Test new DAG HTML

goodwright · Jan 25, 2025 · dfd1efa · dfd1efa
1 parent 6ac7caf
commit dfd1efa
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/integration/base.py b/tests/integration/base.py
@@ -80,7 +80,8 @@ def check_execution(self, execution, line_count=24, output_path=None, version=No
         if dag:
             self.assertIn(dag, os.listdir(self.get_path("rundirectory")))
             with open(os.path.join(self.get_path("rundirectory"), dag)) as f:
-                self.assertIn("Cytoscape.js with Dagre", f.read())
+                html = f.read()
+                self.assertTrue("Cytoscape.js with Dagre" in html or "import mermaid" in html)
         else:
             self.assertNotIn(dag, os.listdir(self.get_path("rundirectory")))