Made zconn_at_test compatible with ZODB3

lib/tests/zconn_at_test.py

@@ -109,10 +109,18 @@ class Zconn_atTest(unittest.TestCase):
             transaction.begin()
             connection = self.db.open()
             T1_is_connected.send(CONNECTED)
-            tids[THREAD1] = u64(zconn_at(connection))
-            T2_is_connected.recv()
-            transaction.abort()            
-            connection.close()
+            try:
+                tids[THREAD1] = u64(zconn_at(connection))
+                T2_is_connected.recv()
+                transaction.abort()            
+                connection.close()
+            except AssertionError as error:
+                T2_is_connected.recv()
+                transaction.abort()            
+                connection.close()
+                raise error
+            #transaction.abort()            
+            #connection.close()
         
         def T2(ctx, self):
             T1_is_connected.recv()
@@ -141,14 +149,21 @@ class Zconn_atTest(unittest.TestCase):
         def T1(ctx, self):
             transaction.begin()
             connection = self.db.open()
-            tids[START] = u64(zconn_at(connection))
-            T1_is_connected.send(CONNECTED)
+            try:
+                tids[START] = u64(zconn_at(connection))
+                T1_is_connected.send(CONNECTED)
             
-            T2_is_done.recv()
-            tids[END] = u64(zconn_at(connection))
+                T2_is_done.recv()
+                tids[END] = u64(zconn_at(connection))
 
-            transaction.abort()            
-            connection.close()
+                transaction.abort()            
+                connection.close()
+            except AssertionError as error:
+                T1_is_connected.send(CONNECTED)
+                T2_is_done.recv()
+                transaction.abort()            
+                connection.close()
+                raise error
         def T2(ctx, self):
             T1_is_connected.recv()