From 1f48edd73b4d3ca2718de79bed33a635fdd1bd66 Mon Sep 17 00:00:00 2001
From: Scott Kitterman <scott@kitterman.com>
Date: Thu, 2 Mar 2017 19:18:02 -0500
Subject: [PATCH] Add test case for Body length not cast as bytes

---
 dkim/tests/test_dkim.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/dkim/tests/test_dkim.py b/dkim/tests/test_dkim.py
index c9918f9..4cf9b58 100644
--- a/dkim/tests/test_dkim.py
+++ b/dkim/tests/test_dkim.py
@@ -16,6 +16,7 @@
 #
 # Copyright (c) 2011 William Grant <me@williamgrant.id.au>
 
+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"):