def test_get_billing_source_for_provider_error(self): """Test to build Koku Provider billing_source json obj with errors.""" test_matrix = [ { "provider_type": Provider.PROVIDER_AWS, "billing_source": { "data_source": "test-bucket" }, "expected_response": ProviderBuilderError, }, { "provider_type": Provider.PROVIDER_AZURE, "billing_source": { "bucket": { "foo": "bar" } }, "expected_response": ProviderBuilderError, }, ] client = ProviderBuilder(auth_header=Config.SOURCES_FAKE_HEADER) for test in test_matrix: with self.assertRaises(test.get("expected_response")): client.get_billing_source_for_provider( test.get("provider_type"), test.get("billing_source"))
def test_get_billing_source_for_provider(self): """Test to build Koku Provider billing_source json obj.""" test_matrix = [ { "provider_type": Provider.PROVIDER_AWS, "billing_source": { "data_source": { "bucket": "test-bucket" } }, "expected_response": { "data_source": { "bucket": "test-bucket" } }, }, { "provider_type": Provider.PROVIDER_OCP, "billing_source": {}, "expected_response": {} }, { "provider_type": Provider.PROVIDER_AZURE, "billing_source": { "data_source": { "foo": "bar" } }, "expected_response": { "data_source": { "foo": "bar" } }, }, ] client = ProviderBuilder(auth_header=Config.SOURCES_FAKE_HEADER) for test in test_matrix: with self.subTest(test=test): response = client.get_billing_source_for_provider( test.get("provider_type"), test.get("billing_source")) self.assertEqual(response, test.get("expected_response"))