external/aurweb
0
0
Fork 0
mirror of https://gitlab.archlinux.org/archlinux/aurweb.git synced 2025-02-03 10:43:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

aurweb.testing.setup_test_db: Expunge objects

Browse source 
This is needed to avoid redundant objects in SQLAlchemy's
IdentityMap, since we pass a direct .execute to delete
the tables passed in. Additionally, remove our engine.connect()
call in favor of relying on the already-established Session.

Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2021-05-31 22:41:06 -07:00
parent b692b11f62
commit 794868b20f
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
aurweb/testing/__init__.py
View file

@ -21,10 +21,12 @@ def setup_test_db(*args):
test_tables = ["Users", "Sessions"];
setup_test_db(*test_tables)
"""
engine = aurweb.db.get_engine()
conn = engine.connect()
# Make sure that we've grabbed the engine before using the session.
aurweb.db.get_engine()
tables = list(args)
for table in tables:
conn.execute(f"DELETE FROM {table}")
conn.close()
aurweb.db.session.execute(f"DELETE FROM {table}")
# Expunge all objects from SQLAlchemy's IdentityMap.
aurweb.db.session.expunge_all()