[tor-commits] [bridgedb/develop] Add some unit tests for Bridges module

[isis at torproject.org]

commit b528a3b1469fe93c427bfbc9363f2bc7f2433312
Author: Matthew Finkel <Matthew.Finkel at gmail.com>
Date:   Sun Feb 2 06:59:16 2014 +0000

    Add some unit tests for Bridges module
---
 lib/bridgedb/test/test_Bridges.py |  111 +++++++++++++++++++++++++++++++++++++
 1 file changed, 111 insertions(+)

diff --git a/lib/bridgedb/test/test_Bridges.py b/lib/bridgedb/test/test_Bridges.py
new file mode 100644
index 0000000..4e0b27e
--- /dev/null
+++ b/lib/bridgedb/test/test_Bridges.py
@@ -0,0 +1,111 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of BridgeDB, a Tor bridge distribution system.
+#
+# :authors: please also see AUTHORS file
+# :copyright: (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.EmailServer` module."""
+
+from twisted.trial import unittest
+from binascii import a2b_hex
+from bridgedb import Bridges
+import hashlib
+try:
+    from cStringIO import StringIO
+except ImportError:
+    from io import StringIO
+
+class BridgeClassTest(unittest.TestCase):
+    """Tests for :class:`bridgedb.Bridges.Bridge`."""
+
+    def setUp(self):
+        self.nickname = 'unnamed'
+        self.ip = '127.0.0.1'
+        self.orport = '9001'
+        self.fingerprint = 'a1cc8dfef1fa11af9c40af1054df9daf45250556'
+        self.id_digest = a2b_hex(self.fingerprint)
+        self.or_addresses = {}
+
+    def test_init(self):
+        try:
+            Bridges.Bridge(self.nickname, self.ip, self.orport,
+                           fingerprint=self.fingerprint)
+            Bridges.Bridge(self.nickname, self.ip, self.orport,
+                           id_digest=self.id_digest)
+        except Exception as e:
+            self.fail("Unexpected %s exception." % type(e))
+
+        self.failUnlessRaises(TypeError, Bridges.Bridge,
+                              self.nickname, self.ip, self.orport,
+                              id_digest=self.id_digest[:-1])
+
+        self.failUnlessRaises(TypeError, Bridges.Bridge,
+                              self.nickname, self.ip, self.orport,
+                              fingerprint=self.fingerprint[:-1])
+
+        self.failUnlessRaises(TypeError, Bridges.Bridge,
+                              self.nickname, self.ip, self.orport)
+
+        invalid_fingerprint = self.fingerprint[:-1] + 'q'
+        self.failUnlessRaises(TypeError, Bridges.Bridge, self.nickname,
+                              self.ip, self.orport,
+                              fingerprint=invalid_fingerprint)
+
+    def test_Bridgehelperfunctions(self):
+        bridge = Bridges.Bridge(self.nickname, self.ip, self.orport,
+                                self.fingerprint)
+        self.assertEqual(self.id_digest, bridge.getID())
+
+        testtext = 'thisisatest'
+        bridge.setDescriptorDigest(testtext)
+        self.assertEqual(bridge.desc_digest, testtext)
+        bridge.setExtraInfoDigest(testtext)
+        self.assertEqual(bridge.ei_digest, testtext)
+        bridge.setVerified()
+        self.assertTrue(bridge.isVerified())
+        self.assertTrue(bridge.verified)
+        self.assertEqual(self.id_digest, bridge.getID())
+
+        self.assertFalse(bridge.running)
+        self.assertFalse(bridge.stable)
+        bridge.setStatus(True, True)
+        self.assertTrue(bridge.running)
+        self.assertTrue(bridge.stable)
+
+#    def test_isBlocked(self):
+        
+
+    def test_getDescriptorDigests(self):
+        sha1hash = hashlib.sha1()
+        ei_digest = 'abcdefghijklmno'
+
+        test = "this is a test line\nFollowed by another\n"
+        test += "extra-info-digest %s\n" % ei_digest
+        sha1hash.update(test)
+        digest = sha1hash.hexdigest()
+        test += "-----BEGIN SIGNATURE-----\n"
+        test += "This is a test line that should be skipped\n"
+        test += "-----END SIGNATURE-----\n"
+        digests = Bridges.getDescriptorDigests(StringIO(test))
+        self.failUnlessIn(digest, digests)
+        self.failUnlessEqual(ei_digest, digests[digest])
+
+    def test_getExtraInfoDigests(self):
+        sha1hash = hashlib.sha1()
+
+        test = "Many words and line all together\n"
+        test += "extra info is nothing like weather\n"
+        test += "it's certain to come, like the key in a hum\n"
+        test += "but sometimes without a transport and rum\n"
+        content = test
+        sha1hash.update(test)
+        digest = sha1hash.hexdigest()
+        test += "-----BEGIN SIGNATURE-----\n"
+        test += "But the rum can't save the world like you\n"
+        test += "-----END SIGNATURE-----\n"
+        digests = Bridges.getExtraInfoDigests(StringIO(test))
+        self.failUnlessIn(digest, digests)
+        self.failUnlessEqual(content, digests[digest].read())