commit 6fcf0121296067f7e47721af315ca5b85021e8b8 Author: Isis Lovecruft isis@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)]