fixed byte issues with test_add_body_length

2017-03-02 16:41:39 -08:00
parent 1f48edd73b
commit 497d74ca4b
1 changed files with 4 additions and 4 deletions
@@ -102,8 +102,8 @@ Y+vtSBczUiKERHv1yRbcaQtZFh5wtiRrN04BLUTD21MycBX5jYchHjPY/wIDAQAB"""
    def test_add_body_length(self):
        sig = dkim.sign(
            self.message, b"test", b"example.com", self.key, length=True)
-        msg = email.message_from_string(self.message)
-        self.assertIn('\n l=%s' % len(msg.get_payload() + '\n'), sig)
+        msg = email.message_from_string(self.message.decode('utf-8'))
+        self.assertIn('\n l=%s' % len(msg.get_payload() + '\n'), sig.decode('utf-8'))
        res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc)
        self.assertTrue(res)