| 1 | -module(beam_stats_consumer_statsd_SUITE). |
| 2 | |
| 3 | -include_lib("beam_stats/include/beam_stats.hrl"). |
| 4 | -include_lib("beam_stats/include/beam_stats_ets_table.hrl"). |
| 5 | |
| 6 | -export( |
| 7 | [ all/0 |
| 8 | , groups/0 |
| 9 | ]). |
| 10 | |
| 11 | %% Test cases |
| 12 | -export( |
| 13 | [ t_send/1 |
| 14 | ]). |
| 15 | |
| 16 | -define(GROUP, beam_stats_consumer_statsd). |
| 17 | |
| 18 | %% ============================================================================ |
| 19 | %% Common Test callbacks |
| 20 | %% ============================================================================ |
| 21 | |
| 22 | all() -> |
| 23 | [{group, ?GROUP}]. |
| 24 | |
| 25 | groups() -> |
| 26 | Tests = |
| 27 | [ t_send |
| 28 | ], |
| 29 | Properties = [], |
| 30 | [{?GROUP, Properties, Tests}]. |
| 31 | |
| 32 | %% ============================================================================= |
| 33 | %% Test cases |
| 34 | %% ============================================================================= |
| 35 | |
| 36 | t_send(_Cfg) -> |
| 37 | ETSTableStatsFoo = |
| 38 | #beam_stats_ets_table |
| 39 | { id = foo |
| 40 | , name = foo |
| 41 | , size = 5 |
| 42 | , memory = 25 |
| 43 | }, |
| 44 | ETSTableStatsBar = |
| 45 | #beam_stats_ets_table |
| 46 | { id = 37 |
| 47 | , name = bar |
| 48 | , size = 8 |
| 49 | , memory = 38 |
| 50 | }, |
| 51 | % TODO: Indent #beam_stats as #beam_stats_ets_table |
| 52 | BEAMStats = #beam_stats |
| 53 | { timestamp = {1, 2, 3} |
| 54 | , node_id = 'node_foo@host_bar' |
| 55 | , memory = [{mem_type_foo, 1}, {mem_type_bar, 2}, {mem_type_baz, 3}] |
| 56 | , io_bytes_in = 3 |
| 57 | , io_bytes_out = 7 |
| 58 | , context_switches = 5 |
| 59 | , reductions = 9 |
| 60 | , run_queue = 17 |
| 61 | , ets = [ETSTableStatsFoo, ETSTableStatsBar] |
| 62 | }, |
| 63 | ServerPort = 8125, |
| 64 | {ok, ServerSocket} = gen_udp:open(ServerPort, [binary, {active, false}]), |
| 65 | BEAMStatsQ = queue:in(BEAMStats, queue:new()), |
| 66 | Options = [{dst_port, ServerPort}], |
| 67 | {_, State1} = beam_stats_consumer_statsd:init(Options), |
| 68 | State2 = beam_stats_consumer_statsd:consume(BEAMStatsQ, State1), |
| 69 | {} = beam_stats_consumer_statsd:terminate(State2), |
| 70 | ResultOfReceive1 = gen_udp:recv(ServerSocket, 0), |
| 71 | {ok, {_, _, PacketReceived1}} = ResultOfReceive1, |
| 72 | ResultOfReceive2 = gen_udp:recv(ServerSocket, 0), |
| 73 | {ok, {_, _, PacketReceived2}} = ResultOfReceive2, |
| 74 | ok = gen_udp:close(ServerSocket), |
| 75 | ct:log("PacketReceived1: ~n~s~n", [PacketReceived1]), |
| 76 | ct:log("PacketReceived2: ~n~s~n", [PacketReceived2]), |
| 77 | PacketsCombined = <<PacketReceived1/binary, PacketReceived2/binary>>, |
| 78 | ct:log("PacketsCombined: ~n~s~n", [PacketsCombined]), |
| 79 | MsgsExpected = |
| 80 | [ <<"beam_stats.node_foo_host_bar.io.bytes_in:3|g">> |
| 81 | , <<"beam_stats.node_foo_host_bar.io.bytes_out:7|g">> |
| 82 | , <<"beam_stats.node_foo_host_bar.context_switches:5|g">> |
| 83 | , <<"beam_stats.node_foo_host_bar.reductions:9|g">> |
| 84 | , <<"beam_stats.node_foo_host_bar.run_queue:17|g">> |
| 85 | , <<"beam_stats.node_foo_host_bar.memory.mem_type_foo:1|g">> |
| 86 | , <<"beam_stats.node_foo_host_bar.memory.mem_type_bar:2|g">> |
| 87 | , <<"beam_stats.node_foo_host_bar.memory.mem_type_baz:3|g">> |
| 88 | , <<"beam_stats.node_foo_host_bar.ets_table.size.foo.foo:5|g">> |
| 89 | , <<"beam_stats.node_foo_host_bar.ets_table.memory.foo.foo:25|g">> |
| 90 | , <<"beam_stats.node_foo_host_bar.ets_table.size.bar.37:8|g">> |
| 91 | , <<"beam_stats.node_foo_host_bar.ets_table.memory.bar.37:38|g">> |
| 92 | ], |
| 93 | MsgsReceived = binary:split(PacketsCombined, <<"\n">>, [global, trim]), |
| 94 | RemoveExpectedFromReceived = |
| 95 | fun (Expected, Received) -> |
| 96 | ct:log( |
| 97 | "Looking for expected msg ~p in remaining received ~p~n", |
| 98 | [Expected, Received] |
| 99 | ), |
| 100 | true = lists:member(Expected, Received), |
| 101 | Received -- [Expected] |
| 102 | end, |
| 103 | [] = lists:foldl(RemoveExpectedFromReceived, MsgsReceived, MsgsExpected). |