|
1 |
| -from datetime import timedelta |
| 1 | +import datetime as dt |
2 | 2 |
|
3 | 3 | from databricks.sdk.errors import NotFound, InvalidParameterValue
|
4 | 4 | from databricks.sdk.retries import retried
|
5 | 5 |
|
6 | 6 | from ..conftest import MockInstallationContext
|
7 | 7 |
|
8 | 8 |
|
9 |
| -@retried(on=[NotFound, InvalidParameterValue], timeout=timedelta(minutes=12)) |
| 9 | +@retried(on=[NotFound, InvalidParameterValue], timeout=dt.timedelta(minutes=12)) |
10 | 10 | def test_running_real_migration_progress_job(installation_ctx: MockInstallationContext) -> None:
|
11 | 11 | """Ensure that the migration-progress workflow can complete successfully."""
|
12 | 12 | installation_ctx.workspace_installation.run()
|
13 | 13 |
|
14 | 14 | # The assessment workflow is a prerequisite for migration-progress: it needs to successfully complete before we can
|
15 | 15 | # test the migration-progress workflow.
|
16 |
| - installation_ctx.deployed_workflows.run_workflow("assessment") |
17 |
| - assert installation_ctx.deployed_workflows.validate_step("assessment") |
| 16 | + workflow = "assessment" |
| 17 | + installation_ctx.deployed_workflows.run_workflow(workflow, max_wait=dt.timedelta(minutes=25)) |
| 18 | + assert installation_ctx.deployed_workflows.validate_step(workflow), f"Workflow failed: {workflow}" |
18 | 19 |
|
19 | 20 | # After the assessment, a user (maybe) installs the progress tracking
|
20 | 21 | installation_ctx.progress_tracking_installation.run()
|
21 | 22 |
|
22 | 23 | # Run the migration-progress workflow until completion.
|
23 |
| - installation_ctx.deployed_workflows.run_workflow("migration-progress-experimental") |
24 |
| - assert installation_ctx.deployed_workflows.validate_step("migration-progress-experimental") |
| 24 | + workflow = "migration-progress-experimental" |
| 25 | + installation_ctx.deployed_workflows.run_workflow(workflow) |
| 26 | + assert installation_ctx.deployed_workflows.validate_step(workflow), f"Workflow failed: {workflow}" |
25 | 27 |
|
26 | 28 | # Ensure that the migration-progress workflow populated the `workflow_runs` table.
|
27 | 29 | query = f"SELECT 1 FROM {installation_ctx.ucx_catalog}.multiworkspace.workflow_runs"
|
|
0 commit comments