From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 20 Sep 2014 22:16:17 +0000 (-0400)
Subject: add missing util lib
X-Git-Tag: rel_0_7_0~68
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=e31274e3289b60c7d0bbb53243c7e4c73ceb3283;p=thirdparty%2Fsqlalchemy%2Falembic.git

add missing util lib
---

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