Skip to content

Commit e10f6f2

Browse files
committed
Add stream tests
1 parent 798bbfb commit e10f6f2

File tree

1 file changed

+33
-9
lines changed

1 file changed

+33
-9
lines changed

deps/rabbit/test/amqp_client_SUITE.erl

Lines changed: 33 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,12 @@ groups() ->
109109
dead_letter_headers_exchange,
110110
dead_letter_reject,
111111
immutable_bare_message,
112-
receive_many_made_available_over_time,
113-
receive_many_auto_flow,
112+
receive_many_made_available_over_time_classic_queue,
113+
receive_many_made_available_over_time_quorum_queue,
114+
receive_many_made_available_over_time_stream,
115+
receive_many_auto_flow_classic_queue,
116+
receive_many_auto_flow_quorum_queue,
117+
receive_many_auto_flow_stream,
114118
incoming_window_closed_transfer_flow_order,
115119
incoming_window_closed_stop_link,
116120
incoming_window_closed_close_link,
@@ -3979,11 +3983,19 @@ footer_checksum(FooterOpt, Config) ->
39793983
ok = end_session_sync(Session),
39803984
ok = amqp10_client:close_connection(Connection).
39813985

3986+
receive_many_made_available_over_time_classic_queue(Config) ->
3987+
receive_many_made_available_over_time(<<"classic">>, Config).
3988+
3989+
receive_many_made_available_over_time_quorum_queue(Config) ->
3990+
receive_many_made_available_over_time(<<"quorum">>, Config).
3991+
3992+
receive_many_made_available_over_time_stream(Config) ->
3993+
receive_many_made_available_over_time(<<"stream">>, Config).
3994+
39823995
%% This test grants many credits to the queue once while
39833996
%% messages are being made available at the source over time.
3984-
receive_many_made_available_over_time(Config) ->
3997+
receive_many_made_available_over_time(QType, Config) ->
39853998
QName = atom_to_binary(?FUNCTION_NAME),
3986-
QType = <<"quorum">>,
39873999
Address = <<"/queue/", QName/binary>>,
39884000
{Connection, Session, LinkPair} = init(Config),
39894001
QProps = #{arguments => #{<<"x-queue-type">> => {utf8, QType}}},
@@ -3994,8 +4006,10 @@ receive_many_made_available_over_time(Config) ->
39944006
%% Send first batch of messages.
39954007
ok = send_messages(Sender, 10, false),
39964008
ok = wait_for_accepts(10),
4009+
Filter = consume_from_first(QType),
39974010
{ok, Receiver} = amqp10_client:attach_receiver_link(
3998-
Session, <<"receiver">>, Address, settled),
4011+
Session, <<"receiver">>, Address,
4012+
settled, configuration, Filter),
39994013
flush(attached),
40004014
%% Grant many credits to the queue once.
40014015
ok = amqp10_client:flow_link_credit(Receiver, 5000, never),
@@ -4022,13 +4036,21 @@ receive_many_made_available_over_time(Config) ->
40224036

40234037
ok = amqp10_client:detach_link(Sender),
40244038
ok = amqp10_client:detach_link(Receiver),
4025-
{ok, #{message_count := 1}} = rabbitmq_amqp_client:delete_queue(LinkPair, QName),
4039+
{ok, _} = rabbitmq_amqp_client:delete_queue(LinkPair, QName),
40264040
ok = rabbitmq_amqp_client:detach_management_link_pair_sync(LinkPair),
40274041
ok = amqp10_client:close_connection(Connection).
40284042

4029-
receive_many_auto_flow(Config) ->
4043+
receive_many_auto_flow_classic_queue(Config) ->
4044+
receive_many_auto_flow(<<"classic">>, Config).
4045+
4046+
receive_many_auto_flow_quorum_queue(Config) ->
4047+
receive_many_auto_flow(<<"quorum">>, Config).
4048+
4049+
receive_many_auto_flow_stream(Config) ->
4050+
receive_many_auto_flow(<<"stream">>, Config).
4051+
4052+
receive_many_auto_flow(QType, Config) ->
40304053
QName = atom_to_binary(?FUNCTION_NAME),
4031-
QType = <<"quorum">>,
40324054
Address = <<"/queue/", QName/binary>>,
40334055
{Connection, Session, LinkPair} = init(Config),
40344056
QProps = #{arguments => #{<<"x-queue-type">> => {utf8, QType}}},
@@ -4041,8 +4063,10 @@ receive_many_auto_flow(Config) ->
40414063
ok = send_messages(Sender, Num, false),
40424064
ok = wait_for_accepts(Num),
40434065

4066+
Filter = consume_from_first(QType),
40444067
{ok, Receiver} = amqp10_client:attach_receiver_link(
4045-
Session, <<"receiver">>, Address, settled),
4068+
Session, <<"receiver">>, Address,
4069+
settled, configuration, Filter),
40464070
receive {amqp10_event, {link, Receiver, attached}} -> ok
40474071
after 5000 -> ct:fail(missing_attached)
40484072
end,

0 commit comments

Comments
 (0)