def test_get_cost_management_application_type_id_error(self): """Test to get application type id with error.""" client = SourcesHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER) with requests_mock.mock() as m: m.get( f'http://www.sources.com/api/v1.0/application_types?filter[name]=/insights/platform/cost-management', exc=requests.exceptions.RequestException) with self.assertRaises(SourcesHTTPClientError): client.get_cost_management_application_type_id()
def test_get_cost_management_application_type_id_not_found(self): """Test to get application type id with invalid prefix.""" client = SourcesHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER) with requests_mock.mock() as m: m.get( "http://www.sources.com/api/v1.0/application_types?filter[name]=/insights/platform/cost-management", status_code=404, json={"data": [{"id": self.application_type}]}, ) with self.assertRaises(SourceNotFoundError): client.get_cost_management_application_type_id()
def test_get_cost_management_application_type_id_exceptions(self): """Test to get application type id with invalid prefix.""" client = SourcesHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER) json_data = [None, [], [{"not_id": 4}]] for test in json_data: with self.subTest(test=test): with requests_mock.mock() as m: m.get( f"{MOCK_URL}/api/v1.0/{ENDPOINT_APPLICATION_TYPES}?filter[name]=/insights/platform/cost-management", # noqa: E501 json={"data": test}, ) with self.assertRaises(SourcesHTTPClientError): client.get_cost_management_application_type_id()
def test_get_cost_management_application_type_id(self): """Test to get application type id.""" client = SourcesHTTPClient(auth_header=Config.SOURCES_FAKE_HEADER) with requests_mock.mock() as m: m.get( "http://www.sources.com/api/v1.0/application_types?filter[name]=/insights/platform/cost-management", status_code=200, json={"data": [{"id": self.application_type}]}, ) response = client.get_cost_management_application_type_id() self.assertEqual(response, self.application_type)