1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
%% This Source Code Form is subject to the terms of the Mozilla Public
%% License, v. 2.0. If a copy of the MPL was not distributed with this
%% file, You can obtain one at https://mozilla.org/MPL/2.0/.
%%
%% Copyright (c) 2018-2020 VMware, Inc. or its affiliates. All rights reserved.
%%
-module(unit_queue_consumers_SUITE).
-include_lib("common_test/include/ct.hrl").
-include_lib("eunit/include/eunit.hrl").
-compile(export_all).
all() ->
[
is_same,
get_consumer,
get,
list_consumers
].
is_same(_Config) ->
?assertEqual(
true,
rabbit_queue_consumers:is_same(
self(), <<"1">>,
consumer(self(), <<"1">>)
)),
?assertEqual(
false,
rabbit_queue_consumers:is_same(
self(), <<"1">>,
consumer(self(), <<"2">>)
)),
Pid = spawn(?MODULE, function_for_process, []),
Pid ! whatever,
?assertEqual(
false,
rabbit_queue_consumers:is_same(
self(), <<"1">>,
consumer(Pid, <<"1">>)
)),
ok.
get(_Config) ->
Pid = spawn(?MODULE, function_for_process, []),
Pid ! whatever,
State = state(consumers([consumer(self(), <<"1">>), consumer(Pid, <<"2">>), consumer(self(), <<"3">>)])),
{Pid, {consumer, <<"2">>, _, _, _, _}} =
rabbit_queue_consumers:get(Pid, <<"2">>, State),
?assertEqual(
undefined,
rabbit_queue_consumers:get(self(), <<"2">>, State)
),
?assertEqual(
undefined,
rabbit_queue_consumers:get(Pid, <<"1">>, State)
),
ok.
get_consumer(_Config) ->
Pid = spawn(unit_queue_consumers_SUITE, function_for_process, []),
Pid ! whatever,
State = state(consumers([consumer(self(), <<"1">>), consumer(Pid, <<"2">>), consumer(self(), <<"3">>)])),
{_Pid, {consumer, _, _, _, _, _}} =
rabbit_queue_consumers:get_consumer(State),
?assertEqual(
undefined,
rabbit_queue_consumers:get_consumer(state(consumers([])))
),
ok.
list_consumers(_Config) ->
State = state(consumers([consumer(self(), <<"1">>), consumer(self(), <<"2">>), consumer(self(), <<"3">>)])),
Consumer = rabbit_queue_consumers:get_consumer(State),
{_Pid, ConsumerRecord} = Consumer,
CTag = rabbit_queue_consumers:consumer_tag(ConsumerRecord),
ConsumersWithSingleActive = rabbit_queue_consumers:all(State, Consumer, true),
?assertEqual(3, length(ConsumersWithSingleActive)),
lists:foldl(fun({Pid, Tag, _, _, Active, ActivityStatus, _, _}, _Acc) ->
?assertEqual(self(), Pid),
case Tag of
CTag ->
?assert(Active),
?assertEqual(single_active, ActivityStatus);
_ ->
?assertNot(Active),
?assertEqual(waiting, ActivityStatus)
end
end, [], ConsumersWithSingleActive),
ConsumersNoSingleActive = rabbit_queue_consumers:all(State, none, false),
?assertEqual(3, length(ConsumersNoSingleActive)),
lists:foldl(fun({Pid, _, _, _, Active, ActivityStatus, _, _}, _Acc) ->
?assertEqual(self(), Pid),
?assert(Active),
?assertEqual(up, ActivityStatus)
end, [], ConsumersNoSingleActive),
ok.
consumers([]) ->
priority_queue:new();
consumers(Consumers) ->
consumers(Consumers, priority_queue:new()).
consumers([H], Q) ->
priority_queue:in(H, Q);
consumers([H | T], Q) ->
consumers(T, priority_queue:in(H, Q)).
consumer(Pid, ConsumerTag) ->
{Pid, {consumer, ConsumerTag, true, 1, [], <<"guest">>}}.
state(Consumers) ->
{state, Consumers, {}}.
function_for_process() ->
receive
_ -> ok
end.
|
