Add discontinue product workflow

Author: AndreuCodina

api/src/api/dependency_container.py

@@ -1,6 +1,9 @@
 from azure.cosmos import CosmosClient, PartitionKey
 
 from api.application_settings import ApplicationSettings
+from api.workflows.products.discontinue_product.discontinue_product_workflow import (
+    DiscontinueProductWorkflow,
+)
 from api.workflows.products.publish_product.publish_product_workflow import (
     PublishProductWorkflow,
 )
@@ -46,3 +49,9 @@ def get_publish_product_workflow(cls) -> PublishProductWorkflow:
         return PublishProductWorkflow(
             application_settings=cls.get_application_settings()
         )
+
+    @classmethod
+    def get_discontinue_product_workflow(cls) -> DiscontinueProductWorkflow:
+        return DiscontinueProductWorkflow(
+            application_settings=cls.get_application_settings()
+        )

api/src/api/workflows/products/discontinue_product/__init__.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+
+class DiscontinueProductRequest(BaseModel):
+    id: str
+    discontinuation_reason: str | None = None

api/src/api/workflows/products/discontinue_product/discontinue_product_request.py

@@ -0,0 +1,28 @@
+from azure.cosmos.aio import CosmosClient
+
+from api.application_settings import ApplicationSettings
+from api.workflows.products.discontinue_product.discontinue_product_request import (
+    DiscontinueProductRequest,
+)
+from domain.entities.product import Product
+
+
+class DiscontinueProductWorkflow:
+    def __init__(self, application_settings: ApplicationSettings) -> None:
+        self.application_settings = application_settings
+
+    async def execute(self, request: DiscontinueProductRequest) -> None:
+        async with CosmosClient(
+            self.application_settings.cosmos_db_no_sql_url,
+            self.application_settings.cosmos_db_no_sql_key.get_secret_value(),
+        ) as cosmosdb_client:
+            cosmosdb_database = cosmosdb_client.get_database_client(
+                self.application_settings.cosmos_db_no_sql_database
+            )
+            product_container = cosmosdb_database.get_container_client(Product.__name__)
+            product = await product_container.read_item(
+                item=request.id, partition_key=request.id
+            )
+            product = Product.model_validate(product)
+            product.discontinue(request.discontinuation_reason)
+            await product_container.replace_item(product.id, product.model_dump())

api/src/api/workflows/products/discontinue_product/discontinue_product_workflow.py

@@ -1,6 +1,9 @@
 from fastapi import APIRouter
 
 from api.dependency_container import DependencyContainer
+from api.workflows.products.discontinue_product.discontinue_product_request import (
+    DiscontinueProductRequest,
+)
 from api.workflows.products.publish_product.publish_product_request import (
     PublishProductRequest,
 )
@@ -17,3 +20,8 @@
 @router.post("")
 async def publish_product(request: PublishProductRequest) -> PublishProductResponse:
     return await DependencyContainer.get_publish_product_workflow().execute(request)
+
+
+@router.post("/discontinue")
+async def discontinue_product(request: DiscontinueProductRequest) -> None:
+    return await DependencyContainer.get_discontinue_product_workflow().execute(request)

api/src/api/workflows/products/product_router.py

@@ -1,8 +1,25 @@
+from collections.abc import AsyncGenerator
+
 import pytest
+from azure.cosmos.aio import CosmosClient, DatabaseProxy
 
 from api.dependency_container import DependencyContainer
 
 
 @pytest.fixture(autouse=True)
 def initialize_dependency_container() -> None:
     DependencyContainer.initialize()
+
+
+@pytest.fixture
+async def cosmosdb_database() -> AsyncGenerator[DatabaseProxy, None]:
+    application_settings = DependencyContainer.get_application_settings()
+
+    async with CosmosClient(
+        application_settings.cosmos_db_no_sql_url,
+        application_settings.cosmos_db_no_sql_key.get_secret_value(),
+    ) as cosmosdb_client:
+        cosmosdb_database = cosmosdb_client.get_database_client(
+            application_settings.cosmos_db_no_sql_database
+        )
+        yield cosmosdb_database

api/tests/integration/conftest.py

@@ -0,0 +1,22 @@
+import pytest
+from azure.cosmos.aio import DatabaseProxy
+
+from api.dependency_container import DependencyContainer
+from api.workflows.products.discontinue_product.discontinue_product_request import (
+    DiscontinueProductRequest,
+)
+from domain.entities.product import Product
+from test_utils.builders.domain.entities.product_builder import ProductBuilder
+
+
+@pytest.mark.integration
+class TestDiscontinueProductWorkflow:
+    async def test_discontinue_product(self, cosmosdb_database: DatabaseProxy) -> None:
+        product = ProductBuilder().build()
+        await cosmosdb_database.get_container_client(Product.__name__).create_item(
+            product.model_dump()
+        )
+        request = DiscontinueProductRequest(id=product.id)
+        workflow = DependencyContainer.get_discontinue_product_workflow()
+
+        await workflow.execute(request)