From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Fri, 4 Jul 2014 20:49:11 +0000 (-0400)
Subject: - add test support for disconnect modification
X-Git-Tag: rel_1_0_0b1~357
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=9b03d6a0a87ab098538e058cc29110b22874eaa9;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

- add test support for disconnect modification
---

diff --git a/test/engine/test_execute.py b/test/engine/test_execute.py
index d511dce532..33e116cdfe 100644
--- a/test/engine/test_execute.py
+++ b/test/engine/test_execute.py
@@ -1659,6 +1659,31 @@ class HandleErrorTest(fixtures.TestBase):
         is_(ctx.is_disconnect, False)
         is_(ctx.original_exception, nope)
 
+    def _test_alter_disconnect(self, orig_error, evt_value):
+        engine = engines.testing_engine()
+
+        @event.listens_for(engine, "handle_error")
+        def evt(ctx):
+            ctx.is_disconnect = evt_value
+
+        with patch.object(engine.dialect, "is_disconnect",
+                Mock(return_value=orig_error)):
+
+            with engine.connect() as c:
+                try:
+                    c.execute("SELECT x FROM nonexistent")
+                    assert False
+                except tsa.exc.StatementError as st:
+                    eq_(st.connection_invalidated, evt_value)
+
+    def test_alter_disconnect_to_true(self):
+        self._test_alter_disconnect(False, True)
+        self._test_alter_disconnect(True, True)
+
+    def test_alter_disconnect_to_false(self):
+        self._test_alter_disconnect(True, False)
+        self._test_alter_disconnect(False, False)
+
 class ProxyConnectionTest(fixtures.TestBase):
     """These are the same tests as EngineEventsTest, except using
     the deprecated ConnectionProxy interface.