def get_postgres_connection(
cls): # TODO externalize use - test schema to initialize
db_url = "postgres:host='localhost' dbname='sales_reporting' user='******' password='******'" # TODO externalize
cls.connection = ConnectionHelper.get_connection(db_url)
cursor = cls.connection.cursor()
schema = "integration_test"
cursor.execute("create schema %s" % schema)
cursor.execute("set schema '%s'" % schema)
return cls.connection
def setUp(self):
# db_url = os.getenv(self.POSTGRES_TEST_URL)
if self.dialect == dialects.DIALECT_POSTGRES:
db_url = "postgres:host='localhost' dbname='sales_reporting_db' user='******' password='******'"
# TODO externalize
test_schema = self.TEST_SCHEMA
elif self.dialect == dialects.DIALECT_SQLITE:
db_url = "sqlite::memory:"
test_schema = None
else:
raise Exception("unsupported dialect %s" % self.dialect)
message = "Skipping test as %s environment variable not set" % self.POSTGRES_TEST_URL
if not db_url:
logging.warning(message)
self.skipTest(message)
self.connection = ConnectionHelper.get_connection(db_url)
SchemaInitter(self.connection, test_schema).process()
ut_condition_init(self.connection).process()
sr_data_init(self.connection).process()
def get_test_postgres_connection():
db_url = "postgres:host='localhost' dbname='sales_reporting_db' user='******' password='******'" # TODO externalize
return ConnectionHelper.get_connection(db_url)