[tor-commits] [bridgedb/master] Add unittests for bridgedb.bridgerequest.

[isis at torproject.org]

commit 1e7e8324ddb113920348990172e194440f322f55
Author: Isis Lovecruft <isis at torproject.org>
Date:   Thu Aug 21 02:42:14 2014 +0000

    Add unittests for bridgedb.bridgerequest.
---
 lib/bridgedb/test/test_bridgerequest.py |   44 +++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/lib/bridgedb/test/test_bridgerequest.py b/lib/bridgedb/test/test_bridgerequest.py
new file mode 100644
index 0000000..1db8814
--- /dev/null
+++ b/lib/bridgedb/test/test_bridgerequest.py
@@ -0,0 +1,44 @@
+# -*- 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) 2007-2014, The Tor Project, Inc.
+#             (c) 2014, Isis Lovecruft
+# :license: see LICENSE for licensing information
+#_____________________________________________________________________________
+
+
+from twisted.trial import unittest
+
+from bridgedb import bridgerequest
+from bridgedb.bridgerequest import IBridgeRequest
+from bridgedb.bridgerequest import BridgeRequestBase
+
+
+class BridgeRequestBaseTests(unittest.TestCase):
+    """Unittests for :class:`bridgedb.bridgerequest.BridgeRequestBase`."""
+
+    def setUp(self):
+        """Setup test run."""
+        self.request = BridgeRequestBase()
+
+    def test_BridgeRequestBase_implements_IRequestBridges(self):
+        """BridgeRequestBase should implement IRequestBridges interface."""
+        self.assertTrue(IBridgeRequest.implementedBy(BridgeRequestBase))
+
+    def test_BridgeRequestBase_withoutBlockInCountry(self):
+        """BridgeRequestBase.withoutBlockInCountry() should add the country CC
+        to the ``notBlockedIn`` attribute.
+        """
+        self.request.withoutBlockInCountry('US')
+        self.assertIn('US', self.request.notBlockedIn)
+
+    def test_BridgeRequestBase_withPluggableTransportType(self):
+        """BridgeRequestBase.withPluggableTransportType() should add the
+        pluggable transport type to the ``transport`` attribute.
+        """
+        self.request.withPluggableTransportType('huggable-transport')
+        self.assertIn('huggable-transport', self.request.transports)