Changeset 5c72356 in flexoentity for tests/conftest.py

Timestamp:

11/02/25 18:49:14 (2 months ago)

Author:

Enrico Schwass <ennoausberlin@…>

Branches:

master

Children:

bf30018

Parents:

8aa20c7

Message:

fix tests due to simplifying state and type

File:

• tests/conftest.py (modified) (5 diffs)

tests/conftest.py

@@ -4 +4 @@
 from dataclasses import dataclass, field
 from typing import List
-from flexoentity import FlexoEntity, EntityType, EntityState, Domain
+from flexoentity import FlexOID, FlexoEntity, EntityType, EntityState, Domain
 
 @pytest.fixture
@@ -37 +37 @@
 class SingleChoiceQuestion(FlexoEntity):
     """A minimal stub to test FlexoEntity integration."""
+    ENTITY_TYPE = EntityType.ITEM
+
     text: str = ""
     options: List[AnswerOption] = field(default_factory=list)
 
+    def __post_init__(self):
+        # If no FlexOID yet, generate a draft ID now.
+        if not getattr(self, "flexo_id", None):
+            domain_code = (
+                self.domain.domain if isinstance(self.domain, Domain) else "GEN"
+            )
+            self.flexo_id = FlexOID.safe_generate(
+                domain=domain_code,
+                entity_type=EntityType.ITEM.value,     # 'I'
+                estate=EntityState.DRAFT.value,        # 'D'
+                text=self.text_seed or self.text,
+                version=1,
+            )
 
     @classmethod
     def default(cls):
-        return cls(domain=Domain(domain="GEN",
-                                 entity_type=EntityType.DOMAIN,
-                                 state=EntityState.DRAFT),
-                   state=EntityState.DRAFT, entity_type=EntityType.ITEM)
+        return cls(domain=Domain(domain="GEN"))
 
     def to_dict(self):
@@ -68 +80 @@
     @classmethod
     def from_dict(cls, data):
-        obj = cls(
+        obj = cls(domain=Domain(domain="GEN"),
             text=data.get("text", ""),
             options=[AnswerOption.from_dict(o) for o in data.get("options", [])],
@@ -74 +86 @@
         # restore FlexoEntity core fields
         obj.domain = data.get("domain")
-        obj.entity_type = EntityType[data.get("etype")] if "etype" in data else EntityType.ITEM
-        obj.state = EntityState[data.get("state")] if "state" in data else EntityState.DRAFT
         if "flexo_id" in data:
-            from flexoentity import FlexOID
             obj.flexo_id = FlexOID.parsed(data["flexo_id"])
         return obj
@@ -87 +96 @@
 @pytest.fixture
 def sample_question():
-    return SingleChoiceQuestion(domain=Domain.default(),
-                               text="What is 2 + 2?",
-                               options=[],
-                               entity_type=EntityType.ITEM,
-                               state=EntityState.DRAFT)
+    q = SingleChoiceQuestion(domain=Domain.default(),
+                             text="What is 2 + 2?",
+                             options=[])
+    q._update_fingerprint()
+    return q