From: Hui HX Xiang <xianghui@cn.ibm.com>
Date: Mon, 19 Aug 2013 15:13:41 +0000 (-0700)
Subject: Enhance unittest for hyperv_neutron_agent
X-Git-Url: https://review.fuel-infra.org/gitweb?a=commitdiff_plain;h=db3b1c01b51f9ebd00a9a63403b1ad16123c2579;p=openstack-build%2Fneutron-build.git

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 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()])