Skip to content

Commit affe04a

Browse files
committed
Fix workflow_start_time should be start_time
1 parent f6473de commit affe04a

File tree

3 files changed

+5
-2
lines changed

3 files changed

+5
-2
lines changed

src/databricks/labs/ucx/contexts/workflow_task.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,5 +116,5 @@ def workflow_run_recorder(self):
116116
workflow_id=int(self.named_parameters["job_id"]),
117117
workflow_run_id=int(self.named_parameters["parent_run_id"]),
118118
workflow_run_attempt=int(self.named_parameters.get("attempt", 0)),
119-
workflow_start_time=self.named_parameters["workflow_start_time"],
119+
workflow_start_time=self.named_parameters["start_time"],
120120
)

tests/integration/progress/test_workflow_runs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ def test_workflow_run_recorder_records_workflow_run(installation_ctx) -> None:
99
"workflow": "test",
1010
"job_id": "123",
1111
"parent_run_id": "456",
12-
"workflow_start_time": start_time.isoformat(),
12+
"start_time": start_time.isoformat(),
1313
}
1414
ctx = installation_ctx.replace(named_parameters=named_parameters)
1515
ctx.progress_tracking_installation.run()

tests/integration/progress/test_workflows.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,9 @@ def test_running_real_migration_progress_job(installation_ctx: MockInstallationC
1616
installation_ctx.deployed_workflows.run_workflow("assessment")
1717
assert installation_ctx.deployed_workflows.validate_step("assessment")
1818

19+
# After the assessment, a user (maybe) installs the progress tracking
20+
installation_ctx.progress_tracking_installation.run()
21+
1922
# Run the migration-progress workflow until completion.
2023
installation_ctx.deployed_workflows.run_workflow("migration-progress-experimental")
2124
assert installation_ctx.deployed_workflows.validate_step("migration-progress-experimental")

0 commit comments

Comments
 (0)