Update tests to check for Content-ID header.

Martin Davidsson · Martin Davidsson · commit 0061106258c2 · 2012-07-08T10:24:51.000-07:00
Change string formatting syntax for style purposes.
diff --git a/sendgrid/transport/smtp.py b/sendgrid/transport/smtp.py
@@ -183,6 +183,6 @@ def _getFileMIME(self, attach):
         msg.add_header('Content-Disposition', 'attachment', filename=filename)
         
         if attach['cid']:
-          msg.add_header('Content-ID', '<{0}>'.format(attach['cid']))
+          msg.add_header('Content-ID', '<%s>' % attach['cid'])
 
         return msg
diff --git a/test/test_smtp_transport.py b/test/test_smtp_transport.py
@@ -43,6 +43,16 @@ def test_smtp_transport_valid_password(self):
         smtp_quit.restore()
 
 
+    def test_smtp_transport_content_id_header(self):
+        smtp_transport = smtp.Smtp('username', 'password', tls=False)
+
+        f = smtp_transport._getFileMIME({'file': 'img.png', 'name': 'contents', 'cid': None})
+        self.assertEqual(None, f.get('Content-ID'))
+
+        f = smtp_transport._getFileMIME({'file': 'img.png', 'name': 'contents', 'cid': 'cid'})
+        self.assertEqual('<cid>', f.get('Content-ID'))
+
+
 class FakeException(IOError):
     def __init__(self, msg):
         self.msg = msg
