Skip to content

Commit 96fae54

Browse files
committed
tweak tests for coverage
1 parent 972a67d commit 96fae54

File tree

3 files changed

+12
-5
lines changed

3 files changed

+12
-5
lines changed

tests/djenum/models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -262,9 +262,10 @@ class EnumFlagTester(EnumFlagTesterBase):
262262

263263
extra_big_neg = EnumField(
264264
ExtraBigNegativeFlagEnum,
265-
default=ExtraBigNegativeFlagEnum(0),
265+
default=None,
266266
db_index=True,
267267
blank=True,
268+
null=True,
268269
)
269270

270271
def __repr__(self):

tests/enum_prop/models.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -431,10 +431,7 @@ class BaseEnumFlagPropTester(models.Model):
431431
)
432432

433433
extra_big_neg = EnumField(
434-
ExtraBigNegativeFlagEnum,
435-
default=ExtraBigNegativeFlagEnum(0),
436-
db_index=True,
437-
blank=True,
434+
ExtraBigNegativeFlagEnum, default=None, db_index=True, blank=True, null=True
438435
)
439436

440437
def __repr__(self):

tests/test_flags.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,3 +448,12 @@ def test_unsupported_flags(self):
448448

449449
with self.assertRaises(FieldError):
450450
self.MODEL_CLASS.objects.filter(**{"field__has_all": EnumClass.ONE})
451+
452+
def test_extra_big_flags(self):
453+
obj = self.MODEL_CLASS.objects.create()
454+
obj.refresh_from_db()
455+
self.assertTrue(obj.extra_big_neg is None)
456+
self.assertEqual(obj.extra_big_pos, 0)
457+
458+
self.assertEqual(obj, self.MODEL_CLASS.objects.get(extra_big_pos=0))
459+
self.assertEqual(obj, self.MODEL_CLASS.objects.get(extra_big_neg__isnull=True))

0 commit comments

Comments
 (0)