diff options
Diffstat (limited to 'ironic/tests/unit/common/test_rpc_service.py')
-rw-r--r-- | ironic/tests/unit/common/test_rpc_service.py | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/ironic/tests/unit/common/test_rpc_service.py b/ironic/tests/unit/common/test_rpc_service.py new file mode 100644 index 000000000..b1e064730 --- /dev/null +++ b/ironic/tests/unit/common/test_rpc_service.py @@ -0,0 +1,53 @@ +# Licensed under the Apache License, Version 2.0 (the "License"); you may +# not use this file except in compliance with the License. You may obtain +# a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the +# License for the specific language governing permissions and limitations +# under the License. + +import mock +from oslo_config import cfg +import oslo_messaging +from oslo_service import service as base_service + +from ironic.common import context +from ironic.common import rpc +from ironic.common import rpc_service +from ironic.conductor import manager +from ironic.objects import base as objects_base +from ironic.tests import base + +CONF = cfg.CONF + + +@mock.patch.object(base_service.Service, '__init__', lambda *_, **__: None) +class TestRPCService(base.TestCase): + + def setUp(self): + super(TestRPCService, self).setUp() + host = "fake_host" + mgr_module = "ironic.conductor.manager" + mgr_class = "ConductorManager" + self.rpc_svc = rpc_service.RPCService(host, mgr_module, mgr_class) + + @mock.patch.object(oslo_messaging, 'Target', autospec=True) + @mock.patch.object(objects_base, 'IronicObjectSerializer', autospec=True) + @mock.patch.object(rpc, 'get_server', autospec=True) + @mock.patch.object(manager.ConductorManager, 'init_host', autospec=True) + @mock.patch.object(context, 'get_admin_context', autospec=True) + def test_start(self, mock_ctx, mock_init_method, + mock_rpc, mock_ios, mock_target): + mock_rpc.return_value.start = mock.MagicMock() + self.rpc_svc.handle_signal = mock.MagicMock() + self.rpc_svc.start() + mock_ctx.assert_called_once_with() + mock_target.assert_called_once_with(topic=self.rpc_svc.topic, + server="fake_host") + mock_ios.assert_called_once_with() + mock_init_method.assert_called_once_with(self.rpc_svc.manager, + mock_ctx.return_value) |