Test refresh report writes query problems

databrickslabs · Sep 25, 2024 · 04fa9d8 · 04fa9d8
1 parent 733146e
commit 04fa9d8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
@@ -111,5 +111,5 @@ def mock_notebook_resolver():
 
 
 @pytest.fixture
-def mock_backend():
+def mock_backend() -> MockBackend:
     return MockBackend()
diff --git a/tests/unit/source_code/test_queries.py b/tests/unit/source_code/test_queries.py
@@ -33,3 +33,13 @@ def test_query_linter_collects_dfsas_from_queries(name, query, dfsa_paths, is_re
     assert set(dfsa.path for dfsa in dfsas) == set(dfsa_paths)
     assert all(dfsa.is_read == is_read for dfsa in dfsas)
     assert all(dfsa.is_write == is_write for dfsa in dfsas)
+
+
+def test_query_liner_refresh_report_writes_query_problems(migration_index, mock_backend) -> None:
+    ws = create_autospec(WorkspaceClient)
+    crawlers = create_autospec(DirectFsAccessCrawler)
+    linter = QueryLinter(ws, migration_index, crawlers, None)
+
+    linter.refresh_report(mock_backend, inventory_database="test")
+
+    assert mock_backend.has_rows_written_for("`test`.query_problems")