Fix cache database for PostgreSQL

1200wd · Feb 7, 2024 · 429e2df · 429e2df
1 parent 52e64fb
commit 429e2df
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/tests/bitcoinlib_cache_unittest1 b/tests/bitcoinlib_cache_unittest1
diff --git a/tests/bitcoinlib_cache_unittest2 b/tests/bitcoinlib_cache_unittest2
diff --git a/tests/test_services.py b/tests/test_services.py
@@ -820,9 +820,13 @@ def setUpClass(cls):
             try:
                 con = psycopg.connect(user='postgres', host='localhost', password='postgres', autocommit=True)
                 cur = con.cursor()
-                cur.execute(sql.SQL("CREATE DATABASE {}").format(
-                    sql.Identifier('bitcoinlibcache.unittest'))
-                )
+                try:
+                    cur.execute(sql.SQL("DROP DATABASE IF EXISTS {}").format(sql.Identifier(CACHE_DBNAME1)))
+                    cur.execute(sql.SQL("DROP DATABASE IF EXISTS {}").format(sql.Identifier(CACHE_DBNAME2)))
+                    cur.execute(sql.SQL("CREATE DATABASE {}").format(sql.Identifier(CACHE_DBNAME1)))
+                    cur.execute(sql.SQL("CREATE DATABASE {}").format(sql.Identifier(CACHE_DBNAME2)))
+                except:
+                    pass
                 cur.close()
                 con.close()
             except Exception: