darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-05 10:42:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

Round 2 of attempting to fix CI test runs

Browse Source 
Signed-off-by: Carlo Costino <carlo.costino@gsa.gov>
This commit is contained in:
Carlo Costino
2023-08-15 10:58:16 -04:00
parent 5519d17e09
commit 17e13c3b3b
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
tests/conftest.py
View File

@@ -106,23 +106,22 @@ def _notify_db(notify_api, worker_id):
ALEMBIC_CONFIG = os.path.join(BASE_DIR, 'migrations') ALEMBIC_CONFIG = os.path.join(BASE_DIR, 'migrations')
config = Config(ALEMBIC_CONFIG + '/alembic.ini') config = Config(ALEMBIC_CONFIG + '/alembic.ini')
config.set_main_option("script_location", ALEMBIC_CONFIG) config.set_main_option("script_location", ALEMBIC_CONFIG)
config.set_main_option(
"sqlalchemy.url",
current_app.config['SQLALCHEMY_DATABASE_URI']
)
with notify_api.app_context(): with notify_api.app_context() as app_context:
# Run DB migrations.
upgrade(config, 'head') upgrade(config, 'head')
# Retrieve the DB object from the initialized app. # Modify the database connection URL to point to the correct
db = notify_api.extensions['sqlalchemy'] # test database.
db = app_context.app.extensions['sqlalchemy']
db.engine.url = current_app.config['SQLALCHEMY_DATABASE_URI']
# Check the DB name. # Return the DB object to the test calling for it.
assert 'test_notification_api' in db.engine.url.database, 'dont run tests against main db' yield db
# Modify the URL to point to the correct test database.
db.engine.url = current_app.config['SQLALCHEMY_DATABASE_URI']
yield db
db.session.remove()
db.engine.dispose()
@pytest.fixture(scope='function') @pytest.fixture(scope='function')