From ad53cec4286e09f2fe7b199ef0e8ab96d28e0f92 Mon Sep 17 00:00:00 2001
From: Roland <roland@learningselfreliance.com>
Date: Fri, 7 Jan 2022 14:03:59 -0700
Subject: [PATCH] Checking database name directly.

---
 wrolpi/db.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wrolpi/db.py b/wrolpi/db.py
index 8958f677..4e8a7572 100644
--- a/wrolpi/db.py
+++ b/wrolpi/db.py
@@ -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