def setUp(self): super(self.__class__, self).setUp() # goal mock p_goal_manager = mock.patch.object(resource, 'GoalManager') self.m_goal_mgr_cls = p_goal_manager.start() self.addCleanup(p_goal_manager.stop) self.m_goal_mgr = mock.Mock() self.m_goal_mgr_cls.return_value = self.m_goal_mgr # strategy mock p_strategy_manager = mock.patch.object(resource, 'StrategyManager') self.m_strategy_mgr_cls = p_strategy_manager.start() self.addCleanup(p_strategy_manager.stop) self.m_strategy_mgr = mock.Mock() self.m_strategy_mgr_cls.return_value = self.m_strategy_mgr # audit template mock p_audit_template_manager = mock.patch.object(resource, 'AuditTemplateManager') self.m_audit_template_mgr_cls = p_audit_template_manager.start() self.addCleanup(p_audit_template_manager.stop) self.m_audit_template_mgr = mock.Mock() self.m_audit_template_mgr_cls.return_value = self.m_audit_template_mgr # stdout mock self.stdout = six.StringIO() self.cmd = shell.WatcherShell(stdout=self.stdout)
def setUp(self): super(self.__class__, self).setUp() p_audit_manager = mock.patch.object(resource, 'AuditManager') p_audit_template_manager = mock.patch.object( resource, 'ActionPlanManager') p_action_plan_manager = mock.patch.object( resource, 'ActionPlanManager') self.m_audit_mgr_cls = p_audit_manager.start() self.m_audit_template_mgr_cls = p_audit_template_manager.start() self.m_action_plan_mgr_cls = p_action_plan_manager.start() self.addCleanup(p_audit_manager.stop) self.addCleanup(p_audit_template_manager.stop) self.addCleanup(p_action_plan_manager.stop) self.m_audit_mgr = mock.Mock() self.m_audit_template_mgr = mock.Mock() self.m_action_plan_mgr = mock.Mock() self.m_audit_mgr_cls.return_value = self.m_audit_mgr self.m_audit_template_mgr_cls.return_value = self.m_audit_template_mgr self.m_action_plan_mgr_cls.return_value = self.m_action_plan_mgr self.stdout = six.StringIO() self.cmd = shell.WatcherShell(stdout=self.stdout)
def setUp(self): super(self.__class__, self).setUp() p_goal_manager = mock.patch.object(resource, 'GoalManager') self.m_goal_mgr_cls = p_goal_manager.start() self.addCleanup(p_goal_manager.stop) self.m_goal_mgr = mock.Mock() self.m_goal_mgr_cls.return_value = self.m_goal_mgr self.stdout = six.StringIO() self.cmd = shell.WatcherShell(stdout=self.stdout)
def setUp(self, os_infra_optim_api_version='1.0'): super( AuditShellTest, self).setUp(os_infra_optim_api_version=os_infra_optim_api_version) # goal mock p_goal_manager = mock.patch.object(resource, 'GoalManager') self.m_goal_mgr_cls = p_goal_manager.start() self.addCleanup(p_goal_manager.stop) self.m_goal_mgr = mock.Mock() self.m_goal_mgr_cls.return_value = self.m_goal_mgr # strategy mock p_strategy_manager = mock.patch.object(resource, 'StrategyManager') self.m_strategy_mgr_cls = p_strategy_manager.start() self.addCleanup(p_strategy_manager.stop) self.m_strategy_mgr = mock.Mock() self.m_strategy_mgr_cls.return_value = self.m_strategy_mgr p_audit_manager = mock.patch.object(resource, 'AuditManager') p_audit_template_manager = mock.patch.object(resource, 'AuditTemplateManager') self.m_audit_mgr_cls = p_audit_manager.start() self.m_audit_template_mgr_cls = p_audit_template_manager.start() self.addCleanup(p_audit_manager.stop) self.addCleanup(p_audit_template_manager.stop) self.m_audit_mgr = mock.Mock() self.m_audit_template_mgr = mock.Mock() self.m_audit_mgr_cls.return_value = self.m_audit_mgr self.m_audit_template_mgr_cls.return_value = self.m_audit_template_mgr # stdout mock self.stdout = six.StringIO() self.cmd = shell.WatcherShell(stdout=self.stdout)