diff --git a/neutron/tests/unit/nec/test_agent_scheduler.py b/neutron/tests/unit/nec/test_agent_scheduler.py index a54f4bdae..68dd70e18 100644 --- a/neutron/tests/unit/nec/test_agent_scheduler.py +++ b/neutron/tests/unit/nec/test_agent_scheduler.py @@ -15,6 +15,9 @@ # See the License for the specific language governing permissions and # limitations under the License. +import mox + +from neutron.plugins.nec.common import ofc_client from neutron.tests.unit.nec import test_nec_plugin from neutron.tests.unit.openvswitch import test_agent_scheduler @@ -23,12 +26,30 @@ class NecAgentSchedulerTestCase( test_agent_scheduler.OvsAgentSchedulerTestCase): plugin_str = test_nec_plugin.PLUGIN_NAME + def setUp(self): + super(NecAgentSchedulerTestCase, self).setUp() + self.mox = mox.Mox() + self.mox.StubOutWithMock(ofc_client.OFCClient, 'do_request') + self.addCleanup(self.mox.UnsetStubs) + class NecDhcpAgentNotifierTestCase( test_agent_scheduler.OvsDhcpAgentNotifierTestCase): plugin_str = test_nec_plugin.PLUGIN_NAME + def setUp(self): + super(NecDhcpAgentNotifierTestCase, self).setUp() + self.mox = mox.Mox() + self.mox.StubOutWithMock(ofc_client.OFCClient, 'do_request') + self.addCleanup(self.mox.UnsetStubs) + class NecL3AgentNotifierTestCase( test_agent_scheduler.OvsL3AgentNotifierTestCase): plugin_str = test_nec_plugin.PLUGIN_NAME + + def setUp(self): + super(NecL3AgentNotifierTestCase, self).setUp() + self.mox = mox.Mox() + self.mox.StubOutWithMock(ofc_client.OFCClient, 'do_request') + self.addCleanup(self.mox.UnsetStubs)