Add assert to make sure installation is finished before re-installation

databrickslabs · Jan 17, 2025 · 7c58407 · 7c58407
1 parent f47722c
commit 7c58407
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integration/install/test_installation.py b/tests/integration/install/test_installation.py
@@ -393,7 +393,9 @@ def test_check_inventory_database_exists(ws, installation_ctx):
 
 
 def test_compare_remote_local_install_versions(ws, installation_ctx):
-    installation_ctx.workspace_installation.run()
+    installation_finished = installation_ctx.workspace_installation.run()
+    assert installation_finished
+
     with pytest.raises(
         RuntimeWarning,
         match="UCX workspace remote and local install versions are same and no override is requested. Exiting...",