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_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)
 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)