b

tests/test_dead_host_mechanism.py

@@ -16,14 +16,12 @@ from main import (
     dead_host_detection,
     dead_host_mitigation,
     main,
+    config
 )
 
-from etcd3_wrapper import Etcd3Wrapper
-
-
 class TestDeadHostMechanism(unittest.TestCase):
     def setUp(self):
-        self.client = Etcd3Wrapper()
+        self.client = config.etcd_client
         self.host_prefix = "/test/host"
         self.vm_prefix = "/test/vm"