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