@@ -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
326329def test_get_model_version ():
0 commit comments