def test_adjust_headers_remove(self):
rules = [(True, "X-Spam-Test", None)]
mock_msg = MagicMock(interpolate_data={"TEST": "test"})
newmsg = MagicMock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset._adjust_headers(mock_msg, newmsg, rules)
newmsg.__delitem__.assert_called_with("X-Spam-Test")
def test_adjust_headers_remove(self):
rules = [(True, "X-Spam-Test", None)]
mock_msg = MagicMock(interpolate_data={"TEST": "test"})
newmsg = MagicMock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset._adjust_headers(mock_msg, newmsg, rules)
newmsg.__delitem__.assert_called_with("X-Spam-Test")
def test_adjust_headers(self):
rules = [(False, "X-Spam-Test", "value")]
mock_msg = MagicMock(interpolate_data={"TEST": "test"})
newmsg = MagicMock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset._adjust_headers(mock_msg, newmsg, rules)
newmsg.add_header.assert_called_with("X-Spam-Test", "value")
def test_adjust_headers(self):
rules = [(False, "X-Spam-Test", "value")]
mock_msg = MagicMock(interpolate_data={"TEST": "test"})
newmsg = MagicMock()
ruleset = pad.rules.ruleset.RuleSet(self.mock_ctxt)
ruleset._adjust_headers(mock_msg, newmsg, rules)
newmsg.add_header.assert_called_with("X-Spam-Test", "value")