Add unit tests

Author: AndreuCodina

common/src/common/business_error.py

@@ -0,0 +1,12 @@
+from http import HTTPStatus
+
+from fastapi import HTTPException
+
+
+class BusinessError(HTTPException):
+    def __init__(
+        self,
+        detail: str | None = None,
+        status_code: int = HTTPStatus.UNPROCESSABLE_ENTITY,
+    ) -> None:
+        super().__init__(status_code=status_code, detail=detail)

domain/src/domain/entities/product.py

@@ -2,6 +2,7 @@
 
 from pydantic import Field
 
+from common.business_error import BusinessError
 from domain.entity import Entity
 
 
@@ -11,3 +12,11 @@ class Product(Entity):
     price: Decimal = Field(gt=Decimal("0.0"))
     is_discontinued: bool
     discontinuation_reason: str | None = None
+
+    def discontinue(self, discontinuation_reason: str | None = None) -> None:
+        if self.is_discontinued:
+            error_message = "The product is already discontinued"
+            raise BusinessError(error_message)
+
+        self.is_discontinued = True
+        self.discontinuation_reason = discontinuation_reason

domain/tests/unit/test_product.py

@@ -1,7 +1,18 @@
 import pytest
 
+from common.business_error import BusinessError
+from test_utils.builders.domain.entities.product_builder import ProductBuilder
+
 
 @pytest.mark.unit
 class TestProduct:
-    def test_create_product(self) -> None:
-        assert 1 == 1
+    def test_discontinue_product(self) -> None:
+        product = ProductBuilder().build()
+
+        product.discontinue()
+
+    def test_fail_when_discontinuing_discontinued_product(self) -> None:
+        product = ProductBuilder().discontinued().build()
+
+        with pytest.raises(BusinessError):
+            product.discontinue()

pyproject.toml

@@ -15,6 +15,10 @@ dev = [
     "api",
     "jupyter==1.1.1",
     "pyright[nodejs]==1.1.400",
+    "pytest==8.3.5",
+    "pytest-asyncio==0.26.0",
+    "pytest-cov==6.1.1",
+    "pytest-mock==3.14.0",
     "ruff==0.11.9",
     "test-utils",
     "transcriber",
@@ -54,4 +58,5 @@ ignore = [
 
 [tool.pytest.ini_options]
 asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
 markers = ["unit: Unit tests", "integration: Integration tests"]

test_utils/src/test_utils/builders/__init__.py

@@ -0,0 +1,21 @@
+from decimal import Decimal
+from typing import Self
+
+from domain.entities.product import Product
+
+
+class ProductBuilder:
+    def __init__(self) -> None:
+        self.is_discontinued = False
+
+    def build(self) -> Product:
+        return Product(
+            name="Table",
+            description="Description",
+            price=Decimal("10.0"),
+            is_discontinued=self.is_discontinued,
+        )
+
+    def discontinued(self) -> Self:
+        self.is_discontinued = True
+        return self