root
/
rabbitmq-server
mirror of https://github.com/rabbitmq/rabbitmq-server.git
1
0
Fork 0
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Pull from socket up to 10 times in stream test utils (#13588) 

To make sure to have enough data to complete a command.
This commit is contained in:
Arnaud Cogoluègnes 2025-03-24 09:13:31 +01:00 committed by GitHub
parent 41dfa6aa99
commit b8244f70f4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 20 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
deps/rabbitmq_ct_helpers/src/stream_test_utils.erl vendored
View File

@ -77,6 +77,11 @@ subscribe(Sock, C0, Stream, SubscriptionId, InitialCredit) ->
{{response, 1, {subscribe, ?RESPONSE_CODE_OK}}, C1} = receive_stream_commands(Sock, C0),
{ok, C1}.
credit(Sock, Subscription, Credit) ->
CreditFrame = rabbit_stream_core:frame({credit, Subscription, Credit}),
ok = gen_tcp:send(Sock, CreditFrame),
ok.
unsubscribe(Sock, C0, SubscriptionId) ->
UnsubscribeFrame = rabbit_stream_core:frame({request, 1, {unsubscribe, SubscriptionId}}),
ok = gen_tcp:send(Sock, UnsubscribeFrame),
@ -149,20 +154,22 @@ sub_batch_entry_compressed(Sequence, Bodies) ->
<<Sequence:64, 1:1, 1:3, 0:4, (length(Bodies)):16, (byte_size(Uncompressed)):32,
CompressedLen:32, Compressed:CompressedLen/binary>>.
receive_stream_commands(Sock, C0) ->
receive_stream_commands(gen_tcp, Sock, C0).
receive_stream_commands(Transport, Sock, C0) ->
receive_stream_commands(Transport, Sock, C0, 10).
receive_stream_commands(_Transport, _Sock, C0, 0) ->
rabbit_stream_core:next_command(C0);
receive_stream_commands(Transport, Sock, C0, N) ->
case rabbit_stream_core:next_command(C0) of
empty ->
case gen_tcp:recv(Sock, 0, 5000) of
case Transport:recv(Sock, 0, 5000) of
{ok, Data} ->
C1 = rabbit_stream_core:incoming_data(Data, C0),
case rabbit_stream_core:next_command(C1) of
empty ->
{ok, Data2} = gen_tcp:recv(Sock, 0, 5000),
rabbit_stream_core:next_command(
rabbit_stream_core:incoming_data(Data2, C1));
Res ->
Res
end;
receive_stream_commands(Transport, Sock, C1, N - 1);
{error, Err} ->
ct:fail("error receiving stream data ~w", [Err])
end;

3
deps/rabbitmq_prometheus/test/rabbit_prometheus_http_SUITE.erl vendored
View File

@ -839,7 +839,8 @@ publish_via_stream_protocol(Stream, MsgPerBatch, Config) ->
{ok, C5} = stream_test_utils:publish(S, C4, PublisherId2, SequenceFrom2, Payloads2),
SubscriptionId = 97,
{ok, C6} = stream_test_utils:subscribe(S, C5, Stream, SubscriptionId, _InitialCredit = 1),
{ok, C6} = stream_test_utils:subscribe(S, C5, Stream, SubscriptionId, _InitialCredit = 0),
ok = stream_test_utils:credit(S, SubscriptionId, 1),
%% delivery of first batch of messages
{{deliver, SubscriptionId, _Bin1}, C7} = stream_test_utils:receive_stream_commands(S, C6),
{ok, S, C7}.

22
deps/rabbitmq_stream/test/rabbit_stream_SUITE.erl vendored
View File

@ -1569,26 +1569,8 @@ wait_for_socket_close(Transport, S, Attempt) ->
closed
end.
receive_commands(Transport, S, C0) ->
case rabbit_stream_core:next_command(C0) of
empty ->
case Transport:recv(S, 0, 5000) of
{ok, Data} ->
C1 = rabbit_stream_core:incoming_data(Data, C0),
case rabbit_stream_core:next_command(C1) of
empty ->
{ok, Data2} = Transport:recv(S, 0, 5000),
rabbit_stream_core:next_command(
rabbit_stream_core:incoming_data(Data2, C1));
Res ->
Res
end;
{error, Err} ->
ct:fail("error receiving data ~w", [Err])
end;
Res ->
Res
end.
receive_commands(Transport, S, C) ->
stream_test_utils:receive_stream_commands(Transport, S, C).
get_osiris_counters(Config) ->
rabbit_ct_broker_helpers:rpc(Config,