i

tlocke · Mar 31, 2024 · 0c5ad61 · 0c5ad61
1 parent 8dc9fd1
commit 0c5ad61
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/native/auth/test_scram-sha-256.py b/test/native/auth/test_scram-sha-256.py
@@ -1,6 +1,6 @@
 import pytest
 
-from pg8000.native import Connection, InterfaceError
+from pg8000.native import Connection, DatabaseError, InterfaceError
 
 # This requires a line in pg_hba.conf that requires scram-sha-256 for the
 # database scram_sha_256
@@ -10,6 +10,10 @@
 
 @pytest.fixture
 def setup(con):
+    try:
+        con.run("CREATE DATABASE {DB}")
+    except DatabaseError:
+        pass
     con.run("ALTER SYSTEM SET ssl = off")
     con.run("SELECT pg_reload_conf()")
     yield