def test_spf_redir(self):
'''
Test for SPF records which use the 'redirect' SPF mechanism
https://en.wikipedia.org/wiki/Sender_Policy_Framework#Mechanisms
'''
dig_mock = MagicMock(side_effect=_spf_side_effect)
with patch.dict(dig.__salt__, {'cmd.run_all': dig_mock}):
self.assertEqual(dig.SPF('xmission-redirect.com'),
['198.60.22.0/24', '166.70.13.0/24'])
def test_spf_include(self):
"""
Test for SPF records which use the 'include' SPF mechanism
https://en.wikipedia.org/wiki/Sender_Policy_Framework#Mechanisms
"""
dig_mock = MagicMock(side_effect=_spf_side_effect)
with patch.dict(dig.__salt__, {"cmd.run_all": dig_mock}):
self.assertEqual(
dig.SPF("xmission.com"), ["198.60.22.0/24", "166.70.13.0/24"]
)
def test_spf(self):
dig.__salt__ = {}
dig_mock = MagicMock(side_effect=_spf_side_effect)
with patch.dict(dig.__salt__, {'cmd.run_all': dig_mock}):
self.assertEqual(dig.SPF('foo.com'),
['216.73.93.70/31', '216.73.93.72/31'])
def test_spf(self):
dig_mock = MagicMock(side_effect=_spf_side_effect)
with patch.dict(dig.__salt__, {"cmd.run_all": dig_mock}):
self.assertEqual(dig.SPF("foo.com"), ["216.73.93.70/31", "216.73.93.72/31"])
