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 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_schema_name(self): """Test provider name getter.""" customer_id = self.customer.id accessor = CustomerDBAccessor(customer_id) self.assertEqual(self.schema, accessor.get_schema_name())