def test_with_router(self): c = DummyConnection(num=0, resp="foo") c2 = DummyConnection(num=1, resp="bar") # test dummy router r = DummyRouter() p = Cluster(hosts={0: c, 1: c2}, router=r) self.assertEquals(p.foo(), "foo") self.assertEquals(p.foo("foo"), "bar") # test default routing behavior p = Cluster(hosts={0: c, 1: c2}) self.assertEquals(p.foo(), ["foo", "bar"]) self.assertEquals(p.foo("foo"), ["foo", "bar"])
def test_with_router(self): c = DummyConnection(num=0, resp='foo') c2 = DummyConnection(num=1, resp='bar') # test dummy router r = DummyRouter() p = Cluster( hosts={0: c, 1: c2}, router=r, ) self.assertEquals(p.foo(), 'foo') self.assertEquals(p.foo('foo'), 'bar') # test default routing behavior p = Cluster( hosts={0: c, 1: c2}, ) self.assertEquals(p.foo(), ['foo', 'bar']) self.assertEquals(p.foo('foo'), ['foo', 'bar'])
def test_proxy(self): c = DummyConnection(num=1, resp="bar") p = Cluster(hosts={0: c}) self.assertEquals(p.foo(), "bar")