use klass, not AsyncoreConnection

mambocab · mambocab · commit 14b810836fe6 · 2018-06-14T17:58:16.000-04:00
diff --git a/tests/integration/standard/test_connection.py b/tests/integration/standard/test_connection.py
@@ -191,7 +191,7 @@ def wait_for_no_connections(self, host, cluster):
 
 class ConnectionTests(object):
 
-    klass = None
+    klass, global_loop = None, False
 
     def setUp(self):
         self.klass.initialize_reactor()
@@ -394,10 +394,13 @@ def test_connect_timeout(self):
         self.assertTrue(exception_thrown)
 
     def test_subclasses_share_loop(self):
-        class C1(AsyncoreConnection):
+        if not self.global_loop:
+            raise unittest.case.SkipTest
+
+        class C1(self.klass):
             pass
 
-        class C2(AsyncoreConnection):
+        class C2(self.klass):
             pass
 
         clusterC1 = Cluster(connection_class=C1)
@@ -412,16 +415,16 @@ class C2(AsyncoreConnection):
 
 
 def get_eventloop_threads(name):
-    import threading
     event_loops_threads = [thread for thread in threading.enumerate() if name == thread.name]
 
     return event_loops_threads
 
 
 class AsyncoreConnectionTests(ConnectionTests, unittest.TestCase):
 
-    klass = AsyncoreConnection
-    event_loop_name = "cassandra_driver_event_loop"
+    klass, global_loop, event_loop_name = (
+        AsyncoreConnection, True, "cassandra_driver_event_loop"
+    )
 
     def setUp(self):
         if is_monkey_patched():
@@ -435,8 +438,9 @@ def clean_global_loop(self):
 
 class LibevConnectionTests(ConnectionTests, unittest.TestCase):
 
-    klass = LibevConnection
-    event_loop_name = "event_loop"
+    klass, global_loop, event_loop_name = (
+        LibevConnection, True, "event_loop"
+    )
 
     def setUp(self):
         if is_monkey_patched():
