Fix dnsfunc definition for test suite for compatibility with dkimpy >= 1.0

This commit is contained in:
Scott Kitterman
2020-04-19 02:28:22 -04:00
parent 7be865d7d7
commit 7b37e2cb8d
+5 -1
View File
@@ -363,8 +363,12 @@ class dkimMilter(Milter.Base):
try: try:
dnsoverride = self.conf.get('DNSOverride') dnsoverride = self.conf.get('DNSOverride')
if isinstance(dnsoverride, str): if isinstance(dnsoverride, str):
timeout = 5
domain = self.fdomain
def dnsfunc(domain, timeout=timeout, dnsoverride=dnsoverride):
return dnsoverride
syslog.syslog("DNSOverride: {0}".format(dnsoverride)) syslog.syslog("DNSOverride: {0}".format(dnsoverride))
res = d.verify(idx=y, dnsfunc=lambda _x: dnsoverride) res = d.verify(idx=y, dnsfunc=dnsfunc)
else: else:
res = d.verify(idx=y) res = d.verify(idx=y)
algo = codecs.decode(d.signature_fields.get(b'a'), 'ascii') algo = codecs.decode(d.signature_fields.get(b'a'), 'ascii')