Fix buildkite env

conbench · May 14, 2021 · d557b0d · d557b0d
1 parent 71cda0a
commit d557b0d
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/conbench/tests/conftest.py b/conbench/tests/conftest.py
@@ -11,11 +11,10 @@
 def create_db():
     configure_engine(TestConfig.SQLALCHEMY_DATABASE_URI)
 
-    try:
-        subprocess.run(["dropdb", TestConfig.DB_NAME])
-        subprocess.run(["createdb", TestConfig.DB_NAME])
-    except:
-        pass
+    command = ["dropdb", TestConfig.DB_NAME, "-U", "postgres", "--if-exists"]
+    subprocess.run(command, capture_output=True)
+    command = ["createdb", TestConfig.DB_NAME, "-U", "postgres"]
+    subprocess.run(command, capture_output=True)
 
     drop_all()
     create_all()