- use a template database for PG so extensions get created automatically

1 files changed, 7 insertions, 4 deletions

diff --git a/lib/sqlalchemy/testing/plugin/provision.py b/lib/sqlalchemy/testing/plugin/provision.py
index d665727f1..e6790f877 100644
--- a/lib/sqlalchemy/testing/plugin/provision.py
+++ b/lib/sqlalchemy/testing/plugin/provision.py
@@ -1,5 +1,6 @@
 from sqlalchemy.engine import url as sa_url
 
+
 def create_follower_db(follower_ident):
     from .. import config, engines
 
@@ -8,6 +9,9 @@ def create_follower_db(follower_ident):
     hosts = set()
 
     for cfg in config.Config.all_configs():
+        cfg.db.dispose()
+
+    for cfg in config.Config.all_configs():
         url = cfg.db.url
         backend = url.get_backend_name()
         host_conf = (
@@ -40,13 +44,12 @@ def _pg_create_db(eng, ident):
             conn.execute("DROP DATABASE %s" % ident)
         except:
             pass
-        conn.execute("CREATE DATABASE %s" % ident)
+        currentdb = conn.scalar("select current_database()")
+        conn.execute("CREATE DATABASE %s TEMPLATE %s" % (ident, currentdb))
 
 
 def _pg_init_db(eng):
-    with eng.connect() as conn:
-        conn.execute("CREATE SCHEMA test_schema")
-        conn.execute("CREATE SCHEMA test_schema_2")
+    pass
 
 
 def _mysql_create_db(eng, ident):