def test_reciver_policy_noDmarc(self):
'Test that we look up the public suffix for the host for "no dmarc"'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.noDmarc)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('my.example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(2, cc)
self.assertEqual(1, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.noDmarc)
def test_receiver_policy_none(self):
'Test that we only look up the reciever policy once if it is "none"'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.none)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(1, cc)
self.assertEqual(0, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.none)
def test_reciver_policy_noDmarc(self):
'Test that we look up the public suffix for the host for "no dmarc"'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.noDmarc)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('my.example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(2, cc)
self.assertEqual(1, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.noDmarc)
def test_receiver_policy_none(self):
'Test that we only look up the reciever policy once if it is "none"'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.none)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(1, cc)
self.assertEqual(0, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.none)
def test_dmarc_host(self):
'Test a host lookup when "_dmarc." something is given'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.none)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('_dmarc.example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(1, cc)
self.assertEqual(0, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.none)
args, kw_args = gs.dmarc.lookup.lookup_receiver_policy.call_args
self.assertEqual('example.com', args[0])
def test_dmarc_host(self):
'Test a host lookup when "_dmarc." something is given'
gs.dmarc.lookup.lookup_receiver_policy = \
MagicMock(return_value=ReceiverPolicy.none)
psl.get_public_suffix = MagicMock(return_value='example.com')
r = receiver_policy('_dmarc.example.com')
cc = gs.dmarc.lookup.lookup_receiver_policy.call_count
self.assertEqual(1, cc)
self.assertEqual(0, psl.get_public_suffix.call_count)
self.assertEqual(r, ReceiverPolicy.none)
args, kw_args = gs.dmarc.lookup.lookup_receiver_policy.call_args
self.assertEqual('example.com', args[0])
