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())