Test for historical records table to be created

Author: JCZuurmond

tests/integration/progress/test_install.py

@@ -1,7 +1,11 @@
-def test_progress_tracking_installer_creates_workflow_runs_table(az_cli_ctx) -> None:
+import pytest
+
+
+@pytest.mark.parametrize("table_name", ["workflow_runs", "historical_records"])
+def test_progress_tracking_installer_creates_table(az_cli_ctx, table_name) -> None:
     az_cli_ctx.progress_tracking_installation.run()
     query = (
         f"SELECT 1 FROM tables WHERE table_catalog = '{az_cli_ctx.config.ucx_catalog}' "
-        "AND table_schema = 'multiworkspace' AND table_name = 'workflow_runs'"
+        f"AND table_schema = 'multiworkspace' AND table_name = '{table_name}'"
     )
     assert any(az_cli_ctx.sql_backend.fetch(query, catalog="system", schema="information_schema"))

tests/unit/progress/test_install.py

@@ -7,8 +7,8 @@ def test_progress_tracking_installation_run_creates_progress_tracking_schema(moc
     assert "CREATE SCHEMA IF NOT EXISTS ucx.multiworkspace" in mock_backend.queries[0]
 
 
-def test_progress_tracking_installation_run_creates_workflow_runs_table(mock_backend) -> None:
+def test_progress_tracking_installation_run_creates_tables(mock_backend) -> None:
     installation = ProgressTrackingInstallation(mock_backend, "ucx")
     installation.run()
     # Dataclass to schema conversion is tested within the lsql package
-    assert any("CREATE TABLE IF NOT EXISTS" in query for query in mock_backend.queries)
+    assert sum("CREATE TABLE IF NOT EXISTS" in query for query in mock_backend.queries) == 2