forked from boorad/gen_bunny
-
Notifications
You must be signed in to change notification settings - Fork 16
/
test_gb.erl
101 lines (78 loc) · 3.27 KB
/
test_gb.erl
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
%% The MIT License
%% Copyright (c) David Reid <dreid@dreid.org>, Andy Gross <andy@andygross.org>
%% Permission is hereby granted, free of charge, to any person obtaining a copy
%% of this software and associated documentation files (the "Software"), to deal
%% in the Software without restriction, including without limitation the rights
%% to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
%% copies of the Software, and to permit persons to whom the Software is
%% furnished to do so, subject to the following conditions:
%% The above copyright notice and this permission notice shall be included in
%% all copies or substantial portions of the Software.
%% THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
%% IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
%% FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
%% AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
%% LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
%% OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
%% THE SOFTWARE.
%% @private
-module(test_gb).
-behavior(gen_bunny).
-export([start_link/1,
start_link/2,
stop/1]).
-export([init/1,
handle_message/2,
handle_call/3,
handle_cast/2,
handle_info/2,
terminate/2]).
-export([ack_stuff/2, get_messages/1, get_calls/1, get_casts/1, get_infos/1]).
-include("gen_bunny.hrl").
-record(state, {messages=[], calls=[], infos=[], casts=[]}).
start_link(Opts) ->
start_link(direct, Opts).
start_link(ConnectInfo, Opts) ->
gen_bunny:start_link(?MODULE, ConnectInfo, <<"bunny.test">>, Opts).
init([]) ->
{ok, #state{}}.
ack_stuff(Pid, Tag) ->
gen_bunny:cast(Pid, {ack_stuff, Tag}).
get_messages(Pid) ->
gen_bunny:call(Pid, get_messages).
get_calls(Pid) ->
gen_bunny:call(Pid, get_calls).
get_casts(Pid) ->
gen_bunny:call(Pid, get_casts).
get_infos(Pid) ->
gen_bunny:call(Pid, get_infos).
stop(Pid) ->
gen_bunny:call(Pid, stop).
handle_message(Message, State=#state{messages=Messages})
when ?is_message(Message) orelse ?is_tagged_message(Message) ->
NewMessages = [Message|Messages],
{noreply, State#state{messages=NewMessages}}.
handle_call(get_messages, _From, State=#state{messages=Messages}) ->
{reply, Messages, State};
handle_call(get_calls, _From, State=#state{calls=Calls}) ->
{reply, Calls, State};
handle_call(get_casts, _From, State=#state{casts=Casts}) ->
{reply, Casts, State};
handle_call(get_infos, _From, State=#state{infos=Infos}) ->
{reply, Infos, State};
handle_call(crash, _From, _State=#state{}) ->
erlang:error({badmatch, crashed});
handle_call(stop, _From, State=#state{}) ->
{stop, normal, ok, State};
handle_call(Msg, _From, State=#state{calls=Calls}) ->
{reply, ok, State#state{calls=[Msg|Calls]}}.
handle_cast({ack_stuff, Tag}, State) ->
gen_bunny:ack(Tag),
{noreply, State};
handle_cast(Msg, State=#state{casts=Casts}) ->
{noreply, State#state{casts=[Msg|Casts]}}.
handle_info(Info, State=#state{infos=Infos}) ->
{noreply, State#state{infos=[Info|Infos]}}.
terminate(Reason, _State) ->
io:format("~p terminating with reason ~p~n", [?MODULE, Reason]),
ok.