[tor-commits] [bridgedb/develop] Add unittests for bridgedb.parse.addr.canonicalizeEmailDomain().

[isis at torproject.org]

commit 46e16fa31c5590e7db2b092029124d90ea061e1d
Author: Isis Lovecruft <isis at torproject.org>
Date:   Wed Apr 23 06:53:50 2014 +0000

    Add unittests for bridgedb.parse.addr.canonicalizeEmailDomain().
---
 lib/bridgedb/test/test_parse_addr.py |   34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/lib/bridgedb/test/test_parse_addr.py b/lib/bridgedb/test/test_parse_addr.py
index 9714cc1..6b22232 100644
--- a/lib/bridgedb/test/test_parse_addr.py
+++ b/lib/bridgedb/test/test_parse_addr.py
@@ -45,6 +45,40 @@ IP6UniqueLocal = "fc00::"
 IP6SiteLocal = "fec0::"
 
 
+class CanonicalizeEmailDomainTests(unittest.TestCase):
+    """Unittests for :func:`bridgedb.parse.addr.canonicalizeEmailDomain`."""
+
+    def test_nonDict(self):
+        """Using a non-dict domainmap as a parameter to
+        canonicalizeEmailDomain() should log an AttributeError and then raise
+        an UnsupportedDomain error.
+        """
+        domainmap = 'example.com'
+        domain = 'fubar.com'
+        self.assertRaises(addr.UnsupportedDomain,
+                          addr.canonicalizeEmailDomain,
+                          domain, domainmap)
+
+    def test_notPermitted(self):
+        """A domain not in the domainmap of allowed domains should raise an
+        UnsupportedDomain error.
+        """
+        domainmap = {'foo.example.com': 'example.com'}
+        domain = 'bar.example.com'
+        self.assertRaises(addr.UnsupportedDomain,
+                          addr.canonicalizeEmailDomain,
+                          domain, domainmap)
+
+    def test_permitted(self):
+        """A domain in the domainmap of allowed domains should return the
+        canonical domain.
+        """
+        domainmap = {'foo.example.com': 'example.com'}
+        domain = 'foo.example.com'
+        canonical = addr.canonicalizeEmailDomain(domain, domainmap)
+        self.assertEquals(canonical, 'example.com')
+
+
 class ParseAddrIsIPAddressTests(unittest.TestCase):
     """Unittests for :func:`bridgedb.parse.addr.isIPAddress`.