[tor-commits] [bridgedb/develop] Add unittests for bridgedb.email.dkim module.

[isis at torproject.org]

commit ac36d66d7e25bbc0aa4dcef2ce364ca7aca841ea
Author: Isis Lovecruft <isis at torproject.org>
Date:   Mon Jun 2 21:08:35 2014 +0000

    Add unittests for bridgedb.email.dkim module.
---
 lib/bridgedb/test/test_email_dkim.py |   59 ++++++++++++++++++++++++++++++++++
 1 file changed, 59 insertions(+)

diff --git a/lib/bridgedb/test/test_email_dkim.py b/lib/bridgedb/test/test_email_dkim.py
new file mode 100644
index 0000000..3350d19
--- /dev/null
+++ b/lib/bridgedb/test/test_email_dkim.py
@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of BridgeDB, a Tor bridge distribution system.
+#
+# :authors: Isis Lovecruft 0xA3ADB67A2CDB8B35 <isis at torproject.org>
+#           please also see AUTHORS file
+# :copyright: (c) 2013, Isis Lovecruft
+#             (c) 2007-2013, The Tor Project, Inc.
+#             (c) 2007-2013, all entities within the AUTHORS file
+# :license: 3-Clause BSD, see LICENSE for licensing information
+
+"""Unittests for the :mod:`bridgedb.email.dkim` module."""
+
+import io
+
+from twisted.mail.smtp import rfc822
+from twisted.trial import unittest
+
+from bridgedb.email import dkim
+
+
+class CheckDKIMTests(unittest.TestCase):
+    """Tests for :func:`email.server.checkDKIM`."""
+
+    def setUp(self):
+        """Create fake email, distributor, and associated context data."""
+        self.goodMessage = io.StringIO(unicode("""\
+From: user at gmail.com
+To: bridges at localhost
+X-DKIM-Authentication-Results: pass
+Subject: testing
+
+get bridges
+"""))
+        self.badMessage = io.StringIO(unicode("""\
+From: user at gmail.com
+To: bridges at localhost
+Subject: testing
+
+get bridges
+"""))
+        self.domainRules = {
+            'gmail.com': ["ignore_dots", "dkim"],
+            'example.com': [],
+            'localhost': [],
+        }
+
+    def test_checkDKIM_good(self):
+        message = rfc822.Message(self.goodMessage)
+        result = dkim.checkDKIM(message,
+                                self.domainRules.get("gmail.com"))
+        self.assertTrue(result)
+                                         
+
+    def test_checkDKIM_bad(self):
+        message = rfc822.Message(self.badMessage)
+        result = dkim.checkDKIM(message,
+                                self.domainRules.get("gmail.com"))
+        self.assertIs(result, False)