Skip to content

Commit b60cab7

Browse files
committed
fixed unit test of get_deployment_status
1 parent 95e5a04 commit b60cab7

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/unitary/with_extras/aqua/test_deployment.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2369,6 +2369,7 @@ def test_validate_multimodel_deployment_feasibility_positive_single(
23692369
)
23702370

23712371
def test_get_deployment_status(self):
2372+
model_deployment = copy.deepcopy(TestDataset.model_deployment_object[0])
23722373
deployment_id = "fakeid.datasciencemodeldeployment.oc1.iad.xxx"
23732374
work_request_id = "fakeid.workrequest.oc1.iad.xxx"
23742375
model_type = "custom"
@@ -2382,7 +2383,11 @@ def test_get_deployment_status(self):
23822383
"ads.model.service.oci_datascience_model_deployment.DataScienceWorkRequest.wait_work_request"
23832384
) as mock_wait:
23842385
self.app.get_deployment_status(
2385-
deployment_id, work_request_id, model_type, model_name
2386+
model_deployment,
2387+
deployment_id,
2388+
work_request_id,
2389+
model_type,
2390+
model_name,
23862391
)
23872392

23882393
mock_ds_work_request.assert_called_with(work_request_id)

0 commit comments

Comments
 (0)