🧪 test: add tests for ProductLine model's clean method

Author: Sardorbek-Zayniyev

drfecommerce/test/conftest.py

@@ -2,11 +2,12 @@
 
 from pytest_factoryboy import register
 from rest_framework.test import APIClient
-from .factories import CategoryFactory, BrandFactory, ProductFactory
+from .factories import CategoryFactory, BrandFactory, ProductFactory, ProductLineFactory
 
 register(CategoryFactory)
 register(BrandFactory)
 register(ProductFactory)
+register(ProductLineFactory)
 
 
 @pytest.fixture

drfecommerce/test/factories.py

@@ -1,5 +1,5 @@
 import factory
-from drfecommerce.product.models import Category, Brand, Product
+from drfecommerce.product.models import Category, Brand, Product, ProductLine
 
 
 class CategoryFactory(factory.django.DjangoModelFactory):
@@ -26,3 +26,13 @@ class Meta:
     description = "test_description"
     is_digital = True
     is_active = True
+
+
+class ProductLineFactory(factory.django.DjangoModelFactory):
+    class Meta:
+        model = ProductLine
+    price = 10.00
+    sku = '12345'
+    stock_qty = 1
+    product = factory.SubFactory(ProductFactory)
+    is_active = True

drfecommerce/test/product/test_models.py

@@ -1,4 +1,5 @@
 import pytest
+from django.core.exceptions import ValidationError
 pytestmark = pytest.mark.django_db
 
 
@@ -27,3 +28,18 @@ def test_str_method(self, product_factory):
         obj = product_factory(name="test_product")
         # Assert
         assert obj.__str__() == "test_product"
+
+
+class TestProductLineModel:
+    def test_str_method(self, product_line_factory):
+        # Arrange
+        # Act
+        obj = product_line_factory(sku="12345")
+        # Assert
+        assert obj.__str__() == "12345"
+
+    def test_duplicate_order_values(self, product_line_factory, product_factory):
+        obj = product_factory()
+        product_line_factory(order=1, product=obj)
+        with pytest.raises(ValidationError):
+            product_line_factory(order=1, product=obj).clean()

pytest.ini

@@ -0,0 +1,6 @@
+[tool:pytest]
+DJANGO_SETTINGS_MODULE = drfecommerce.settings.local
+
+python_files = test_*.py
+
+; addopts = --cov