def test_initializer(self):
"""Test Initializer"""
customer_id = '1'
accessor = CustomerDBAccessor(customer_id)
self.assertIsNotNone(accessor._session)
self.assertTrue(accessor.does_db_entry_exist())
accessor.close_session()
def test_get_customer_uuid(self):
"""Test provider billing_source getter."""
uuid = '6e212746-484a-40cd-bba0-09a19d132d64'
accessor = ProviderDBAccessor(uuid)
customer = CustomerDBAccessor(1)
expected_uuid = customer.get_uuid()
self.assertEqual(expected_uuid, accessor.get_customer_uuid())
accessor.close_session()
customer.close_session()
def test_get_customer_uuid(self):
"""Test provider billing_source getter."""
uuid = self.aws_test_provider_uuid
accessor = ProviderDBAccessor(uuid)
customer = CustomerDBAccessor(1)
expected_uuid = customer.get_uuid()
self.assertEqual(expected_uuid, accessor.get_customer_uuid())
accessor.close_session()
customer.close_session()
def get_schema(self):
"""
Return the schema for the customer.
Args:
None
Returns:
(String): "Name of the database schema",
"""
obj = self._get_db_obj_query().first()
customer_id = obj.customer_id
customer_accessor = CustomerDBAccessor(customer_id)
schema_name = customer_accessor.get_schema_name()
return schema_name
def get_customer_uuid(self):
"""
Return the provider's customer uuid.
Args:
None
Returns:
(String): "UUID v4",
example: "edf94475-235e-4b64-ba18-0b81f2de9c9e"
"""
obj = self._get_db_obj_query().first()
customer_id = obj.customer_id
customer_accessor = CustomerDBAccessor(customer_id)
return customer_accessor.get_uuid()
def get_customer_name(self):
"""
Return the provider's customer name.
Args:
None
Returns:
(String): "Name of the customer",
example: "Customer 1 Inc."
"""
obj = self._get_db_obj_query().first()
customer_id = obj.customer_id
customer_accessor = CustomerDBAccessor(customer_id)
group_ptr_id = customer_accessor.get_group_ptr_id()
auth_accessor = AuthDBAccessor(group_ptr_id)
return auth_accessor.get_name()
def test_get_customer_uuid(self):
"""Test provider billing_source getter."""
expected_uuid = None
with CustomerDBAccessor(self.customer.id) as customer_accessor:
expected_uuid = customer_accessor.get_uuid()
uuid = self.aws_provider_uuid
with ProviderDBAccessor(uuid) as accessor:
self.assertEqual(expected_uuid, accessor.get_customer_uuid())
def test_get_schema_name(self):
"""Test provider name getter."""
customer_id = '1'
accessor = CustomerDBAccessor(customer_id)
self.assertEqual('acct10001', accessor.get_schema_name())
accessor.close_session()
def test_get_uuid(self):
"""Test uuid getter."""
customer_id = '1'
accessor = CustomerDBAccessor(customer_id)
self.assertIsNotNone(accessor.get_uuid())
accessor.close_session()
def test_get_group_ptr_id(self):
"""Test provider name getter."""
customer_id = '1'
accessor = CustomerDBAccessor(customer_id)
self.assertEqual(1, accessor.get_group_ptr_id())
accessor.close_session()
def test_get_schema_name(self):
"""Test provider name getter."""
customer_id = self.customer.id
accessor = CustomerDBAccessor(customer_id)
self.assertEqual(self.schema, accessor.get_schema_name())
def test_get_uuid(self):
"""Test uuid getter."""
customer_id = self.customer.id
accessor = CustomerDBAccessor(customer_id)
self.assertIsNotNone(accessor.get_uuid())
def test_initializer(self):
"""Test Initializer."""
customer_id = self.customer.id
accessor = CustomerDBAccessor(customer_id)
self.assertTrue(accessor.does_db_entry_exist())
