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)