def get_accounts_from_source(self):
"""
Retrieve all accounts from the Koku database.
This will return a list of dicts for the Orchestrator to use to access reports.
Args:
None
Returns:
([{}]) : A list of dicts
"""
collector = ProviderCollector()
all_providers = collector.get_providers()
collector.close_session()
accounts = []
for provider in all_providers:
provider_accessor = ProviderDBAccessor(provider.uuid)
accounts.append({
'authentication':
provider_accessor.get_authentication(),
'billing_source':
provider_accessor.get_billing_source(),
'customer_name':
provider_accessor.get_customer_name(),
'provider_type':
provider_accessor.get_type(),
'schema_name':
provider_accessor.get_schema()
})
provider_accessor.close_session()
return accounts
def test_get_billing_source(self):
"""Test provider billing_source getter."""
uuid = '6e212746-484a-40cd-bba0-09a19d132d64'
expected_billing_source = 'test-bucket'
accessor = ProviderDBAccessor(uuid)
self.assertEqual(expected_billing_source,
accessor.get_billing_source())
accessor.close_session()
def test_get_billing_source(self):
"""Test provider billing_source getter."""
uuid = self.aws_test_provider_uuid
expected_billing_source = 'test-bucket'
accessor = ProviderDBAccessor(uuid)
self.assertEqual(expected_billing_source,
accessor.get_billing_source())
accessor.close_session()
