From: Mike Bayer <mike_mp@zzzcomputing.com>
Date: Sat, 29 Jun 2013 02:29:45 +0000 (-0400)
Subject: improve the close handling here so that we don't double-close a connection;
X-Git-Tag: rel_0_9_0b1~220
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=83f3dbc83d1066216084a01b32cddcc090f697d5;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

improve the close handling here so that we don't double-close a connection;
we log this now so it apparently happens a bunch
---

diff --git a/lib/sqlalchemy/testing/engines.py b/lib/sqlalchemy/testing/engines.py
index d8c1709e74..29c8b6a032 100644
--- a/lib/sqlalchemy/testing/engines.py
+++ b/lib/sqlalchemy/testing/engines.py
@@ -21,7 +21,7 @@ class ConnectionKiller(object):
         self.testing_engines[engine] = True
 
     def connect(self, dbapi_conn, con_record):
-        self.conns.add(dbapi_conn)
+        self.conns.add((dbapi_conn, con_record))
 
     def checkout(self, dbapi_con, con_record, con_proxy):
         self.proxy_refs[con_proxy] = True
@@ -52,7 +52,7 @@ class ConnectionKiller(object):
         # is collecting in finalize_fairy, deadlock.
         # not sure if this should be if pypy/jython only.
         # note that firebird/fdb definitely needs this though
-        for conn in self.conns:
+        for conn, rec in self.conns:
             self._safe(conn.rollback)
 
     def _stop_test_ctx(self):
@@ -72,8 +72,10 @@ class ConnectionKiller(object):
 
     def _stop_test_ctx_aggressive(self):
         self.close_all()
-        for conn in self.conns:
+        for conn, rec in self.conns:
             self._safe(conn.close)
+            rec.connection = None
+            
         self.conns = set()
         for rec in list(self.testing_engines):
             rec.dispose()