Add test case for Body length not cast as bytes

dkim/tests/test_dkim.py

@@ -16,6 +16,7 @@
 #
 # Copyright (c) 2011 William Grant <[email protected]>
 
+import email
 import os.path
 import unittest
 import time
@@ -98,6 +99,14 @@ Y+vtSBczUiKERHv1yRbcaQtZFh5wtiRrN04BLUTD21MycBX5jYchHjPY/wIDAQAB"""
                 res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc)
                 self.assertTrue(res)
 
+    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)
+        res = dkim.verify(sig + self.message, dnsfunc=self.dnsfunc)
+        self.assertTrue(res)
+
     def test_altered_body_fails(self):
         # An altered body fails verification.
         for header_algo in (b"simple", b"relaxed"):