Implement rotate_logs command
This commit is contained in:
		
							parent
							
								
									ce72d5409c
								
							
						
					
					
						commit
						17fff860c8
					
				|  | @ -0,0 +1,38 @@ | ||||||
|  | ## 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 RotateLogsCommand do | ||||||
|  |   @behaviour CommandBehaviour | ||||||
|  |   @flags [] | ||||||
|  | 
 | ||||||
|  |   def merge_defaults(args, opts), do: {args, opts} | ||||||
|  |   def validate([_|_] = args, _) when length(args) > 0, do: {:validation_failure, :too_many_args}   | ||||||
|  |   def validate([], _), do: :ok | ||||||
|  |   def switches(), do: [] | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  |   def run([], %{node: node_name}) do | ||||||
|  |     node_name | ||||||
|  |     |> Helpers.parse_node | ||||||
|  |     |> :rabbit_misc.rpc_call(:rabbit, :rotate_logs, []) | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   def usage, do: "rotate_logs" | ||||||
|  | 
 | ||||||
|  |   def flags, do: @flags | ||||||
|  | 
 | ||||||
|  |   def banner(_, %{node: node_name}), do: "Rotating logs for node #{node_name} ..." | ||||||
|  | end | ||||||
|  | @ -0,0 +1,57 @@ | ||||||
|  | ## 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 RotateLogsCommandTest do | ||||||
|  |   use ExUnit.Case, async: false | ||||||
|  |   import TestHelper | ||||||
|  | 
 | ||||||
|  |   @command RotateLogsCommand | ||||||
|  | 
 | ||||||
|  |   setup_all do | ||||||
|  |     :net_kernel.start([:rabbitmqctl, :shortnames]) | ||||||
|  |     :net_kernel.connect_node(get_rabbit_hostname) | ||||||
|  | 
 | ||||||
|  |     on_exit([], fn -> | ||||||
|  |       :erlang.disconnect_node(get_rabbit_hostname) | ||||||
|  |       :net_kernel.stop() | ||||||
|  |     end) | ||||||
|  | 
 | ||||||
|  |     :ok | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   setup do | ||||||
|  |     {:ok, opts: %{node: get_rabbit_hostname}} | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   test "validate: with extra arguments returns an arg count error", context do | ||||||
|  |     assert @command.validate(["extra"], context[:opts]) == {:validation_failure, :too_many_args} | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   test "run: request to a named, active node succeeds", context do | ||||||
|  |     assert @command.run([], context[:opts]) == :ok  | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   test "run: request to a non-existent node returns nodedown" do | ||||||
|  |     target = :jake@thedog | ||||||
|  |     :net_kernel.connect_node(target) | ||||||
|  |     opts = %{node: target} | ||||||
|  |     assert match?({:badrpc, :nodedown}, @command.run([], opts)) | ||||||
|  |   end | ||||||
|  | 
 | ||||||
|  |   test "banner", context do | ||||||
|  |     assert @command.banner([], context[:opts]) =~ ~r/Rotating logs for node #{get_rabbit_hostname}/ | ||||||
|  |   end | ||||||
|  | end | ||||||
		Loading…
	
		Reference in New Issue