From e602a731974bface5c25ff4e83e7e2bd153daa61 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Fri, 25 Jul 2014 20:19:06 -0400 Subject: - use a template database for PG so extensions get created automatically --- lib/sqlalchemy/testing/plugin/provision.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) (limited to 'lib/sqlalchemy/testing/plugin/provision.py') 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 @@ -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): -- cgit v1.2.1