def test_random_ready_conn(): conn_1 = Nsqd('nsq1') conn_2 = Nsqd('nsq2') reader = Reader('test', 'test', 'localhost:4150') reader.max_in_flight = 2 reader.conns = [conn_2] reader.last_random_ready = 0 assert reader.random_ready_conn(conn_1) is conn_1 assert reader.last_random_ready == 0 reader.max_in_flight = 1 reader.conns = [conn_2, conn_2] reader.last_random_ready = last_random_ready = time.time() assert reader.random_ready_conn(conn_1) is conn_1 assert reader.last_random_ready == last_random_ready reader.max_in_flight = 1 reader.conns = [conn_2, conn_2] reader.last_random_ready = 0 assert reader.random_ready_conn(conn_1) is conn_2 assert reader.last_random_ready != 0 reader.max_in_flight = 1 reader.conns = [conn_2, conn_2] reader.last_random_ready = 0 conn_2.ready_count = 1 assert reader.random_ready_conn(conn_1) is conn_1 assert reader.last_random_ready == 0