From 9490255ee80fdf933eccc7688a9526dbc051d9a3 Mon Sep 17 00:00:00 2001 From: rajdeep11 Date: Mon, 11 Dec 2023 16:05:09 +0530 Subject: [PATCH] adding test cases Issue-id: AIMLFW-65 Change-Id: Iae1540046b3c6a3ff232fe43f7eb034ba283f901 Signed-off-by: rajdeep11 --- tests/test_trainingmgr_operations.py | 29 ++++++++++++++++++++++++++++- 1 file changed, 28 insertions(+), 1 deletion(-) diff --git a/tests/test_trainingmgr_operations.py b/tests/test_trainingmgr_operations.py index 1b885dc..0951fc3 100644 --- a/tests/test_trainingmgr_operations.py +++ b/tests/test_trainingmgr_operations.py @@ -18,7 +18,7 @@ import json import requests from unittest import mock -from mock import patch +from mock import patch, MagicMock import pytest import flask from requests.models import Response @@ -188,3 +188,30 @@ class Test_delete_dme_filtered_data_job: assert "URL validation error: " in err.message except Exception: assert False + +class Test_get_model_info: + + @patch('trainingmgr.common.trainingmgr_operations.requests.get') + def test_get_model_info(self,mock_requests_get): + training_config_obj = DummyVariable() + model_name="abc" + mock_response=MagicMock(spec=Response) + mock_response.status_code=200 + mock_response.json.return_value={'message':'{"abc":"bca"}'} + mock_requests_get.return_value= mock_response + model_info=trainingmgr_operations.get_model_info(training_config_obj, model_name) + expected_model_info={'abc': 'bca'} + assert model_info==expected_model_info, "get model info failed" + + @patch('trainingmgr.common.trainingmgr_operations.requests.get') + def test_negative_get_model_info(self,mock_requests_get): + training_config_obj = DummyVariable() + model_name="abc" + mock_response=MagicMock(spec=Response) + mock_response.status_code=500 + mock_response.json.return_value={'message':'{"abc":"bca"}'} + mock_requests_get.return_value= mock_response + try: + model_info=trainingmgr_operations.get_model_info(training_config_obj, model_name) + except TMException as err: + assert "model info can't be fetched, model_name:" in err.message -- 2.16.6