DandelionNStuff/dkimpy-smtputf8
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed byte issues with test_add_body_length

Browse Source
This commit is contained in:
Gene Shuman
2017-03-02 16:41:39 -08:00
parent 1f48edd73b
commit 497d74ca4b
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dkim/tests/test_dkim.py
+4 -4
View File
@@ -102,8 +102,8 @@ Y+vtSBczUiKERHv1yRbcaQtZFh5wtiRrN04BLUTD21MycBX5jYchHjPY/wIDAQAB"""
def test_add_body_length(self): def test_add_body_length(self):
sig = dkim.sign( sig = dkim.sign(
self.message, b"test", b"example.com", self.key, length=True) self.message, b"test", b"example.com", self.key, length=True)
msg = email.message_from_string(self.message) msg = email.message_from_string(self.message.decode('utf-8'))
self.assertIn('\n l=%s' % len(msg.get_payload() + '\n'), sig) self.assertIn('\n l=%s' % len(msg.get_payload() + '\n'), sig.decode('utf-8'))
res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc) res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc)
self.assertTrue(res) self.assertTrue(res)
@@ -163,7 +163,7 @@ b/mPfjC0QJTocVBq6Za/PlzfV+Py92VaCak19F4WrbVTK5Gg5tW220MCAwEAAQ==
# sig_value with empty b= before hashing, and then appends the # sig_value with empty b= before hashing, and then appends the
# signature. So folding dkim_header again adds FWS to # signature. So folding dkim_header again adds FWS to
# the b= tag only. This should be ignored even with # the b= tag only. This should be ignored even with
# simple canonicalization. # simple canonicalization.
# http://tools.ietf.org/html/rfc4871#section-3.5 # http://tools.ietf.org/html/rfc4871#section-3.5
signed = dkim.fold(dkim_header) + sample_msg signed = dkim.fold(dkim_header) + sample_msg
result = dkim.verify(signed,dnsfunc=self.dnsfunc, result = dkim.verify(signed,dnsfunc=self.dnsfunc,
@@ -184,7 +184,7 @@ b/mPfjC0QJTocVBq6Za/PlzfV+Py92VaCak19F4WrbVTK5Gg5tW220MCAwEAAQ==
dv = dkim.DKIM(message) dv = dkim.DKIM(message)
res = dv.verify(dnsfunc=self.dnsfunc) res = dv.verify(dnsfunc=self.dnsfunc)
self.assertTrue(res) self.assertTrue(res)
def test_extra_headers(self): def test_extra_headers(self):
# <https://bugs.launchpad.net/dkimpy/+bug/737311> # <https://bugs.launchpad.net/dkimpy/+bug/737311>
# extra headers above From caused failure # extra headers above From caused failure