test: fixed unit tests

datahub-project · Dec 16, 2024 · a3160d3 · a3160d3
1 parent 612ffa4
commit a3160d3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/metadata-ingestion/tests/unit/stateful_ingestion/state/test_redundant_run_skip_handler.py b/metadata-ingestion/tests/unit/stateful_ingestion/state/test_redundant_run_skip_handler.py
@@ -37,7 +37,11 @@ def stateful_source(mock_datahub_graph: DataHubGraph) -> Iterable[SnowflakeV2Sou
         ),
     )
 
-    with mock.patch("snowflake.connector.connect"):
+    with mock.patch(
+        "datahub.sql_parsing.sql_parsing_aggregator.ToolMetaExtractor.create",
+    ) as mock_checkpoint, mock.patch("snowflake.connector.connect"):
+        mock_checkpoint.return_value = mock.MagicMock()
+
         yield SnowflakeV2Source(ctx=ctx, config=config)