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)