rt/wrolpi
1
0
Fork 0
mirror of https://github.com/lrnselfreliance/wrolpi.git synced 2026-10-26 19:01:39 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Checking database name directly.

Browse source
This commit is contained in:
Roland 2022-01-07 14:03:59 -07:00
parent b9606c1b31
commit ad53cec428
No known key found for this signature in database
GPG key ID: 9D6DBD3ACF11AB93
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
wrolpi/db.py
View file

@ -72,7 +72,7 @@ def get_db_context() -> Tuple[create_engine, Session]:
Get a DB engine and session.
"""
local_engine, session = _get_db_session()
if PYTEST and 'wrolpi_testing' not in str(local_engine.url):
if PYTEST and not local_engine.url.database.startswith('wrolpi_testing_'):
raise ValueError(f'Running tests, but a test database is not being used!! {local_engine.url=}')
return local_engine, session