Modify HAProxy driver tests for mock OS services.

libra/worker/drivers/haproxy/driver.py

@@ -19,8 +19,11 @@ from libra.worker.drivers.haproxy.services_base import ServicesBase
 
 class HAProxyDriver(LoadBalancerDriver):
 
-    def __init__(self, ossvc='ubuntu_services.UbuntuServices'):
+    def __init__(
-        ossvc_driver = import_class('libra.worker.drivers.haproxy.' + ossvc)
+        self,
+        ossvc='libra.worker.drivers.haproxy.ubuntu_services.UbuntuServices'
+    ):
+        ossvc_driver = import_class(ossvc)
         self.ossvc = ossvc_driver()
         if not isinstance(self.ossvc, ServicesBase):
             raise Exception('Class is not derived from ServicesBase: %s' %

tests/mock_objects.py

@@ -1,6 +1,8 @@
 import json
 import logging
 
+from libra.worker.drivers.haproxy.services_base import ServicesBase
+
 
 class FakeJob(object):
     def __init__(self, data):
@@ -10,6 +12,23 @@ class FakeJob(object):
         self.data = json.dumps(data)
 
 
+class FakeOSServices(ServicesBase):
+    def service_stop(self):
+        pass
+
+    def service_start(self):
+        pass
+
+    def service_restart(self):
+        pass
+
+    def write_config(self):
+        pass
+
+    def remove_configs(self):
+        pass
+
+
 class MockLoggingHandler(logging.Handler):
     """Mock logging handler to check for expected logs."""
 
@@ -28,3 +47,5 @@ class MockLoggingHandler(logging.Handler):
             'error': [],
             'critical': [],
         }
+
+

tests/test_haproxy_driver.py

@@ -4,7 +4,7 @@ from libra.worker.drivers.haproxy.driver import HAProxyDriver
 
 class TestHAProxyDriver(unittest.TestCase):
     def setUp(self):
-        self.driver = HAProxyDriver()
+        self.driver = HAProxyDriver('mock_objects.FakeOSServices')
 
     def tearDown(self):
         pass