- [bug] Postgresql dialect memoizes that an ENUM of a

particular name was processed
during a create/drop sequence.  This allows
a create/drop sequence to work without any
calls to "checkfirst", and also means with
"checkfirst" turned on it only needs to
check for the ENUM once.  [ticket:2311]

1 files changed, 7 insertions, 1 deletions

diff --git a/test/lib/engines.py b/test/lib/engines.py
index 649c3e31a..c8a44dc44 100644
--- a/test/lib/engines.py
+++ b/test/lib/engines.py
@@ -247,12 +247,18 @@ def mock_engine(dialect_name=None):
     def assert_sql(stmts):
         recv = [re.sub(r'[\n\t]', '', str(s)) for s in buffer]
         assert  recv == stmts, recv
-
+    def print_sql():
+        d = engine.dialect
+        return "\n".join(
+            str(s.compile(dialect=d)) 
+            for s in engine.mock
+        )
     engine = create_engine(dialect_name + '://',
                            strategy='mock', executor=executor)
     assert not hasattr(engine, 'mock')
     engine.mock = buffer
     engine.assert_sql = assert_sql
+    engine.print_sql = print_sql
     return engine
 
 class DBAPIProxyCursor(object):