From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Mon, 24 Mar 2014 23:30:44 +0000 (-0400)
Subject: add some more mock structure so tricky DBAPIs like pypy work
X-Git-Tag: rel_0_9_4~27
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=74f7509d663641547054146a986a9634f9198857;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

add some more mock structure so tricky DBAPIs like pypy work
---

diff --git a/test/engine/test_execute.py b/test/engine/test_execute.py
index 73e7080855..62dea83b0f 100644
--- a/test/engine/test_execute.py
+++ b/test/engine/test_execute.py
@@ -1611,6 +1611,14 @@ class DialectEventTest(fixtures.TestBase):
         e.dialect.do_executemany = m1.real_do_executemany
         e.dialect.do_execute_no_params = m1.real_do_execute_no_params
 
+        def mock_the_cursor(cursor, *arg):
+            arg[-1].get_result_proxy = Mock(return_value=Mock(context=arg[-1]))
+            return retval
+
+        m1.real_do_execute.side_effect = m1.do_execute.side_effect = mock_the_cursor
+        m1.real_do_executemany.side_effect = m1.do_executemany.side_effect = mock_the_cursor
+        m1.real_do_execute_no_params.side_effect = m1.do_execute_no_params.side_effect = mock_the_cursor
+
         with e.connect() as conn:
             yield conn, m1