def test_do_audit_template_list_filter_by_goal_name(self): goal1 = resource.Goal(mock.Mock(), GOAL_1) strategy1 = resource.Strategy(mock.Mock(), STRATEGY_1) audit_template1 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) audit_template2 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_2) self.m_goal_mgr.get.return_value = goal1 self.m_strategy_mgr.get.return_value = strategy1 self.m_audit_template_mgr.list.return_value = [ audit_template1, audit_template2 ] exit_code, results = self.run_cmd( 'audittemplate list --goal SERVER_CONSOLIDATION') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template1, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS), self.resource_as_dict(audit_template2, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with( detail=False, goal='SERVER_CONSOLIDATION', )
def test_do_audit_template_list_detail(self): audit_template1 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) audit_template2 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_2) self.m_audit_template_mgr.list.return_value = [ audit_template1, audit_template2 ] exit_code, results = self.run_cmd('audittemplate list --detail') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template1, self.FIELDS, self.FIELD_LABELS), self.resource_as_dict(audit_template2, self.FIELDS, self.FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with(detail=True)
def test_do_audit_template_show_by_name(self): audit_template = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.get.return_value = audit_template exit_code, result = self.run_cmd('audittemplate show at1') self.assertEqual(0, exit_code) self.assertEqual( self.resource_as_dict(audit_template, self.FIELDS, self.FIELD_LABELS), result) self.m_audit_template_mgr.get.assert_called_once_with('at1')
def test_do_audit_template_create_with_aggregate(self): audit_template = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.create.return_value = audit_template exit_code, result = self.run_cmd( 'audittemplate create at1 fc087747-61be-4aad-8126-b701731ae836') self.assertEqual(0, exit_code) self.assertEqual( self.resource_as_dict(audit_template, self.FIELDS, self.FIELD_LABELS), result) self.m_audit_template_mgr.create.assert_called_once_with( goal='fc087747-61be-4aad-8126-b701731ae836', name='at1')
def test_do_audit_template_show_by_uuid(self): audit_template = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.get.return_value = audit_template exit_code, result = self.run_cmd( 'audittemplate show f8e47706-efcf-49a4-a5c4-af604eb492f2') self.assertEqual(0, exit_code) self.assertEqual( self.resource_as_dict(audit_template, self.FIELDS, self.FIELD_LABELS), result) self.m_audit_template_mgr.get.assert_called_once_with( 'f8e47706-efcf-49a4-a5c4-af604eb492f2')
def test_do_audit_template_list_filter_by_goal_uuid(self): audit_template1 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) audit_template2 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_2) self.m_audit_template_mgr.list.return_value = [ audit_template1, audit_template2 ] exit_code, results = self.run_cmd( 'audittemplate list --goal ' 'fc087747-61be-4aad-8126-b701731ae836') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template1, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS), self.resource_as_dict(audit_template2, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with( detail=False, goal='fc087747-61be-4aad-8126-b701731ae836', )
def test_do_audit_template_list_marker(self): audit_template2 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_2) self.m_audit_template_mgr.list.return_value = [audit_template2] exit_code, results = self.run_cmd( 'audittemplate list --marker ' 'f8e47706-efcf-49a4-a5c4-af604eb492f2') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template2, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with( detail=False, marker='f8e47706-efcf-49a4-a5c4-af604eb492f2')
def test_do_audit_create_with_audit_template_name(self): audit = resource.Audit(mock.Mock(), self.AUDIT_3) audit_template = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.get.return_value = audit_template self.m_audit_mgr.create.return_value = audit exit_code, result = self.run_cmd('audit create -a at1') self.assertEqual(0, exit_code) self.assertEqual( self.resource_as_dict(audit, self.FIELDS, self.FIELD_LABELS), result) self.m_audit_mgr.create.assert_called_once_with( audit_template_uuid='f8e47706-efcf-49a4-a5c4-af604eb492f2', auto_trigger=False, audit_type='ONESHOT')
def test_do_audit_template_update(self): audit_template = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.update.return_value = audit_template exit_code, result = self.run_cmd( 'audittemplate update at1 replace description="New description"') self.assertEqual(0, exit_code) self.assertEqual( self.resource_as_dict(audit_template, self.FIELDS, self.FIELD_LABELS), result) self.m_audit_template_mgr.update.assert_called_once_with( 'at1', [{ 'op': 'replace', 'path': '/description', 'value': 'New description' }])
def test_do_audit_template_list_filter_by_strategy_name(self): audit_template1 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_audit_template_mgr.list.return_value = [audit_template1] exit_code, results = self.run_cmd('audittemplate list --strategy ' 'basic') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template1, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with( detail=False, strategy='basic', )
def test_do_audit_template_list_filter_by_strategy_uuid(self): goal1 = resource.Goal(mock.Mock(), GOAL_1) strategy1 = resource.Strategy(mock.Mock(), STRATEGY_1) audit_template1 = resource.AuditTemplate(mock.Mock(), AUDIT_TEMPLATE_1) self.m_goal_mgr.get.return_value = goal1 self.m_strategy_mgr.get.return_value = strategy1 self.m_audit_template_mgr.list.return_value = [audit_template1] exit_code, results = self.run_cmd( 'audittemplate list --strategy ' '2cf86250-d309-4b81-818e-1537f3dba6e5') self.assertEqual(0, exit_code) self.assertEqual([ self.resource_as_dict(audit_template1, self.SHORT_LIST_FIELDS, self.SHORT_LIST_FIELD_LABELS) ], results) self.m_audit_template_mgr.list.assert_called_once_with( detail=False, strategy='2cf86250-d309-4b81-818e-1537f3dba6e5', )