reorganize monolithic test file into many smaller files fix #70

Author: bckohan

doc/source/changelog.rst

@@ -5,6 +5,7 @@ Change Log
 v2.0.0
 ======
 
+* Completed `Reorganize tests <https://github.com/bckohan/django-enum/issues/70>`_
 * Implemented `Add database constraints on enum fields by default. <https://github.com/bckohan/django-enum/issues/45>`_
 * Implemented `Provide an optional enum path converter. <https://github.com/bckohan/django-enum/issues/22>`_
 * Implemented `Supply a mixin for DRF ModelSerializers that instantiates the provided DRF EnumField type for model EnumFields. <https://github.com/bckohan/django-enum/issues/47>`_

pyproject.toml

@@ -148,7 +148,7 @@ sphinx = true
 [tool.pytest.ini_options]
 # py.test options:
 DJANGO_SETTINGS_MODULE = "tests.settings"
-python_files = "tests.py"
+python_files = "test*.py"
 norecursedirs = "*.egg .eggs dist build docs .tox .git __pycache__"
 
 addopts = [

tests/test_admin.py

@@ -0,0 +1,53 @@
+from tests.utils import EnumTypeMixin
+from django.test import LiveServerTestCase
+from tests.djenum.models import AdminDisplayBug35
+from django.urls import reverse
+
+
+class TestAdmin(EnumTypeMixin, LiveServerTestCase):
+
+    BUG35_CLASS = AdminDisplayBug35
+
+    def test_admin_list_display_bug35(self):
+        from django.contrib.auth import get_user_model
+
+        get_user_model().objects.create_superuser(
+            username="admin",
+            email="[email protected]",
+            password="admin_password",
+        )
+        self.client.login(username="admin", password="admin_password")
+
+        obj = self.BUG35_CLASS.objects.create()
+
+        resp = self.client.get(
+            reverse(
+                f'admin:{self.BUG35_CLASS._meta.label_lower.replace(".", "_")}_changelist'
+            )
+        )
+        self.assertContains(resp, '<td class="field-int_enum">Value 2</td>')
+        change_link = reverse(
+            f'admin:{self.BUG35_CLASS._meta.label_lower.replace(".", "_")}_change',
+            args=[obj.id],
+        )
+        self.assertContains(resp, f'<a href="{change_link}">Value1</a>')
+
+    def test_admin_change_display_bug35(self):
+        from django.contrib.auth import get_user_model
+
+        get_user_model().objects.create_superuser(
+            username="admin",
+            email="[email protected]",
+            password="admin_password",
+        )
+        self.client.login(username="admin", password="admin_password")
+
+        obj = self.BUG35_CLASS.objects.create()
+        resp = self.client.get(
+            reverse(
+                f'admin:{self.BUG35_CLASS._meta.label_lower.replace(".", "_")}_change',
+                args=[obj.id],
+            )
+        )
+        self.assertContains(resp, '<div class="readonly">Value1</div>')
+        self.assertContains(resp, '<div class="readonly">Value 2</div>')

tests/test_admin_ep.py

@@ -0,0 +1,12 @@
+import pytest
+
+pytest.importorskip("enum_properties")
+
+from tests.test_admin import TestAdmin
+from tests.enum_prop.models import AdminDisplayBug35
+
+class TestEnumPropAdmin(TestAdmin):
+
+    BUG35_CLASS = AdminDisplayBug35
+
+TestAdmin = None

tests/test_bulk.py

@@ -0,0 +1,88 @@
+from django.test import TestCase
+from tests.utils import EnumTypeMixin
+from tests.djenum.models import EnumTester
+
+
+class TestBulkOperations(EnumTypeMixin, TestCase):
+    """Tests bulk insertions and updates"""
+
+    MODEL_CLASS = EnumTester
+    NUMBER = 250
+
+    def setUp(self):
+        self.MODEL_CLASS.objects.all().delete()
+
+    @property
+    def create_params(self):
+        return {
+            "small_pos_int": self.SmallPosIntEnum.VAL2,
+            "small_int": self.SmallIntEnum.VALn1,
+            "pos_int": 2147483647,
+            "int": -2147483648,
+            "big_pos_int": self.BigPosIntEnum.VAL3,
+            "big_int": self.BigIntEnum.VAL2,
+            "constant": self.Constants.GOLDEN_RATIO,
+            "text": self.TextEnum.VALUE2,
+            "extern": self.ExternEnum.TWO,
+            "date_enum": self.DateEnum.HUGO,
+            "datetime_enum": self.DateTimeEnum.KATRINA,
+            "time_enum": self.TimeEnum.COB,
+            "duration_enum": self.DurationEnum.FORTNIGHT,
+            "decimal_enum": self.DecimalEnum.FIVE,
+            "dj_int_enum": 3,
+            "dj_text_enum": self.DJTextEnum.A,
+            "non_strict_int": 15,
+            "non_strict_text": "arbitrary",
+            "no_coerce": "0",
+        }
+
+    @property
+    def update_params(self):
+        return {
+            "non_strict_int": 100,
+            "constant": self.Constants.PI,
+            "big_int": -2147483649,
+            "date_enum": self.DateEnum.BRIAN,
+            "datetime_enum": self.DateTimeEnum.ST_HELENS,
+            "time_enum": self.TimeEnum.LUNCH,
+            "duration_enum": self.DurationEnum.WEEK,
+            "decimal_enum": self.DecimalEnum.TWO,
+        }
+
+    def test_bulk_create(self):
+
+        objects = []
+        for obj in range(0, self.NUMBER):
+            objects.append(self.MODEL_CLASS(**self.create_params))
+
+        self.MODEL_CLASS.objects.bulk_create(objects)
+
+        self.assertEqual(
+            self.MODEL_CLASS.objects.filter(**self.create_params).count(), self.NUMBER
+        )
+
+    def test_bulk_update(self):
+        objects = []
+        for obj in range(0, self.NUMBER):
+            obj = self.MODEL_CLASS.objects.create(**self.create_params)
+            for param, value in self.update_params.items():
+                setattr(obj, param, value)
+            objects.append(obj)
+
+        self.assertEqual(len(objects), self.NUMBER)
+        to_update = ["constant", "non_strict_int"]
+        self.MODEL_CLASS.objects.bulk_update(objects, to_update)
+
+        self.assertEqual(
+            self.MODEL_CLASS.objects.filter(
+                **{
+                    **self.create_params,
+                    **{
+                        param: val
+                        for param, val in self.update_params.items()
+                        if param in to_update
+                    },
+                }
+            ).count(),
+            self.NUMBER,
+        )

tests/test_bulk_ep.py

@@ -0,0 +1,41 @@
+import pytest
+
+pytest.importorskip("enum_properties")
+
+from tests.test_bulk import TestBulkOperations
+from tests.enum_prop.models import EnumTester
+
+    
+class TestBulkOperationsProps(TestBulkOperations):
+    MODEL_CLASS = EnumTester
+
+    @property
+    def create_params(self):
+        return {
+            "small_pos_int": self.SmallPosIntEnum.VAL2,
+            "small_int": "Value -32768",
+            "pos_int": 2147483647,
+            "int": -2147483648,
+            "big_pos_int": "Value 2147483648",
+            "big_int": "VAL2",
+            "constant": "φ",
+            "text": "V TWo",
+            "extern": "One",
+            "dj_int_enum": 3,
+            "dj_text_enum": self.DJTextEnum.A,
+            "non_strict_int": 15,
+            "non_strict_text": "arbitrary",
+            "no_coerce": "Value 2",
+        }
+
+    @property
+    def update_params(self):
+        return {
+            "non_strict_int": 100,
+            "non_strict_text": self.TextEnum.VALUE3,
+            "constant": "π",
+            "big_int": -2147483649,
+            "coerce": 2,
+        }
+
+TestBulkOperations = None