[tor-commits] [bridgedb/develop] Add email.server.generateResponse() and MailResponse unittests.

[isis at torproject.org]

commit 9cf6d069e7c14edfc563617e1553afc899d576da
Author: Isis Lovecruft <isis at torproject.org>
Date:   Tue May 6 12:25:38 2014 +0000

    Add email.server.generateResponse() and MailResponse unittests.
---
 lib/bridgedb/test/test_email_server.py |   60 ++++++++++++++++++++++++++------
 1 file changed, 50 insertions(+), 10 deletions(-)

diff --git a/lib/bridgedb/test/test_email_server.py b/lib/bridgedb/test/test_email_server.py
index c7fdefb..5323e38 100644
--- a/lib/bridgedb/test/test_email_server.py
+++ b/lib/bridgedb/test/test_email_server.py
@@ -208,18 +208,58 @@ class CreateResponseBodyTests(unittest.TestCase):
         self.assertSubstring("obfs2", ret)
 
 
-class EmailReplyTests(unittest.TestCase):
-    """Tests for ``EmailServer.replyToMail()``."""
+class MailResponseTests(unittest.TestCase):
+    """Tests for ``generateResponse()`` and ``MailResponse``."""
 
     def setUp(self):
-        """Create fake email, distributor, and associated context data."""
-        # TODO: Add headers if we start validating them
-        self.lines = ["From: %s@%s.com",
-                      "To: bridges at example.net",
-                      "Subject: testing",
-                      "\n",
-                      "get bridges"]
-        self.ctx = _createMailContext()
+        self.fromAddr = "bridges at torproject.org"
+        self.clientAddr = "user at example.com"
+        self.body = """\
+People think that time is strictly linear, but, in reality, it's actually just
+a ball of timey-wimey, wibbly-warbly... stuff."""
+
+    def tearDown(self):
+        server.safelog.safe_logging = True
+
+    def test_generateResponse(self):
+        response = server.generateResponse(self.fromAddr, self.clientAddr,
+                                           self.body)
+        self.assertIsInstance(response, server.MailResponse)
+
+    def test_generateResponse_noSafelog(self):
+        server.safelog.safe_logging = False
+        response = server.generateResponse(self.fromAddr, self.clientAddr,
+                                           self.body)
+        self.assertIsInstance(response, server.MailResponse)
+
+    def test_generateResponse_mailfile(self):
+        response = server.generateResponse(self.fromAddr, self.clientAddr,
+                                           self.body)
+        self.assertIsInstance(response.mailfile, (io.BytesIO, io.StringIO))
+
+    def test_generateResponse_withInReplyTo(self):
+        response = server.generateResponse(self.fromAddr, self.clientAddr,
+                                           self.body, messageID="NSA")
+        contents = str(response.readContents()).replace('\x00', '')
+        self.assertIsInstance(response.mailfile, (io.BytesIO, io.StringIO))
+        self.assertSubstring("In-Reply-To: NSA", contents)
+
+    def test_generateResponse_readContents(self):
+        response = server.generateResponse(self.fromAddr, self.clientAddr,
+                                           self.body)
+        contents = str(response.readContents()).replace('\x00', '')
+        self.assertSubstring('timey-wimey, wibbly-warbly... stuff.', contents)
+
+    def test_MailResponse_additionalHeaders(self):
+        response = server.MailResponse()
+        response.writeHeaders(self.fromAddr, self.clientAddr,
+                              subject="Re: echelon", inReplyTo="NSA",
+                              X_been_there="They were so 2004")
+        contents = str(response.readContents()).replace('\x00', '')
+        self.assertIsInstance(response.mailfile, (io.BytesIO, io.StringIO))
+        self.assertSubstring("In-Reply-To: NSA", contents)
+        self.assertSubstring("X-been-there: They were so 2004", contents)
+
 
     def test_replyToMail(self):
         self.skip = True