Skip to content
This repository was archived by the owner on Mar 13, 2020. It is now read-only.

Commit b67ccfc

Browse files
committed
add setup db scripts for unit test target db
1 parent 860f27f commit b67ccfc

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tests/unit_tests/test_DataLoadTrackerRepository.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,14 @@ def setUpClass(cls):
3535
conn.execute(f"CREATE DATABASE {TEST_DB};")
3636
conn.close()
3737

38+
cls.target_engine = create_engine(gen_connection_string.format(db=TEST_DB))
39+
target_db_conn = cls.target_engine.connect()
40+
target_db_conn.execute('CREATE EXTENSION IF NOT EXISTS CITEXT;')
41+
target_db_conn.execute('CREATE EXTENSION IF NOT EXISTS "uuid-ossp";')
42+
target_db_conn.close()
43+
3844
subprocess.call(f"alembic -c rdl/alembic.ini -x {gen_connection_string.format(db=TEST_DB)} upgrade head")
3945

40-
cls.target_engine = create_engine(gen_connection_string.format(db=TEST_DB))
4146
cls.data_load_tracker = DataLoadTrackerRepository(sessionmaker(bind=cls.target_engine))
4247

4348
@classmethod

0 commit comments

Comments
 (0)