def test_vrouter_running(self): vr1 = test_utils.create_test_virtual_router('vrouter1') vr2 = test_utils.create_test_virtual_router('vrouter2') self.scheduler.get_analytics_client = mock.MagicMock() def query_uve_side_effect(query_str): if 'NodeStatus' in query_str: return AGENTS_STATUS elif 'VrouterAgent' in query_str: return VROUTERS_MODE else: return {} self.scheduler.query_uve = query_uve_side_effect self.assertTrue(VirtualRouterSM.get('vrouter1').agent_state) self.assertTrue(VirtualRouterSM.get('vrouter2').agent_state)
def tearDown(self): self.analytics_patch.stop() VirtualRouterSM.reset() ServiceInstanceSM.reset() super(TestRandomScheduler, self).tearDown()