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