Add scheduled run name to the test.

sfczekalski · sfczekalski · commit 6beaf8d16ed3 · 2024-08-06T11:45:15.000+02:00
diff --git a/tests/test_vertex_ai_client.py b/tests/test_vertex_ai_client.py
@@ -19,6 +19,7 @@ def create_client(self):
                     "root": "BUCKET/PREFIX",
                     "network": {"vpc": "my-vpc"},
                     "experiment_name": "experiment-name",
+                    "scheduled_run_name": "scheduled-run",
                 },
             }
         )
@@ -80,10 +81,9 @@ def test_should_schedule_pipeline(self):
                 ScheduleConfig(cron_expression="0 0 12 * *", timezone="Etc/UTC"),
             )
 
-            # job.create_schedule.assert_not_called()
             _, kwargs = job.create_schedule.call_args
             assert kwargs["cron"] == "TZ=Etc/UTC 0 0 12 * *"
-            assert kwargs["display_name"] is None
+            assert kwargs["display_name"] == "scheduled-run"
             assert kwargs["start_time"] is None
             assert kwargs["end_time"] is None
             assert kwargs["allow_queueing"] is False

Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,7 @@ def create_client(self):`
`19`	`19`	`"root": "BUCKET/PREFIX",`
`20`	`20`	`"network": {"vpc": "my-vpc"},`
`21`	`21`	`"experiment_name": "experiment-name",`
	`22`	`+ "scheduled_run_name": "scheduled-run",`
`22`	`23`	`},`
`23`	`24`	`}`
`24`	`25`	`)`
`@@ -80,10 +81,9 @@ def test_should_schedule_pipeline(self):`
`80`	`81`	`ScheduleConfig(cron_expression="0 0 12 * *", timezone="Etc/UTC"),`
`81`	`82`	`)`
`82`	`83`
`83`		`- # job.create_schedule.assert_not_called()`
`84`	`84`	`_, kwargs = job.create_schedule.call_args`
`85`	`85`	`assert kwargs["cron"] == "TZ=Etc/UTC 0 0 12 * *"`
`86`		`- assert kwargs["display_name"] is None`
	`86`	`+ assert kwargs["display_name"] == "scheduled-run"`
`87`	`87`	`assert kwargs["start_time"] is None`
`88`	`88`	`assert kwargs["end_time"] is None`
`89`	`89`	`assert kwargs["allow_queueing"] is False`