def test_add_rule_should_not_check(self):
mock_rule = Mock(**{"should_check.return_value": False})
name_mock = PropertyMock(return_value="TEST_RULE")
type(mock_rule).name = name_mock
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.add_rule(mock_rule)
self.assertEqual(ruleset.checked, {})
self.assertEqual(ruleset.not_checked, {"TEST_RULE": mock_rule})
def test_add_rule_should_not_check(self):
mock_rule = Mock(**{"should_check.return_value": False})
name_mock = PropertyMock(return_value="TEST_RULE")
type(mock_rule).name = name_mock
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.add_rule(mock_rule)
self.assertEqual(ruleset.checked, {})
self.assertEqual(ruleset.not_checked, {"TEST_RULE": mock_rule})
def test_add_rule_postprocess(self):
mock_rule = Mock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.add_rule(mock_rule)
mock_rule.postprocess.assert_called_with(ruleset)
def test_add_rule_postprocess(self):
mock_rule = Mock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.add_rule(mock_rule)
mock_rule.postprocess.assert_called_with(ruleset)
