commit 7e6a652878ba3e7c29751540579ca12f01424a3a Author: Isis Lovecruft <isis@torproject.org> Date: Wed May 21 04:12:10 2014 +0000 Add unittests for MailResponse close(), write(), and writelines() methods. --- lib/bridgedb/test/test_email_server.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/lib/bridgedb/test/test_email_server.py b/lib/bridgedb/test/test_email_server.py index 01be6a7..7805458 100644 --- a/lib/bridgedb/test/test_email_server.py +++ b/lib/bridgedb/test/test_email_server.py @@ -266,6 +266,32 @@ a ball of timey-wimey, wibbly-warbly... stuff.""" self.assertSubstring("In-Reply-To: NSA", contents) self.assertSubstring("X-been-there: They were so 2004", contents) + def test_MailResponse_close(self): + """Calling MailResponse.close() should close the ``mailfile`` and set + ``closed=True``. + """ + response = server.MailResponse() + self.assertEqual(response.closed, False) + response.close() + self.assertEqual(response.closed, True) + self.assertRaises(ValueError, response.write, self.body) + + def test_MailResponse_write(self): + """Calling MailResponse.write() should write to the mailfile.""" + response = server.MailResponse() + response.write(self.body) + contents = str(response.readContents()).replace('\x00', '') + self.assertEqual(self.body.replace('\n', '\r\n') + '\r\n', contents) + + def test_MailResponse_writelines(self): + """Calling MailResponse.writelines() with a list should write the + concatenated contents of the list into the mailfile. + """ + response = server.MailResponse() + response.writelines(self.body.split('\n')) + contents = str(response.readContents()).replace('\x00', '') + self.assertEqual(self.body.replace('\n', '\r\n') + '\r\n', contents) + class MailMessageTests(unittest.TestCase): """Unittests for :class:`bridgedb.email.server.MailMessage`."""