Merge "Add Unit Tests for Pool Manager"

designate/tests/__init__.py

@@ -38,6 +38,7 @@ from designate import exceptions
 from designate.network_api import fake as fake_network_api
 from designate import network_api
 from designate import objects
+from designate import rpc
 from designate import storage
 from designate.manage import database as manage_database
 from designate.sqlalchemy import utils as sqlalchemy_utils
@@ -71,6 +72,21 @@ class NotifierFixture(fixtures.Fixture):
         return test_notifier.reset()
 
 
+class RPCFixture(fixtures.Fixture):
+
+    def __init__(self, conf):
+        self.conf = conf
+
+    def setUp(self):
+        super(RPCFixture, self).setUp()
+        rpc.init(self.conf)
+        self.addCleanup(self.deinit)
+
+    def deinit(self):
+        if rpc.initialized():
+            rpc.cleanup()
+
+
 class ServiceFixture(fixtures.Fixture):
     def __init__(self, svc_name, *args, **kw):
         cls = importutils.import_class(
@@ -318,6 +334,8 @@ class TestCase(base.BaseTestCase):
 
         self.notifications = self.useFixture(NotifierFixture())
 
+        self.useFixture(RPCFixture(cfg.CONF))
+
         self.config(
             storage_driver='sqlalchemy',
             group='service:central'