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

diff --git a/lib/sqlalchemy/testing/plugin/provision.py b/lib/sqlalchemy/testing/plugin/provision.py
index d665727f1632f3e24d4ce1cb009eccc1476e5c66..e6790f877fd9c70f5f934e33568d9a1b509d39db 100644 (file)
--- 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
 
@@ -7,6 +8,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()
@@ -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):