X-Git-Url: https://git.xandkar.net/?a=blobdiff_plain;f=test%2Fbeam_stats_consumer_statsd_SUITE.erl;h=161eac199910d9a771c739d36073e8887a38c7eb;hb=b4e2333fc5fd9f32c8a0a39db4c6faacdbb15a91;hp=0f5f3d771e52961f84c06d2fc19f9ea96567e3d5;hpb=b0ab6ee2c0bea2f50a60d2260fc2e71dfb01320b;p=beam_stats.git

diff --git a/test/beam_stats_consumer_statsd_SUITE.erl b/test/beam_stats_consumer_statsd_SUITE.erl
index 0f5f3d7..161eac1 100644
--- a/test/beam_stats_consumer_statsd_SUITE.erl
+++ b/test/beam_stats_consumer_statsd_SUITE.erl
@@ -1,5 +1,7 @@
 -module(beam_stats_consumer_statsd_SUITE).
 
+-include_lib("beam_stats/include/beam_stats.hrl").
+
 -export(
     [ all/0
     , groups/0
@@ -7,15 +9,10 @@
 
 %% Test cases
 -export(
-    [ ct_test__beam_stats_to_bins/1
-    , ct_test__memory_component_to_statsd_msg/1
-    , ct_test__statsd_msg_add_name_prefix/1
-    , ct_test__statsd_msg_to_bin/1
-    , ct_test__node_id_to_bin/1
+    [ t_send/1
     ]).
 
--define(statsd_module, beam_stats_consumer_statsd).
--define(GROUP, ?statsd_module).
+-define(GROUP, beam_stats_consumer_statsd).
 
 %% ============================================================================
 %% Common Test callbacks
@@ -26,11 +23,7 @@ all() ->
 
 groups() ->
     Tests =
-        [ ct_test__beam_stats_to_bins
-        , ct_test__memory_component_to_statsd_msg
-        , ct_test__statsd_msg_add_name_prefix
-        , ct_test__statsd_msg_to_bin
-        , ct_test__node_id_to_bin
+        [ t_send
         ],
     Properties = [],
     [{?GROUP, Properties, Tests}].
@@ -39,17 +32,25 @@ groups() ->
 %%  Test cases
 %% =============================================================================
 
-ct_test__beam_stats_to_bins(_Cfg) ->
-    ?statsd_module:ct_test__beam_stats_to_bins(_Cfg).
-
-ct_test__memory_component_to_statsd_msg(_Cfg) ->
-    ?statsd_module:ct_test__memory_component_to_statsd_msg(_Cfg).
-
-ct_test__statsd_msg_add_name_prefix(_Cfg) ->
-    ?statsd_module:ct_test__statsd_msg_add_name_prefix(_Cfg).
-
-ct_test__statsd_msg_to_bin(_Cfg) ->
-    ?statsd_module:ct_test__statsd_msg_to_bin(_Cfg).
-
-ct_test__node_id_to_bin(_Cfg) ->
-    ?statsd_module:ct_test__node_id_to_bin(_Cfg).
+t_send(_Cfg) ->
+    BEAMStats = #beam_stats
+    { timestamp = {1, 2, 3}
+    , node_id   = 'node_foo@host_bar'
+    , memory    = [{mem_type_foo, 1}]
+    , io_bytes_in  = 3
+    , io_bytes_out = 7
+    },
+    ServerPort = 8125,
+    {ok, ServerSocket} = gen_udp:open(ServerPort, [binary, {active, false}]),
+    BEAMStatsQ = queue:in(BEAMStats, queue:new()),
+    Options = [{dst_port, ServerPort}],
+    {_, State1} = beam_stats_consumer_statsd:init(Options),
+    State2 = beam_stats_consumer_statsd:consume(BEAMStatsQ, State1),
+    {} = beam_stats_consumer_statsd:terminate(State2),
+    ResultOfReceive = gen_udp:recv(ServerSocket, 0),
+    ok = gen_udp:close(ServerSocket),
+    {ok, {_, _, Data}} = ResultOfReceive,
+    << "beam_stats.node_foo_host_bar.io.bytes_in:3|g\n"
+     , "beam_stats.node_foo_host_bar.io.bytes_out:7|g\n"
+     , "beam_stats.node_foo_host_bar.memory.mem_type_foo:1|g\n"
+    >> = Data.