lots of tests

1 files changed, 44 insertions, 19 deletions

diff --git a/src/rabbit_tests.erl b/src/rabbit_tests.erl
index 65d16c94..1853a855 100644
--- a/src/rabbit_tests.erl
+++ b/src/rabbit_tests.erl
@@ -35,6 +35,7 @@
 
 -import(lists).
 
+-include("rabbit.hrl").
 -include_lib("kernel/include/file.hrl").
 
 test_content_prop_roundtrip(Datum, Binary) ->
@@ -497,27 +498,42 @@ test_user_management() ->
 
 test_server_status() ->
 
-    % list queues
-    ok = control_action(list_queues, []),
-    ok = control_action(list_queues, ["name", "durable", "auto_delete",
-            "arguments", "pid", "messages_ready", "messages_unacknowledged", 
-            "messages_uncommitted", "messages", "acks_uncommitted", "consumers", 
-            "transactions", "memory"]),
-
-    % list exchanges
-    ok = control_action(list_exchanges, []),
-    ok = control_action(list_exchanges, ["name", "type", "durable", "auto_delete", 
-            "arguments"]),
-
-    % list bindings
+    %% create a queue so we have something to list
+    Q = #amqqueue{} = rabbit_amqqueue:declare(
+                        rabbit_misc:r(<<"/">>, queue, <<"foo">>),
+                        false, false, []),
+
+    %% list queues
+    ok = info_action(
+           list_queues,
+           [name, durable, auto_delete, arguments, pid,
+            messages_ready, messages_unacknowledged, messages_uncommitted,
+            messages, acks_uncommitted, consumers, transactions, memory],
+           true),
+
+    %% list exchanges
+    ok = info_action(
+           list_exchanges,
+           [name, type, durable, auto_delete, arguments],
+           true),
+
+    %% list bindings
     ok = control_action(list_bindings, []),
 
-    % list connections
-    ok = control_action(list_connections, []),
-    ok = control_action(list_connections, ["pid", "address", "port", "peer_address", 
-            "peer_port", "state", "channels", "user", "vhost", "timeout", 
-            "frame_max", "recv_oct", "recv_cnt", "send_oct", "send_cnt", 
-            "send_pend passed"]),
+    %% cleanup
+    {ok, _} = rabbit_amqqueue:delete(Q, false, false),
+
+    %% list connections
+    [#listener{host = H, port = P} | _] = rabbit_networking:active_listeners(),
+    {ok, C} = gen_tcp:connect(H, P, []),
+    timer:sleep(100),
+    ok = info_action(
+           list_connections,
+           [pid, address, port, peer_address, peer_port, state,
+            channels, user, vhost, timeout, frame_max,
+            recv_oct, recv_cnt, send_oct, send_cnt, send_pend],
+           false),
+    ok = gen_tcp:close(C),
 
     passed.
 
@@ -539,6 +555,15 @@ control_action(Command, Node, Args) ->
             Other
     end.
 
+info_action(Command, Args, CheckVHost) ->
+    ok = control_action(Command, []),
+    if CheckVHost -> ok = control_action(Command, ["-p", "/"]);
+       true       -> ok
+    end,
+    ok = control_action(Command, lists:map(fun atom_to_list/1, Args)),
+    {bad_argument, dummy} = control_action(Command, ["dummy"]),
+    ok.
+
 empty_files(Files) ->
     [case file:read_file_info(File) of
          {ok, FInfo} -> FInfo#file_info.size == 0;