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")
