From e31274e3289b60c7d0bbb53243c7e4c73ceb3283 Mon Sep 17 00:00:00 2001 From: Mike Bayer Date: Sat, 20 Sep 2014 18:16:17 -0400 Subject: [PATCH] add missing util lib --- alembic/testing/util.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 alembic/testing/util.py diff --git a/alembic/testing/util.py b/alembic/testing/util.py new file mode 100644 index 00000000..466dea30 --- /dev/null +++ b/alembic/testing/util.py @@ -0,0 +1,19 @@ +from sqlalchemy.util import decorator + + +@decorator +def provide_metadata(fn, *args, **kw): + """Provide bound MetaData for a single test, dropping afterwards.""" + + from . import config + from sqlalchemy import schema + + metadata = schema.MetaData(config.db) + self = args[0] + prev_meta = getattr(self, 'metadata', None) + self.metadata = metadata + try: + return fn(*args, **kw) + finally: + metadata.drop_all() + self.metadata = prev_meta -- 2.47.2