Ignore asyncpg driver for test database engine

encode · Sep 24, 2021 · 1bff5c9 · 1bff5c9
1 parent cba7e57
commit 1bff5c9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_databases.py b/tests/test_databases.py
@@ -87,6 +87,8 @@ def create_test_database():
             url = str(database_url.replace(driver="pymysql"))
         elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
+        elif database_url.scheme == "postgresql+asyncpg":
+            url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.create_all(engine)
 
@@ -100,6 +102,8 @@ def create_test_database():
             url = str(database_url.replace(driver="pymysql"))
         elif database_url.scheme == "postgresql+aiopg":
             url = str(database_url.replace(driver=None))
+        elif database_url.scheme == "postgresql+asyncpg":
+            url = str(database_url.replace(driver=None))
         engine = sqlalchemy.create_engine(url)
         metadata.drop_all(engine)