def test_message_merging(disable_reduce): if not disable_reduce: args = set_pod_parser().parse_args([]) else: args = set_pod_parser().parse_args(['--disable-reduce']) args.polling = PollingType.ALL cancel_event, handle_queue, runtime_thread = _create_runtime(args) assert handle_queue.empty() _add_worker(args, 'ip1', shard_id=0) _add_worker(args, 'ip2', shard_id=1) _add_worker(args, 'ip3', shard_id=2) assert handle_queue.empty() data_request = _create_test_data_message() result = GrpcConnectionPool.send_requests_sync( [data_request, data_request], f'{args.host}:{args.port}') assert result assert _queue_length(handle_queue) == 3 assert len(result.response.docs) == 2 if disable_reduce else 1 _destroy_runtime(args, cancel_event, runtime_thread)
def test_message_merging(disable_reduce): if not disable_reduce: args = set_pod_parser().parse_args([]) else: args = set_pod_parser().parse_args(['--disable-reduce']) args.polling = PollingType.ALL connection_list_dict = { 0: [f'ip1:8080'], 1: [f'ip2:8080'], 2: [f'ip3:8080'] } args.connection_list = json.dumps(connection_list_dict) cancel_event, handle_queue, runtime_thread = _create_runtime(args) assert handle_queue.empty() data_request = _create_test_data_message() result = GrpcConnectionPool.send_requests_sync( [data_request, data_request], f'{args.host}:{args.port}') assert result assert _queue_length(handle_queue) == 3 assert len(result.response.docs) == 2 if disable_reduce else 1 _destroy_runtime(args, cancel_event, runtime_thread)