Skipped some tests if LibevConnection is not installed

bjmb · bjmb · commit 8025dc6c71df · 2017-03-31T11:26:48.000-04:00
diff --git a/tests/unit/test_cluster.py b/tests/unit/test_cluster.py
@@ -33,9 +33,6 @@
 except ImportError:
     LibevConnection = None  # noqa
 
-def setUp():
-    LibevConnection.initialize_reactor()
-
 
 class ExceptionTypeTest(unittest.TestCase):
 
@@ -131,6 +128,11 @@ def test_event_delay_timing(self, *_):
 
 
 class SessionTest(unittest.TestCase):
+    def setUp(self):
+        if LibevConnection is None:
+            raise unittest.SkipTest('libev does not appear to be installed correctly')
+        LibevConnection.initialize_reactor()
+
     # TODO: this suite could be expanded; for now just adding a test covering a PR
     @mock_session_pools
     def test_default_serial_consistency_level(self, *_):
@@ -161,6 +163,10 @@ def test_default_serial_consistency_level(self, *_):
 
 
 class ExecutionProfileTest(unittest.TestCase):
+    def setUp(self):
+        if LibevConnection is None:
+            raise unittest.SkipTest('libev does not appear to be installed correctly')
+        LibevConnection.initialize_reactor()
 
     def _verify_response_future_profile(self, rf, prof):
         self.assertEqual(rf._load_balancer, prof.load_balancing_policy)
