tweak tests for coverage

bckohan · Sep 7, 2024 · 96fae54 · 96fae54
1 parent 972a67d
commit 96fae54
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/tests/djenum/models.py b/tests/djenum/models.py
@@ -262,9 +262,10 @@ class EnumFlagTester(EnumFlagTesterBase):
 
     extra_big_neg = EnumField(
         ExtraBigNegativeFlagEnum,
-        default=ExtraBigNegativeFlagEnum(0),
+        default=None,
         db_index=True,
         blank=True,
+        null=True,
     )
 
     def __repr__(self):

diff --git a/tests/enum_prop/models.py b/tests/enum_prop/models.py
@@ -431,10 +431,7 @@ class BaseEnumFlagPropTester(models.Model):
     )
 
     extra_big_neg = EnumField(
-        ExtraBigNegativeFlagEnum,
-        default=ExtraBigNegativeFlagEnum(0),
-        db_index=True,
-        blank=True,
+        ExtraBigNegativeFlagEnum, default=None, db_index=True, blank=True, null=True
     )
 
     def __repr__(self):

diff --git a/tests/test_flags.py b/tests/test_flags.py
@@ -448,3 +448,12 @@ def test_unsupported_flags(self):
 
             with self.assertRaises(FieldError):
                 self.MODEL_CLASS.objects.filter(**{"field__has_all": EnumClass.ONE})
+
+    def test_extra_big_flags(self):
+        obj = self.MODEL_CLASS.objects.create()
+        obj.refresh_from_db()
+        self.assertTrue(obj.extra_big_neg is None)
+        self.assertEqual(obj.extra_big_pos, 0)
+
+        self.assertEqual(obj, self.MODEL_CLASS.objects.get(extra_big_pos=0))
+        self.assertEqual(obj, self.MODEL_CLASS.objects.get(extra_big_neg__isnull=True))