def test_get_aa_policy_id_of_server_exception_501(self, mock_clc_sdk): error = APIFailedResponse() error.response_text = 'Mock failure message' error.response_status_code = 501 mock_clc_sdk.v2.API.Call.side_effect = error self.module.check_mode = False under_test = ClcModifyServer(self.module) under_test._get_aa_policy_id_of_server(mock_clc_sdk, self.module, 'alias', 'dummy_server') self.module.fail_json.assert_called_once_with(msg='Unable to fetch anti affinity policy for server "dummy_server". Mock failure message')
def test_get_aa_policy_id_of_server_exception_404(self, mock_clc_sdk): error = APIFailedResponse() error.response_text = 'Mock failure message' error.response_status_code = 404 mock_clc_sdk.v2.API.Call.side_effect = error self.module.check_mode = False under_test = ClcModifyServer(self.module) res = under_test._get_aa_policy_id_of_server(mock_clc_sdk, self.module, 'alias', 'dummy_server') self.assertEqual(self.module.fail_json.called, False) self.assertEqual(res, None)