fixed byte issues with test_add_body_length

This commit is contained in:
Gene Shuman
2017-03-02 16:41:39 -08:00
parent 1f48edd73b
commit 497d74ca4b
+2 -2
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)