blob: 930741997ab2ecb32c8497b20d157cbc88cd061a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
# Copyright (C) 2016 by the Free Software Foundation, Inc.
#
# This file is part of GNU Mailman.
#
# GNU Mailman is free software: you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# GNU Mailman is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# GNU Mailman. If not, see <http://www.gnu.org/licenses/>.
"""Support for mocking dnspython calls from dmarc rules."""
from dns.rdatatype import TXT
from dns.resolver import NXDOMAIN, NoAnswer
from public import public
from unittest import mock
@public
def get_dns_resolver():
"""Create a dns.resolver.Resolver mock.
This is used to return a predictable response to a _dmarc query. It
returns p=reject for the example.biz domain and raises either NXDOMAIN
or NoAnswer for any other.
It only implements those classes and attributes used by the dmarc rule.
"""
class Name:
# mock answer.name
def __init__(self):
pass
def to_text(self):
return '_dmarc.example.biz.'
class Item:
# mock answer.items
def __init__(self):
self.strings = [b'v=DMARC1; p=reject;']
class Ans_e:
# mock answer element
def __init__(self):
self.rdtype = TXT
self.items = [Item()]
self.name = Name()
class Answer:
# mock answer
def __init__(self):
self.answer = [Ans_e()]
class Resolver:
# mock dns.resolver.Resolver class.
def __init__(self):
pass
def query(self, domain, data_type):
if data_type != TXT:
raise NoAnswer
dparts = domain.split('.')
if len(dparts) < 3:
raise NXDOMAIN
if len(dparts) > 3:
raise NoAnswer
if dparts[0] != '_dmarc':
raise NoAnswer
if dparts[1] != 'example' or dparts[2] != 'biz':
raise NXDOMAIN
self.response = Answer()
return self
patcher = mock.patch('dns.resolver.Resolver', Resolver)
return patcher
|
