add in unknown parsing for WFTypes (#653)

v7labs · Sep 7, 2023 · a4bc974 · a4bc974
1 parent 3173988
commit a4bc974
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/darwin/future/data_objects/workflow.py b/darwin/future/data_objects/workflow.py
@@ -87,6 +87,11 @@ class WFType(Enum):
     ARCHIVE = "archive"
     CONSENSUS_TEST = "consensus_test"
     CONSENSUS_ENTRYPOINT = "consensus_entrypoint"
+    UNKNOWN = "unknown"
+
+    @classmethod
+    def _missing_(cls, value: str) -> "WFType":
+        return WFType.UNKNOWN
 
 
 class WFUser(DefaultDarwin):

diff --git a/darwin/future/tests/meta/queries/test_stage.py b/darwin/future/tests/meta/queries/test_stage.py
@@ -35,6 +35,11 @@ def multi_stage_workflow_object(base_single_workflow_object: dict) -> dict:
     return base_single_workflow_object
 
 
+def test_WFTypes_accept_unknonwn() -> None:
+    assert WFType("unknown") == WFType.UNKNOWN
+    assert WFType("test") == WFType.UNKNOWN
+
+
 def test_stage_collects_basic(
     filled_query: StageQuery, base_single_workflow_object: dict, base_workflow_meta: WorkflowMeta
 ) -> None: