|
9 | 9 |
|
10 | 10 | -include_lib("common_test/include/ct.hrl"). |
11 | 11 | -include_lib("eunit/include/eunit.hrl"). |
| 12 | + |
| 13 | +-include_lib("rabbitmq_ct_helpers/include/rabbit_assert.hrl"). |
| 14 | + |
12 | 15 | -compile([export_all, nowarn_export_all]). |
13 | 16 |
|
14 | 17 | -import(rabbit_ct_broker_helpers, [rpc/5]). |
@@ -136,6 +139,14 @@ shovel(SrcNode, DestNode, ShovelNode, Config) -> |
136 | 139 | ok = rpc(Config, ShovelNode, rabbit_runtime_parameters, clear, |
137 | 140 | [<<"/">>, <<"shovel">>, ShovelName, none]), |
138 | 141 | ExpectedQueueLen = 0, |
| 142 | + ?awaitMatch( |
| 143 | + [ExpectedQueueLen], |
| 144 | + rpc(Config, ?OLD, ?MODULE, queues_length, []), |
| 145 | + 30000), |
| 146 | + ?awaitMatch( |
| 147 | + [ExpectedQueueLen], |
| 148 | + rpc(Config, ?NEW, ?MODULE, queues_length, []), |
| 149 | + 30000), |
139 | 150 | ?assertEqual([ExpectedQueueLen], rpc(Config, ?OLD, ?MODULE, delete_queues, [])), |
140 | 151 | ?assertEqual([ExpectedQueueLen], rpc(Config, ?NEW, ?MODULE, delete_queues, [])). |
141 | 152 |
|
@@ -170,6 +181,12 @@ flush(Prefix) -> |
170 | 181 | ok |
171 | 182 | end. |
172 | 183 |
|
| 184 | +queues_length() -> |
| 185 | + [begin |
| 186 | + [{messages, N}] = rabbit_amqqueue:info(Q, [messages]), |
| 187 | + N |
| 188 | + end || Q <- rabbit_amqqueue:list()]. |
| 189 | + |
173 | 190 | delete_queues() -> |
174 | 191 | [begin |
175 | 192 | {ok, N} = rabbit_amqqueue:delete(Q, false, false, <<"tests">>), |
|
0 commit comments