Fix (mock) prompts

databrickslabs · Sep 17, 2024 · 0a94d02 · 0a94d02
1 parent cf22438
commit 0a94d02
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/databricks/labs/ucx/cli.py b/src/databricks/labs/ucx/cli.py
@@ -547,7 +547,7 @@ def migrate_tables(
             if prompts.confirm(
                 f"Found {len(external_ctas_tables)} ({percentage_external_ctas_tables:.2f}%) external tables which "
                 f"cannot be migrated using sync in {workspace_context.workspace_client.config.host}, do you want to "
-                "run the `migrate-external-tables-ctas workflow`?"
+                "run the `migrate-external-tables-ctas` workflow?"
             ):
                 deployed_workflows.run_workflow("migrate-external-tables-ctas")
 

diff --git a/tests/unit/test_cli.py b/tests/unit/test_cli.py
@@ -750,8 +750,8 @@ def test_migrate_tables_calls_external_hiveserde_tables_job_run_now(ws) -> None:
     )
 
     prompt = (
-        "Found 1 (.*) hiveserde tables, do you want to run the "
-        "migrate-external-hiveserde-tables-in-place-experimental workflow?"
+        "Found 1 (.*) hiveserde tables in https://localhost, do you want to run the "
+        "`migrate-external-hiveserde-tables-in-place-experimental` workflow?"
     )
     prompts = MockPrompts({prompt: "Yes"})
 
@@ -781,8 +781,8 @@ def test_migrate_tables_calls_external_tables_ctas_job_run_now(ws) -> None:
     )
 
     prompt = (
-        "Found 1 (.*) external tables which cannot be migrated using sync, do you want to run the "
-        "migrate-external-tables-ctas workflow?"
+        "Found 1 (.*) external tables which cannot be migrated using sync in https://localhost, do you want to run the "
+        "`migrate-external-tables-ctas` workflow?"
     )
 
     prompts = MockPrompts({prompt: "Yes"})