1 files changed, 90 insertions, 0 deletions
diff --git a/deps/rabbitmq_cli/test/ctl/list_connections_command_test.exs b/deps/rabbitmq_cli/test/ctl/list_connections_command_test.exs
new file mode 100644
index 0000000000..9cfcb8787f
--- /dev/null
+++ b/deps/rabbitmq_cli/test/ctl/list_connections_command_test.exs
@@ -0,0 +1,90 @@
+defmodule ListConnectionsCommandTest do
+  use ExUnit.Case, async: false
+  import TestHelper
+
+  @command RabbitMQ.CLI.Ctl.Commands.ListConnectionsCommand
+  @user "guest"
+  @default_timeout 15000
+  @default_options %{table_headers: true}
+
+  setup_all do
+    RabbitMQ.CLI.Core.Distribution.start()
+
+    close_all_connections(get_rabbit_hostname())
+
+    on_exit([], fn ->
+      close_all_connections(get_rabbit_hostname())
+    end)
+
+    :ok
+  end
+
+  setup context do
+    {
+      :ok,
+      opts: %{
+        node: get_rabbit_hostname(),
+        timeout: context[:test_timeout] || @default_timeout
+      }
+    }
+  end
+
+  test "merge_defaults: user, peer_host, peer_port and state by default" do
+    assert @command.merge_defaults([], %{}) == {~w(user peer_host peer_port state), @default_options}
+  end
+
+  test "merge_defaults: includes table headers by default", _context do
+    {_, opts} = @command.merge_defaults([], %{})
+    assert opts[:table_headers]
+  end
+
+  test "validate: returns bad_info_key on a single bad arg", context do
+    assert @command.validate(["quack"], context[:opts]) ==
+      {:validation_failure, {:bad_info_key, [:quack]}}
+  end
+
+  test "validate: multiple bad args return a list of bad info key values", context do
+    assert @command.validate(["quack", "oink"], context[:opts]) ==
+      {:validation_failure, {:bad_info_key, [:oink, :quack]}}
+  end
+
+  test "validate: return bad_info_key on mix of good and bad args", context do
+    assert @command.validate(["quack", "peer_host"], context[:opts]) ==
+      {:validation_failure, {:bad_info_key, [:quack]}}
+    assert @command.validate(["user", "oink"], context[:opts]) ==
+      {:validation_failure, {:bad_info_key, [:oink]}}
+    assert @command.validate(["user", "oink", "peer_host"], context[:opts]) ==
+      {:validation_failure, {:bad_info_key, [:oink]}}
+  end
+
+  @tag test_timeout: 0
+  test "run: timeout causes command to return badrpc", context do
+    assert run_command_to_list(@command, [["name"], context[:opts]]) ==
+      [{:badrpc, {:timeout, 0.0}}]
+  end
+
+  test "run: filter single key", context do
+    vhost = "/"
+    with_connection(vhost, fn(_conn) ->
+      conns = run_command_to_list(@command, [["name"], context[:opts]])
+      assert (Enum.map(conns, &Keyword.keys/1) |> Enum.uniq) == [[:name]]
+      assert Enum.any?(conns, fn(conn) -> conn[:name] != nil end)
+    end)
+  end
+
+  test "run: show connection vhost", context do
+    vhost = "custom_vhost"
+    add_vhost vhost
+    set_permissions @user, vhost, [".*", ".*", ".*"]
+    on_exit(fn ->
+      delete_vhost vhost
+    end)
+    with_connection(vhost, fn(_conn) ->
+      conns = run_command_to_list(@command, [["vhost"], context[:opts]])
+      assert (Enum.map(conns, &Keyword.keys/1) |> Enum.uniq) == [[:vhost]]
+      assert Enum.any?(conns, fn(conn) -> conn[:vhost] == vhost end)
+    end)
+  end
+
+
+end