test/beam_stats_consumer_statsd_SUITE.erl

   1 -module(beam_stats_consumer_statsd_SUITE).
   2
   3 -include_lib("beam_stats/include/beam_stats.hrl").
   4
   5 -export(
   6     [ all/0
   7     , groups/0
   8     ]).
   9
  10 %% Test cases
  11 -export(
  12     [ t_send/1
  13     ]).
  14
  15 -define(GROUP, beam_stats_consumer_statsd).
  16
  17 %% ============================================================================
  18 %% Common Test callbacks
  19 %% ============================================================================
  20
  21 all() ->
  22     [{group, ?GROUP}].
  23
  24 groups() ->
  25     Tests =
  26         [ t_send
  27         ],
  28     Properties = [],
  29     [{?GROUP, Properties, Tests}].
  30
  31 %% =============================================================================
  32 %%  Test cases
  33 %% =============================================================================
  34
  35 t_send(_Cfg) ->
  36     BEAMStats = #beam_stats
  37     { timestamp = {1, 2, 3}
  38     , node_id   = 'node_foo@host_bar'
  39     , memory    = [{mem_type_foo, 1}]
  40     , io_bytes_in  = 3
  41     , io_bytes_out = 7
  42     , context_switches = 5
  43     , reductions       = 9
  44     },
  45     ServerPort = 8125,
  46     {ok, ServerSocket} = gen_udp:open(ServerPort, [binary, {active, false}]),
  47     BEAMStatsQ = queue:in(BEAMStats, queue:new()),
  48     Options = [{dst_port, ServerPort}],
  49     {_, State1} = beam_stats_consumer_statsd:init(Options),
  50     State2 = beam_stats_consumer_statsd:consume(BEAMStatsQ, State1),
  51     {} = beam_stats_consumer_statsd:terminate(State2),
  52     ResultOfReceive = gen_udp:recv(ServerSocket, 0),
  53     ok = gen_udp:close(ServerSocket),
  54     {ok, {_, _, Data}} = ResultOfReceive,
  55     << "beam_stats.node_foo_host_bar.io.bytes_in:3|g\n"
  56      , "beam_stats.node_foo_host_bar.io.bytes_out:7|g\n"
  57      , "beam_stats.node_foo_host_bar.context_switches:5|g\n"
  58      , "beam_stats.node_foo_host_bar.reductions:9|g\n"
  59      , "beam_stats.node_foo_host_bar.memory.mem_type_foo:1|g\n"
  60     >> = Data.