From db3b1c01b51f9ebd00a9a63403b1ad16123c2579 Mon Sep 17 00:00:00 2001 From: Hui HX Xiang Date: Mon, 19 Aug 2013 08:13:41 -0700 Subject: [PATCH] Enhance unittest for hyperv_neutron_agent Change-Id: Idc726ba668eba56a64dc498b49ec0e5e397590b5 --- .../unit/hyperv/test_hyperv_neutron_agent.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py b/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py index 2f598e896..52e6b4c8d 100644 --- a/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py +++ b/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py @@ -108,3 +108,19 @@ class TestHyperVNeutronAgent(base.BaseTestCase): def test_treat_devices_removed_ignores_missing_port(self): self.mock_treat_devices_removed(False) + + def test_main(self): + with mock.patch.object(hyperv_neutron_agent, + 'HyperVNeutronAgent') as plugin: + with mock.patch.object(hyperv_neutron_agent, 'cfg') as cfg: + with mock.patch('eventlet.monkey_patch') as eventlet: + with mock.patch.object( + hyperv_neutron_agent, + 'logging_config') as logging_config: + + hyperv_neutron_agent.main() + + self.assertTrue(cfg.called) + self.assertTrue(eventlet.called) + self.assertTrue(logging_config.setup_logging.called) + plugin.assert_has_calls([mock.call().daemon_loop()]) -- 2.45.2