run optional dependency tests in real missing and fake missing contexts

Author: bckohan

django_enum/tests/tests.py

@@ -25,6 +25,7 @@
 from django_enum.tests.djenum.models import EnumTester
 from django_test_migrations.constants import MIGRATION_TEST_MARKER
 from django_test_migrations.contrib.unittest_case import MigratorTestCase
+from django_enum import filters
 
 
 def set_models(version):
@@ -346,6 +347,76 @@ def test_clean(self):
             self.assertTrue('constant' in ve.message_dict)
             self.assertTrue('text' in ve.message_dict)
 
+    def test_django_filters_missing(self):
+        import sys
+        from importlib import reload
+        from unittest.mock import patch
+
+        with patch.dict(sys.modules, {'django_filters': None}):
+            reload(sys.modules['django_enum.filters'])
+            from django_enum.filters import EnumFilter
+            from django_enum.filters import FilterSet as EnumFilterSet
+
+            class EnumTesterFilter(EnumFilterSet):
+                class Meta:
+                    model = EnumTester
+                    fields = '__all__'
+
+            self.assertRaises(ImportError, EnumTesterFilter)
+            self.assertRaises(ImportError, EnumFilter)
+
+    def test_enum_properties_missing(self):
+        import enum
+        import sys
+        from importlib import reload
+        from unittest.mock import patch
+
+        with patch.dict(sys.modules, {'enum_properties': None}):
+            reload(sys.modules['django_enum.choices'])
+            from django_enum.choices import (
+                DjangoEnumPropertiesMeta,
+                DjangoSymmetricMixin,
+                FloatChoices,
+                IntegerChoices,
+                TextChoices,
+            )
+
+            with self.assertRaises(ImportError):
+                class ThrowsEnum(DjangoSymmetricMixin, enum.Enum):
+                    A = 1
+                    B = 2
+                    C = 3
+
+            with self.assertRaises(ImportError):
+                class ThrowsEnum(
+                    enum.Enum,
+                    metaclass=DjangoEnumPropertiesMeta
+                ):
+                    A = 1
+                    B = 2
+                    C = 3
+
+            with self.assertRaises(ImportError):
+                class ThrowsEnum(IntegerChoices):
+                    A = 1
+                    B = 2
+                    C = 3
+
+            with self.assertRaises(ImportError):
+                class ThrowsEnum(TextChoices):
+                    A = 'A'
+                    B = 'B'
+                    C = 'C'
+
+            with self.assertRaises(ImportError):
+                class ThrowsEnum(FloatChoices):
+                    A = 1.1
+                    B = 2.2
+                    C = 3.3
+
+            self.do_test_integer_choices()
+            self.do_test_text_choices()
+
 
 class TestFieldTypeResolution(EnumTypeMixin, TestCase):
 
@@ -2048,77 +2119,6 @@ def test_validate(self):
             self.assertTrue(tester._meta.get_field('dj_text_enum').validate('A', tester) is None)
             self.assertTrue(tester._meta.get_field('non_strict_int').validate(20, tester) is None)
 
-        def test_django_filters_missing(self):
-            import sys
-            from importlib import reload
-            from unittest.mock import patch
-
-            with patch.dict(sys.modules, {'django_filters': None}):
-                reload(sys.modules['django_enum.filters'])
-                from django_enum.filters import EnumFilter
-                from django_enum.filters import FilterSet as EnumFilterSet
-
-                class EnumTesterFilter(EnumFilterSet):
-                    class Meta:
-                        model = EnumTester
-                        fields = '__all__'
-
-                self.assertRaises(ImportError, EnumTesterFilter)
-                self.assertRaises(ImportError, EnumFilter)
-
-        def test_enum_properties_missing(self):
-            import enum
-            import sys
-            from importlib import reload
-            from unittest.mock import patch
-
-            with patch.dict(sys.modules, {'enum_properties': None}):
-                reload(sys.modules['django_enum.choices'])
-                from django_enum.choices import (
-                    DjangoEnumPropertiesMeta,
-                    DjangoSymmetricMixin,
-                    FloatChoices,
-                    IntegerChoices,
-                    TextChoices,
-                )
-
-                with self.assertRaises(ImportError):
-                    class ThrowsEnum(DjangoSymmetricMixin, enum.Enum):
-                        A = 1
-                        B = 2
-                        C = 3
-
-                with self.assertRaises(ImportError):
-                    class ThrowsEnum(
-                        enum.Enum,
-                        metaclass=DjangoEnumPropertiesMeta
-                    ):
-                        A = 1
-                        B = 2
-                        C = 3
-
-                with self.assertRaises(ImportError):
-                    class ThrowsEnum(IntegerChoices):
-                        A = 1
-                        B = 2
-                        C = 3
-
-                with self.assertRaises(ImportError):
-                    class ThrowsEnum(TextChoices):
-                        A = 'A'
-                        B = 'B'
-                        C = 'C'
-
-                with self.assertRaises(ImportError):
-                    class ThrowsEnum(FloatChoices):
-                        A = 1.1
-                        B = 2.2
-                        C = 3.3
-
-                self.do_test_integer_choices()
-                self.do_test_text_choices()
-
-
     class PerformanceTest(TestCase):
 
         COUNT = 10000