test workarounds for #123

Author: bckohan

tests/djenum/admin.py

@@ -27,7 +27,14 @@
 
 
 class AdminDisplayBug35Admin(admin.ModelAdmin):
-    list_display = ("text_enum", "int_enum")
+    list_display = (
+        "text_enum",
+        "int_enum",
+        "status_basic",
+        "status_basic_int",
+        "status_str",
+        "status_int",
+    )
     readonly_fields = ("text_enum", "int_enum", "blank_int", "blank_txt")
 
 

tests/djenum/enums.py

@@ -398,3 +398,61 @@ class GNSSConstellation(IntFlag):
     GALILEO = 1 << 2
     BEIDOU = 1 << 3
     QZSS = 1 << 4
+
+
+class NestStatusBasic(Enum):
+    INIT = "INIT"
+    LOADED = "LOADED"
+    ACTIVE = "ACTIVE"
+    DONE = "DONE"
+    REJECTED = "REJECTED"
+    CANCELLED = "CANCELLED"
+
+    def __hash__(self):
+        return hash(self.value)
+
+    def __eq__(self, value) -> bool:
+        if isinstance(value, self.__class__):
+            return self.value == value.value
+        try:
+            return self.value == self.__class__(value).value
+        except (ValueError, TypeError):
+            return False
+
+
+class NestStatusStr(str, Enum):
+    INIT = "INIT"
+    LOADED = "LOADED"
+    ACTIVE = "ACTIVE"
+    DONE = "DONE"
+    REJECTED = "REJECTED"
+    CANCELLED = "CANCELLED"
+
+
+class NestStatusBasicInt(Enum):
+    INIT = 0
+    LOADED = 1
+    ACTIVE = 2
+    DONE = 3
+    REJECTED = 4
+    CANCELLED = 5
+
+    def __hash__(self):
+        return hash(self.value)
+
+    def __eq__(self, value) -> bool:
+        if isinstance(value, self.__class__):
+            return self.value == value.value
+        try:
+            return self.value == self.__class__(value).value
+        except (ValueError, TypeError):
+            return False
+
+
+class NestStatusInt(int, Enum):
+    INIT = 0
+    LOADED = 1
+    ACTIVE = 2
+    DONE = 3
+    REJECTED = 4
+    CANCELLED = 5

tests/djenum/models.py

@@ -41,6 +41,10 @@
     TimeEnum,
     NullableConstants,
     GNSSConstellation,
+    NestStatusBasic,
+    NestStatusInt,
+    NestStatusBasicInt,
+    NestStatusStr,
 )
 
 
@@ -159,6 +163,11 @@ class AdminDisplayBug35(models.Model):
 
     blank_txt = EnumField(TextEnum, null=True, default=None)
 
+    status_basic = EnumField(NestStatusBasic, null=True, default=None)
+    status_basic_int = EnumField(NestStatusBasicInt, null=True, default=None)
+    status_int = EnumField(NestStatusInt, null=True, default=None)
+    status_str = EnumField(NestStatusStr, null=True, default=None)
+
 
 class EmptyEnumValueTester(models.Model):
     class BlankTextEnum(TextChoices):