Test close all connections command

deps/rabbitmq_cli/lib/rabbitmq/cli/ctl/commands/close_all_connections_command.ex

@@ -35,9 +35,14 @@ defmodule RabbitMQ.CLI.Ctl.Commands.CloseAllConnectionsCommand do
                 :rabbit_misc.rpc_call(node_name, :rabbit_connection_tracking,
                   :list_on_node, [node_name])
             end
-    :rabbit_misc.rpc_call(node_name, :rabbit_connection_tracking_handler,
-      :close_connections, [conns, explanation, delay])
-    {:ok, "Closed #{length(conns)} connections"}
+    case conns do
+      {:badrpc, _} = err ->
+        err
+      _ ->
+        :rabbit_misc.rpc_call(node_name, :rabbit_connection_tracking_handler,
+          :close_connections, [conns, explanation, delay])
+        {:ok, "Closed #{length(conns)} connections"}
+    end
   end
 
   def output({:stream, stream}, _opts) do

deps/rabbitmq_cli/test/close_all_connections_command_test.exs

@@ -0,0 +1,118 @@
+## The contents of this file are subject to the Mozilla Public License
+## Version 1.1 (the "License"); you may not use this file except in
+## compliance with the License. You may obtain a copy of the License
+## at http://www.mozilla.org/MPL/
+##
+## Software distributed under the License is distributed on an "AS IS"
+## basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
+## the License for the specific language governing rights and
+## limitations under the License.
+##
+## The Original Code is RabbitMQ.
+##
+## The Initial Developer of the Original Code is GoPivotal, Inc.
+## Copyright (c) 2007-2016 Pivotal Software, Inc.  All rights reserved.
+
+
+defmodule CloseAllConnectionsCommandTest do
+  use ExUnit.Case, async: false
+  import TestHelper
+
+  alias RabbitMQ.CLI.Ctl.RpcStream, as: RpcStream
+
+  @helpers RabbitMQ.CLI.Core.Helpers
+
+  @command RabbitMQ.CLI.Ctl.Commands.CloseAllConnectionsCommand
+
+  @vhost "/"
+  
+  setup_all do
+    RabbitMQ.CLI.Core.Distribution.start()
+    :net_kernel.connect_node(get_rabbit_hostname)
+    close_all_connections(get_rabbit_hostname)
+
+    on_exit([], fn ->
+      close_all_connections(get_rabbit_hostname)
+      :erlang.disconnect_node(get_rabbit_hostname)
+
+    end)
+
+    :ok
+  end
+
+  test "validate: with an invalid number of arguments returns an arg count error", context do
+    assert @command.validate(["random", "explanation"], context[:opts]) == {:validation_failure, :too_many_args}
+    assert @command.validate([], context[:opts]) == {:validation_failure, :not_enough_args}
+  end
+
+  test "validate: with the correct number of arguments returns ok", context do
+    assert @command.validate(["explanation"], context[:opts]) == :ok
+  end
+
+  test "run: a close connections request on an existing vhost", context do
+    with_connection(@vhost, fn(_) ->
+      node = @helpers.parse_node(context[:node])
+      nodes = @helpers.nodes_in_cluster(node)
+      [[vhost: @vhost]] = fetch_connections_vhosts(node, nodes)
+      opts = %{node: node, vhost: @vhost, global: false, per_connection_delay: 0}
+      assert {:ok, "Closed 1 connections"} == @command.run(["test"], opts)
+      assert fetch_connections_vhosts(node, nodes) == []
+    end)
+  end
+
+  test "run: a close connections request for a non-existing vhost does nothing", context do
+    with_connection(@vhost, fn(_) ->
+      node = @helpers.parse_node(context[:node])
+      nodes = @helpers.nodes_in_cluster(node)
+      [[vhost: @vhost]] = fetch_connections_vhosts(node, nodes)
+      opts = %{node: node, vhost: "burrow", global: false, per_connection_delay: 0}
+      assert {:ok, "Closed 0 connections"} == @command.run(["test"], opts)
+      assert fetch_connections_vhosts(node, nodes) == [[vhost: @vhost]]
+      close_all_connections(node)
+    end)
+  end
+
+  test "run: a close connections request on an existing node", context do
+    with_connection(@vhost, fn(_) ->
+      node = @helpers.parse_node(context[:node])
+      nodes = @helpers.nodes_in_cluster(node)
+      [[vhost: @vhost]] = fetch_connections_vhosts(node, nodes)
+      opts = %{node: node, vhost: "fakeit", global: true, per_connection_delay: 0}
+      assert {:ok, "Closed 1 connections"} == @command.run(["test"], opts)
+      assert fetch_connections_vhosts(node, nodes) == []
+    end)
+  end
+
+  test "run: a close_all_connections request on nonexistent RabbitMQ node returns nodedown" do
+    target = :jake@thedog
+    :net_kernel.connect_node(target)
+    opts = %{node: target, vhost: @vhost, global: true, per_connection_delay: 0}
+    assert match?({:badrpc, :nodedown}, @command.run(["test"], opts))
+  end
+
+  test "banner for vhost option", context do
+    opts = %{node: node, vhost: "burrow", global: false, per_connection_delay: 0}
+    s = @command.banner(["some reason"], opts)
+    assert s =~ ~r/Closing all connections to vhost burrow/
+    assert s =~ ~r/some reason/
+  end
+
+  test "banner for global option", context do
+    opts = %{node: :test@localhost, vhost: "burrow", global: true, per_connection_delay: 0}
+    s = @command.banner(["some reason"], opts)
+    assert s =~ ~r/Closing all connections to node test@localhost/
+    assert s =~ ~r/some reason/
+  end
+
+  defp fetch_connections_vhosts(node, nodes) do
+      RpcStream.receive_list_items(node,
+                                   :rabbit_networking,
+                                   :emit_connection_info_all,
+                                   [nodes, [:vhost]],
+                                   :infinity,
+                                   [:vhost],
+                                   Kernel.length(nodes))
+      |> Enum.to_list
+  end
+
+end