Skip to content

Commit f61eb04

Browse files
committed
update model_repository unit tests to accomodate version checking
1 parent 0d5fd79 commit f61eb04

File tree

1 file changed

+29
-26
lines changed

1 file changed

+29
-26
lines changed

tests/unit/test_model_repository.py

Lines changed: 29 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -290,37 +290,40 @@ def test_list_model_versions():
290290
with mock.patch(
291291
"sasctl._services.model_repository.ModelRepository.request_link"
292292
) as request_link:
293+
with mock.patch(
294+
"sasctl.core.Session.version_info"
295+
) as version:
296+
version.return_value = VersionInfo(4)
297+
get_model_link.return_value = None
298+
with pytest.raises(ValueError):
299+
mr.list_model_versions(
300+
model="12345",
301+
)
293302

294-
get_model_link.return_value = None
295-
with pytest.raises(ValueError):
296-
mr.list_model_versions(
297-
model="12345",
298-
)
299-
300-
get_model_link_mock = {
301-
"method": "GET",
302-
"rel": "modelHistory",
303-
"href": "/modelRepository/models/12345/history",
304-
"uri": "/modelRepository/models/12345/history",
305-
"type": "application/vnd.sas.collection",
306-
"responseItemType": "application/vnd.sas.models.model.version",
307-
}
303+
get_model_link_mock = {
304+
"method": "GET",
305+
"rel": "modelHistory",
306+
"href": "/modelRepository/models/12345/history",
307+
"uri": "/modelRepository/models/12345/history",
308+
"type": "application/vnd.sas.collection",
309+
"responseItemType": "application/vnd.sas.models.model.version",
310+
}
308311

309-
get_model_link.return_value = get_model_link_mock
312+
get_model_link.return_value = get_model_link_mock
310313

311-
response = mr.list_model_versions(model="12345")
312-
assert response
314+
response = mr.list_model_versions(model="12345")
315+
assert response
313316

314-
request_link.return_value = RestObj({"id": "12345"})
315-
response = mr.list_model_versions(model="12345")
316-
assert isinstance(response, list)
317+
request_link.return_value = RestObj({"id": "12345"})
318+
response = mr.list_model_versions(model="12345")
319+
assert isinstance(response, list)
317320

318-
request_link.return_value = [
319-
RestObj({"id": "12345"}),
320-
RestObj({"id": "3456"}),
321-
]
322-
response = mr.list_model_versions(model="12345")
323-
assert isinstance(response, list)
321+
request_link.return_value = [
322+
RestObj({"id": "12345"}),
323+
RestObj({"id": "3456"}),
324+
]
325+
response = mr.list_model_versions(model="12345")
326+
assert isinstance(response, list)
324327

325328

326329
def test_get_model_version():

0 commit comments

Comments
 (0)