|
40 | 40 |
|
41 | 41 | _TEST_PROJECT = "test-project"
|
42 | 42 | _TEST_LOCATION = "us-central1"
|
| 43 | +_TEST_PIPELINE_JOB_DISPLAY_NAME = "sample-pipeline-job-display-name" |
43 | 44 | _TEST_PIPELINE_JOB_ID = "sample-test-pipeline-202111111"
|
44 | 45 | _TEST_GCS_BUCKET_NAME = "my-bucket"
|
45 | 46 | _TEST_CREDENTIALS = auth_credentials.AnonymousCredentials()
|
@@ -199,7 +200,7 @@ def test_run_call_pipeline_service_create(
|
199 | 200 | )
|
200 | 201 |
|
201 | 202 | job = pipeline_jobs.PipelineJob(
|
202 |
| -display_name=_TEST_PIPELINE_JOB_ID, |
| 203 | +display_name=_TEST_PIPELINE_JOB_DISPLAY_NAME, |
203 | 204 | template_path=_TEST_TEMPLATE_PATH,
|
204 | 205 | job_id=_TEST_PIPELINE_JOB_ID,
|
205 | 206 | parameter_values=_TEST_PIPELINE_PARAMETER_VALUES,
|
@@ -222,7 +223,7 @@ def test_run_call_pipeline_service_create(
|
222 | 223 |
|
223 | 224 | # Construct expected request
|
224 | 225 | expected_gapic_pipeline_job = gca_pipeline_job_v1beta1.PipelineJob(
|
225 |
| -display_name=_TEST_PIPELINE_JOB_ID, |
| 226 | +display_name=_TEST_PIPELINE_JOB_DISPLAY_NAME, |
226 | 227 | name=_TEST_PIPELINE_JOB_NAME,
|
227 | 228 | pipeline_spec={
|
228 | 229 | "components": {},
|
@@ -233,7 +234,9 @@ def test_run_call_pipeline_service_create(
|
233 | 234 | )
|
234 | 235 |
|
235 | 236 | mock_pipeline_service_create.assert_called_once_with(
|
236 |
| -parent=_TEST_PARENT, pipeline_job=expected_gapic_pipeline_job, |
| 237 | +parent=_TEST_PARENT, |
| 238 | +pipeline_job=expected_gapic_pipeline_job, |
| 239 | +pipeline_job_id=_TEST_PIPELINE_JOB_ID, |
237 | 240 | )
|
238 | 241 |
|
239 | 242 | mock_pipeline_service_get.assert_called_with(name=_TEST_PIPELINE_JOB_NAME)
|
@@ -242,6 +245,14 @@ def test_run_call_pipeline_service_create(
|
242 | 245 | gca_pipeline_state_v1beta1.PipelineState.PIPELINE_STATE_SUCCEEDED
|
243 | 246 | )
|
244 | 247 |
|
| 248 | +@pytest.mark.usefixtures("mock_pipeline_service_get") |
| 249 | +def test_get_pipeline_job(self, mock_pipeline_service_get): |
| 250 | +aiplatform.init(project=_TEST_PROJECT) |
| 251 | +job = pipeline_jobs.PipelineJob.get(resource_name=_TEST_PIPELINE_JOB_ID) |
| 252 | + |
| 253 | +mock_pipeline_service_get.assert_called_once_with(name=_TEST_PIPELINE_JOB_NAME) |
| 254 | +assert isinstance(job, pipeline_jobs.PipelineJob) |
| 255 | + |
245 | 256 | @pytest.mark.usefixtures(
|
246 | 257 | "mock_pipeline_service_create", "mock_pipeline_service_get", "mock_load_json",
|
247 | 258 | )
|
@@ -255,7 +266,7 @@ def test_cancel_pipeline_job(
|
255 | 266 | )
|
256 | 267 |
|
257 | 268 | job = pipeline_jobs.PipelineJob(
|
258 |
| -display_name=_TEST_PIPELINE_JOB_ID, |
| 269 | +display_name=_TEST_PIPELINE_JOB_DISPLAY_NAME, |
259 | 270 | template_path=_TEST_TEMPLATE_PATH,
|
260 | 271 | job_id=_TEST_PIPELINE_JOB_ID,
|
261 | 272 | )
|
@@ -267,14 +278,6 @@ def test_cancel_pipeline_job(
|
267 | 278 | name=_TEST_PIPELINE_JOB_NAME
|
268 | 279 | )
|
269 | 280 |
|
270 |
| -@pytest.mark.usefixtures("mock_pipeline_service_get") |
271 |
| -def test_get_training_job(self, mock_pipeline_service_get): |
272 |
| -aiplatform.init(project=_TEST_PROJECT) |
273 |
| -job = pipeline_jobs.PipelineJob.get(resource_name=_TEST_PIPELINE_JOB_ID) |
274 |
| - |
275 |
| -mock_pipeline_service_get.assert_called_once_with(name=_TEST_PIPELINE_JOB_NAME) |
276 |
| -assert isinstance(job, pipeline_jobs.PipelineJob) |
277 |
| - |
278 | 281 | @pytest.mark.usefixtures(
|
279 | 282 | "mock_pipeline_service_create", "mock_pipeline_service_get", "mock_load_json",
|
280 | 283 | )
|
@@ -288,7 +291,7 @@ def test_cancel_pipeline_job_without_running(
|
288 | 291 | )
|
289 | 292 |
|
290 | 293 | job = pipeline_jobs.PipelineJob(
|
291 |
| -display_name=_TEST_PIPELINE_JOB_ID, |
| 294 | +display_name=_TEST_PIPELINE_JOB_DISPLAY_NAME, |
292 | 295 | template_path=_TEST_TEMPLATE_PATH,
|
293 | 296 | job_id=_TEST_PIPELINE_JOB_ID,
|
294 | 297 | )
|
@@ -313,7 +316,7 @@ def test_pipeline_failure_raises(self, sync):
|
313 | 316 | )
|
314 | 317 |
|
315 | 318 | job = pipeline_jobs.PipelineJob(
|
316 |
| -display_name=_TEST_PIPELINE_JOB_ID, |
| 319 | +display_name=_TEST_PIPELINE_JOB_DISPLAY_NAME, |
317 | 320 | template_path=_TEST_TEMPLATE_PATH,
|
318 | 321 | job_id=_TEST_PIPELINE_JOB_ID,
|
319 | 322 | parameter_values=_TEST_PIPELINE_PARAMETER_VALUES,
|
|
0 commit comments