Skip to content

Commit 10d06cc

Browse files
Adding UT for delete model
1 parent e1b985f commit 10d06cc

File tree

1 file changed

+15
-1
lines changed

1 file changed

+15
-1
lines changed

tests/unitary/with_extras/aqua/test_model_handler.py

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919
AquaModelLicenseHandler,
2020
)
2121
from ads.aqua.model import AquaModelApp
22-
from ads.aqua.model.constants import ModelTask
2322
from ads.aqua.model.entities import AquaModel, AquaModelSummary, HFModelSummary
2423

2524

@@ -79,6 +78,21 @@ def test_delete(self, mock_urlparse, mock_clear_model_list_cache):
7978
mock_urlparse.assert_called()
8079
mock_clear_model_list_cache.assert_called()
8180

81+
@patch("ads.aqua.extension.model_handler.urlparse")
82+
@patch.object(AquaModelApp, "delete_model")
83+
def test_delete_with_id(self, mock_delete, mock_urlparse):
84+
request_path = MagicMock(path="aqua/model/ocid1.datasciencemodel.oc1.iad.xxx")
85+
mock_urlparse.return_value = request_path
86+
mock_delete.return_value = {"state": "DELETED"}
87+
with patch(
88+
"ads.aqua.extension.base_handler.AquaAPIhandler.finish"
89+
) as mock_finish:
90+
mock_finish.side_effect = lambda x: x
91+
result = self.model_handler.delete(id="ocid1.datasciencemodel.oc1.iad.xxx")
92+
assert result["state"] is "DELETED"
93+
mock_urlparse.assert_called()
94+
mock_delete.assert_called()
95+
8296
@patch.object(AquaModelApp, "list")
8397
def test_list(self, mock_list):
8498
with patch(

0 commit comments

Comments
 (0)