def test_realtime_host_w_failover(self): servers = ["127.0.0.1:21211", "127.0.0.1:21212"] mc = Client(servers) failover_mc = Client(servers, failover=True) hosts_visited = set() for i in range(1000): key = "test:realtime:route:%d" % i h1 = mc.get_host_by_key(key) h2 = mc.get_realtime_host_by_key(key) assert h1 is not None assert h1 == h2 h3 = failover_mc.get_host_by_key(key) h4 = failover_mc.get_realtime_host_by_key(key) assert h3 == h1 assert h4 == h1 hosts_visited.add(h1) assert len(hosts_visited) == len(servers)
def test_none_host(self): existed_server = "127.0.0.1:21211" not_existed_server = "127.0.0.1:1" servers = [existed_server, not_existed_server] mc = Client(servers) failover_mc = Client(servers, failover=True) hosts_visited = set() for i in range(1000): key = "test:realtime:route:%d" % i h1 = mc.get_host_by_key(key) h2 = mc.get_realtime_host_by_key(key) if h1 == existed_server: assert h1 == h2 elif h1 == not_existed_server: assert h2 is None h3 = failover_mc.get_host_by_key(key) h4 = failover_mc.get_realtime_host_by_key(key) assert h3 == h1 assert h4 == existed_server hosts_visited.add(h1) assert len(hosts_visited) == len(servers)