passes product router instead of product to backend

Co-authored-by: Tyler <[email protected]>
Co-authored-by: Jarrett Keifer <[email protected]>

Author: 3 people

src/stapi_fastapi/backends/product_backend.py

@@ -6,13 +6,13 @@
 
 from stapi_fastapi.models.opportunity import Opportunity, OpportunityRequest
 from stapi_fastapi.models.order import Order
-from stapi_fastapi.models.product import Product
+from stapi_fastapi.routers.product_router import ProductRouter
 
 
 class ProductBackend(Protocol):
     async def search_opportunities(
         self,
-        product: Product,
+        product_router: ProductRouter,
         search: OpportunityRequest,
         request: Request,
     ) -> list[Opportunity]:
@@ -26,7 +26,7 @@ async def search_opportunities(
 
     async def create_order(
         self,
-        product: Product,
+        product_router: ProductRouter,
         search: OpportunityRequest,
         request: Request,
     ) -> Order:

src/stapi_fastapi/routers/product_router.py

@@ -90,7 +90,7 @@ async def search_opportunities(
         """
         try:
             opportunities = await self.product.backend.search_opportunities(
-                self.product, search, request
+                self, search, request
             )
         except ConstraintsException as exc:
             raise HTTPException(status.HTTP_422_UNPROCESSABLE_ENTITY, detail=exc.detail)
@@ -111,7 +111,7 @@ async def create_order(
         """
         try:
             order = await self.product.backend.create_order(
-                self.product,
+                self,
                 payload,
                 request,
             )

tests/backends.py

@@ -5,7 +5,7 @@
 from stapi_fastapi.exceptions import ConstraintsException, NotFoundException
 from stapi_fastapi.models.opportunity import Opportunity, OpportunityRequest
 from stapi_fastapi.models.order import Order
-from stapi_fastapi.models.product import Product
+from stapi_fastapi.routers.product_router import ProductRouter
 
 
 class MockOrderDB(dict[int | str, Order]):
@@ -39,12 +39,18 @@ def __init__(self, orders: MockOrderDB) -> None:
         self._orders = orders
 
     async def search_opportunities(
-        self, product: Product, search: OpportunityRequest, request: Request
+        self,
+        product_router: ProductRouter,
+        search: OpportunityRequest,
+        request: Request,
     ) -> list[Opportunity]:
         return [o.model_copy(update=search.model_dump()) for o in self._opportunities]
 
     async def create_order(
-        self, product: Product, payload: OpportunityRequest, request: Request
+        self,
+        product_router: ProductRouter,
+        payload: OpportunityRequest,
+        request: Request,
     ) -> Order:
         """
         Create a new order.
@@ -57,7 +63,7 @@ async def create_order(
                 properties={
                     "filter": payload.filter,
                     "datetime": payload.datetime,
-                    "product_id": product.id,
+                    "product_id": product_router.product.id,
                 },
                 links=[],
             )