diff options
Diffstat (limited to 'lib/sqlalchemy/testing/provision.py')
| -rw-r--r-- | lib/sqlalchemy/testing/provision.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/lib/sqlalchemy/testing/provision.py b/lib/sqlalchemy/testing/provision.py index 13a5ea078..21bacfca2 100644 --- a/lib/sqlalchemy/testing/provision.py +++ b/lib/sqlalchemy/testing/provision.py @@ -21,9 +21,10 @@ class register(object): def init(cls, fn): return register().for_db("*")(fn) - def for_db(self, dbname): + def for_db(self, *dbnames): def decorate(fn): - self.fns[dbname] = fn + for dbname in dbnames: + self.fns[dbname] = fn return self return decorate @@ -138,6 +139,7 @@ def _generate_driver_urls(url, extra_drivers): main_driver = url.get_driver_name() extra_drivers.discard(main_driver) + url = generate_driver_url(url, main_driver) yield str(url) for drv in list(extra_drivers): |