Fix the change made for issue 1243654.

Surprisingly, it turns out there was no test that exercised this code path.

2 files changed, 13 insertions, 1 deletions

diff --git a/Lib/email/generator.py b/Lib/email/generator.py
index 510f68b..9d33f1c 100644
--- a/Lib/email/generator.py
+++ b/Lib/email/generator.py
@@ -226,7 +226,8 @@ class Generator:
             # Create a boundary that doesn't appear in any of the
             # message texts.
             alltext = self._encoded_NL.join(msgtexts)
-            msg.set_boundary(self._make_boundary(alltext))
+            boundary = self._make_boundary(alltext)
+            msg.set_boundary(boundary)
         # If there's a preamble, write it out, with a trailing CRLF
         if msg.preamble is not None:
             self.write(msg.preamble + self._NL)
diff --git a/Lib/email/test/test_email.py b/Lib/email/test/test_email.py
index e5eece2..a54c1a3 100644
--- a/Lib/email/test/test_email.py
+++ b/Lib/email/test/test_email.py
@@ -180,6 +180,17 @@ class TestMessageAPI(TestEmailBase):
         self.assertRaises(errors.HeaderParseError,
                           msg.set_boundary, 'BOUNDARY')
 
+    def test_make_boundary(self):
+        msg = MIMEMultipart('form-data')
+        # Note that when the boundary gets created is an implementation
+        # detail and might change.
+        self.assertEqual(msg.items()[0][1], 'multipart/form-data')
+        # Trigger creation of boundary
+        msg.as_string()
+        self.assertEqual(msg.items()[0][1][:33],
+                        'multipart/form-data; boundary="==')
+        # XXX: there ought to be tests of the uniqueness of the boundary, too.
+
     def test_message_rfc822_only(self):
         # Issue 7970: message/rfc822 not in multipart parsed by
         # HeaderParser caused an exception when flattened.