def test_schema_name_present_on_customer(self):
"""Test that schema_name is returned on customer."""
iam_arn = 'arn:aws:s3:::my_s3_bucket'
bucket_name = 'my_s3_bucket'
provider = {
'name': 'test_provider',
'type': Provider.PROVIDER_AWS,
'authentication': {
'provider_resource_name': iam_arn
},
'billing_source': {
'bucket': bucket_name
}
}
with patch.object(ProviderAccessor,
'cost_usage_source_ready',
returns=True):
serializer = AdminProviderSerializer(data=provider,
context=self.request_context)
if serializer.is_valid(raise_exception=True):
serializer.save()
account = self.customer.account_id
expected_schema_name = create_schema_name(account)
schema_name = serializer.data['customer'].get('schema_name')
self.assertIsNotNone(schema_name)
self.assertEqual(schema_name, expected_schema_name)
def test_schema_name_present_on_customer(self):
"""Test that schema_name is returned on customer."""
iam_arn = "arn:aws:s3:::my_s3_bucket"
bucket_name = "my_s3_bucket"
provider = {
"name": "test_provider",
"type": Provider.PROVIDER_AWS.lower(),
"authentication": {
"credentials": {
"role_arn": iam_arn
}
},
"billing_source": {
"data_source": {
"bucket": bucket_name
}
},
}
with patch.object(ProviderAccessor,
"cost_usage_source_ready",
returns=True):
serializer = AdminProviderSerializer(data=provider,
context=self.request_context)
if serializer.is_valid(raise_exception=True):
serializer.save()
account = self.customer.account_id
expected_schema_name = create_schema_name(account)
schema_name = serializer.data["customer"].get("schema_name")
self.assertIsNotNone(schema_name)
self.assertEqual(schema_name, expected_schema_name)