Enhance unittest for hyperv_neutron_agent

Change-Id: Idc726ba668eba56a64dc498b49ec0e5e397590b5

diff --git a/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py b/neutron/tests/unit/hyperv/test_hyperv_neutron_agent.py
index 2f598e8969c44571041e3704126f013720c4cd43..52e6b4c8d49482633ffd4ad78aca00f788820b08 100644 (file)
--- 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()])