diff options
Diffstat (limited to 'src/mailman/rules/tests/test_dmarc.py')
| -rw-r--r-- | src/mailman/rules/tests/test_dmarc.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/mailman/rules/tests/test_dmarc.py b/src/mailman/rules/tests/test_dmarc.py index 9af3e9347..640dd5cda 100644 --- a/src/mailman/rules/tests/test_dmarc.py +++ b/src/mailman/rules/tests/test_dmarc.py @@ -28,12 +28,12 @@ from mailman.rules import dmarc from mailman.testing.helpers import ( LogFileMark, specialized_message_from_string as mfs) from mailman.testing.layers import ConfigLayer +from mailman.utilities.protocols import get as _get from pkg_resources import resource_filename from public import public from unittest import TestCase from unittest.mock import patch from urllib.error import URLError -from urllib.request import urlopen @public @@ -101,12 +101,12 @@ def get_org_data(): """Create a mock to load the organizational domain data from our local test data. """ - def ouropen(url): + def ourget(url): datapath = resource_filename( 'mailman.rules.tests.data', 'org_domain.txt') org_data_url = 'file:///{}'.format(datapath) - return urlopen(org_data_url) - return patch('mailman.rules.dmarc.request.urlopen', ouropen) + return _get(org_data_url) + return patch('mailman.rules.dmarc.protocols.get', ourget) class TestDMARCRules(TestCase): @@ -146,7 +146,7 @@ class TestDMARCRules(TestCase): def test_no_publicsuffix_dot_org(self): mark = LogFileMark('mailman.error') - with patch('mailman.rules.dmarc.request.urlopen', + with patch('mailman.rules.dmarc.protocols.get', side_effect=URLError('no internet')): domain = dmarc._get_org_dom('ssub.sub.city.kobe.jp') line = mark.readline() @@ -182,7 +182,7 @@ To: ant@example.com """) mark = LogFileMark('mailman.error') rule = dmarc.DMARCMitigation() - with get_dns_resolver(): + with get_dns_resolver(), get_org_data(): self.assertFalse(rule.check(mlist, msg, {})) line = mark.readline() self.assertEqual( |