fix(deps): upgrade to sqlalchemy version 2

Bump sqlalchemy to version 2.0.22

There are quite some changes that happened with v2.
We are currently relying on the "auto-commit" feature which was removed.
For the moment we can use a wrapper class to mimic the auto-commit
behavior allowing us to move to v2.

Ultimately, the (db) session management needs some overhaul though.

Signed-off-by: moson <moson@archlinux.org>

test/conftest.py

@@ -44,7 +44,7 @@ from multiprocessing import Lock
 import py
 import pytest
 from prometheus_client import values
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, text
 from sqlalchemy.engine import URL
 from sqlalchemy.engine.base import Engine
 from sqlalchemy.exc import ProgrammingError
@@ -113,15 +113,16 @@ def _create_database(engine: Engine, dbname: str) -> None:
     :param dbname: Database name to create
     """
     conn = engine.connect()
-    try:
-        conn.execute(f"CREATE DATABASE {dbname}")
-    except ProgrammingError:  # pragma: no cover
-        # The database most likely already existed if we hit
-        # a ProgrammingError. Just drop the database and try
-        # again. If at that point things still fail, any
-        # exception will be propogated up to the caller.
-        conn.execute(f"DROP DATABASE {dbname} WITH (FORCE)")
-        conn.execute(f"CREATE DATABASE {dbname}")
+    with conn.begin():
+        try:
+            conn.execute(text(f"CREATE DATABASE {dbname}"))
+        except ProgrammingError:  # pragma: no cover
+            # The database most likely already existed if we hit
+            # a ProgrammingError. Just drop the database and try
+            # again. If at that point things still fail, any
+            # exception will be propogated up to the caller.
+            conn.execute(text(f"DROP DATABASE {dbname} WITH (FORCE)"))
+            conn.execute(text(f"CREATE DATABASE {dbname}"))
     conn.close()
     initdb.run(AlembicArgs)
 
@@ -133,8 +134,9 @@ def _drop_database(engine: Engine, dbname: str) -> None:
     :param engine: Engine returned by test_engine()
     :param dbname: Database name to drop
     """
+    aurweb.schema.metadata.drop_all(bind=engine)
     conn = engine.connect()
-    conn.execute(f"DROP DATABASE {dbname} WITH (FORCE)")
+    conn.execute(text(f"DROP DATABASE {dbname}"))
     conn.close()