def test_get_authentication(self): """Test provider authentication getter.""" uuid = self.aws_test_provider_uuid expected_auth_string = self.aws_provider_resource_name accessor = ProviderDBAccessor(uuid) self.assertEqual(expected_auth_string, accessor.get_authentication()) accessor.close_session()
def test_get_authentication(self): """Test provider authentication getter.""" uuid = '6e212746-484a-40cd-bba0-09a19d132d64' expected_auth_string = 'arn:aws:iam::111111111111:role/CostManagement' accessor = ProviderDBAccessor(uuid) self.assertEqual(expected_auth_string, accessor.get_authentication()) accessor.close_session()
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