def test_adjusted_all_spam(self):
mock_bounce = patch("pad.rules.ruleset.RuleSet."
"_get_bounce_message").start()
mock_adjust = patch("pad.rules.ruleset.RuleSet."
"_adjust_headers").start()
mock_msg = MagicMock(score=6)
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.header_mod["all"].append("All mod")
ruleset.header_mod["spam"].append("Spam mod")
result = ruleset.get_adjusted_message(mock_msg)
newmsg = mock_bounce(mock_msg)
self.assertEqual(result, newmsg.as_string())
calls = [
call(mock_msg, newmsg, ["All mod"]),
call(mock_msg, newmsg, ["Spam mod"])
]
mock_adjust.assert_has_calls(calls)
def test_adjusted_all_spam(self):
mock_bounce = patch("pad.rules.ruleset.RuleSet."
"_get_bounce_message").start()
mock_adjust = patch("pad.rules.ruleset.RuleSet."
"_adjust_headers").start()
mock_msg = MagicMock(score=6)
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.header_mod["all"].append("All mod")
ruleset.header_mod["spam"].append("Spam mod")
result = ruleset.get_adjusted_message(mock_msg)
newmsg = mock_bounce(mock_msg)
self.assertEqual(result, newmsg.as_string())
calls = [
call(mock_msg, newmsg, ["All mod"]),
call(mock_msg, newmsg, ["Spam mod"])
]
mock_adjust.assert_has_calls(calls)
def test_adjusted_header_only_not_spam(self):
mock_email = patch("pad.rules.ruleset."
"email.message_from_string").start()
mock_bounce = patch("pad.rules.ruleset.RuleSet."
"_get_bounce_message").start()
mock_adjust = patch("pad.rules.ruleset.RuleSet."
"_adjust_headers").start()
mock_msg = MagicMock(score=4)
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.header_mod["all"].append("All mod")
ruleset.header_mod["ham"].append("Ham mod")
result = ruleset.get_adjusted_message(mock_msg, True)
newmsg = mock_email(mock_msg.raw_msg)
self.assertEqual(
result, newmsg.as_string().split("\n\n", 1)[0] + "\n\n")
calls = [
call(mock_msg, newmsg, ["All mod"]),
call(mock_msg, newmsg, ["Ham mod"])
]
mock_adjust.assert_has_calls(calls)
def test_adjusted_header_only_not_spam(self):
mock_email = patch("pad.rules.ruleset."
"email.message_from_string").start()
mock_bounce = patch("pad.rules.ruleset.RuleSet."
"_get_bounce_message").start()
mock_adjust = patch("pad.rules.ruleset.RuleSet."
"_adjust_headers").start()
mock_msg = MagicMock(score=4)
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset.header_mod["all"].append("All mod")
ruleset.header_mod["ham"].append("Ham mod")
result = ruleset.get_adjusted_message(mock_msg, True)
newmsg = mock_email(mock_msg.raw_msg)
self.assertEqual(result,
newmsg.as_string().split("\n\n", 1)[0] + "\n\n")
calls = [
call(mock_msg, newmsg, ["All mod"]),
call(mock_msg, newmsg, ["Ham mod"])
]
mock_adjust.assert_has_calls(calls)
