[tor-commits] [bridgedb/develop] Add a mocked DummyIPBasedDistributor class to test_HTTPServer.py.

[isis at torproject.org]

commit 6fcf0121296067f7e47721af315ca5b85021e8b8
Author: Isis Lovecruft <isis at torproject.org>
Date:   Wed Mar 19 17:56:26 2014 +0000

     Add a mocked DummyIPBasedDistributor class to test_HTTPServer.py.
    
     A distributor is a required parameter to the
    `HTTPServer.WebResourceBridges` class. This one merely creates N random
    `DummyBridges`, where N is the number of bridges we're told to return to
    the client.
---
 lib/bridgedb/test/test_HTTPServer.py |   27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/lib/bridgedb/test/test_HTTPServer.py b/lib/bridgedb/test/test_HTTPServer.py
index 05e041c..e2856c7 100644
--- a/lib/bridgedb/test/test_HTTPServer.py
+++ b/lib/bridgedb/test/test_HTTPServer.py
@@ -418,3 +418,30 @@ class DummyBridge(object):
         bridgeLine = " ".join([item for item in line])
         #print "Created config line: %r" % bridgeLine
         return bridgeLine
+
+
+class DummyIPBasedDistributor(object):
+    """A mocked :class:`bridgedb.Dist.IPBasedDistributor` which is used to test
+    :class:`bridgedb.HTTPServer.WebResourceBridges.
+    """
+
+    def _dumbAreaMapper(ip): return ip
+
+    def __init__(self, areaMapper=None, nClusters=None, key=None,
+                 ipCategories=None, answerParameters=None):
+        """None of the parameters are really used, they are just there to retain
+        an identical method signature.
+        """
+        self.areaMapper = self._dumbAreaMapper
+        self.nClusters = 3
+        self.nBridgesToGive = 3
+        self.key = self.__class__.__name__
+        self.ipCategories = ipCategories
+        self.answerParameters = answerParameters
+
+    def getBridgesForIP(self, ip=None, epoch=None, N=1,
+                        countyCode=None, bridgeFilterRules=None):
+        """Needed because it's called in
+        :meth:`WebResourceBridges.getBridgesForIP`.
+        """
+        return [DummyBridge() for _ in xrange(N)]