fix: tests fix

Signed-off-by: 35C4n0r <[email protected]>
35C4n0r · Sep 27, 2024 · 85f11b4 · 85f11b4
1 parent bc1d08f
commit 85f11b4
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/keep/workflowmanager/workflowstore.py b/keep/workflowmanager/workflowstore.py
@@ -307,15 +307,25 @@ def provision_workflows_from_directory(
                         elif "alert" in workflow_yaml:
                             workflow_yaml = workflow_yaml["alert"]
 
-                    workflow_name = workflow_yaml.get("name") or workflow_yaml.get("id")
+                    workflow_id = workflow_yaml.get("id", str(uuid.uuid4()))
+                    workflow_yaml["id"] = workflow_id
+                    workflow_name = workflow_yaml.get(
+                        "name", "Workflow-" + workflow_yaml.get("id")
+                    )
                     if not workflow_name:
                         logger.error(f"Workflow from {file} does not have a name or id")
                         continue
-                    workflow_id = str(uuid.uuid4())
                     workflow_description = workflow_yaml.get("description")
                     workflow_interval = parser.parse_interval(workflow_yaml)
                     workflow_disabled = parser.parse_disabled(workflow_yaml)
-
+                    is_valid = True
+                    try:
+                        parser = Parser()
+                        parser.parse(
+                            tenant_id=tenant_id, parsed_workflow_yaml=workflow_yaml
+                        )
+                    except Exception as e:
+                        is_valid = False
                     add_or_update_workflow(
                         id=workflow_id,
                         name=workflow_name,
@@ -324,6 +334,7 @@ def provision_workflows_from_directory(
                         created_by="system",
                         interval=workflow_interval,
                         is_disabled=workflow_disabled,
+                        is_valid=is_valid,
                         workflow_raw=yaml.dump(workflow_yaml),
                         provisioned=True,
                         provisioned_file=workflow_path,