Fixes failing test

PrefectHQ · Jul 19, 2024 · cde51ec · cde51ec
1 parent a62d9b2
commit cde51ec
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/deployment/test_base.py b/tests/deployment/test_base.py
@@ -150,7 +150,7 @@ async def test_initialize_project_with_docker_recipe_default_image(self, recipe)
 class TestDiscoverFlows:
     async def test_find_all_flows_in_dir_tree(self, project_dir):
         flows = await _search_for_flow_functions(str(project_dir))
-        assert len(flows) == 6, f"Expected 6 flows, found {len(flows)}"
+        assert len(flows) == 7, f"Expected 7 flows, found {len(flows)}"
 
         expected_flows = [
             {
@@ -191,6 +191,11 @@ async def test_find_all_flows_in_dir_tree(self, project_dir):
                     project_dir / "import-project" / "my_module" / "flow.py"
                 ),
             },
+            {
+                "flow_name": "uses_block",
+                "function_name": "uses_block",
+                "filepath": str(project_dir / "flows" / "uses_block.py"),
+            },
         ]
 
         for flow in flows: