minestrone 0.0.1

data/test/command_test.rb

@@ -0,0 +1,305 @@
+require "utils"
+require 'minestrone/command'
+require 'minestrone/configuration'
+
+class CommandTest < Test::Unit::TestCase
+  class FakeChannel < Hash
+    def exec(*); end
+    def send_data(*); end
+    def eof!; end
+    def close; end
+    def request_pty(*); end
+    def on_data(*); end
+    def on_extended_data(*); end
+    def on_request(*); end
+    def on_close(*); end
+  end
+
+  def test_command_should_keep_session
+    session = mock_session
+    assert_equal session, Minestrone::Command.new("ls", session).session
+  end
+
+  def test_command_with_newlines_should_be_properly_escaped
+    cmd = Minestrone::Command.new("ls\necho", mock_session)
+    assert_equal "ls\\\necho", cmd.command
+  end
+
+  def test_command_with_crlf_newlines_should_be_properly_escaped
+    cmd = Minestrone::Command.new("ls\r\necho", mock_session)
+    assert_equal "ls\\\necho", cmd.command
+  end
+
+  def test_command_with_pty_should_request_pty_and_register_success_callback
+    session = setup_for_extracting_channel_action(:request_pty, true) do |ch|
+      ch.expects(:exec).with(%(sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :pty => true)
+  end
+
+  def test_command_with_env_key_should_have_environment_constructed_and_prepended
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:request_pty).never
+      ch.expects(:exec).with(%(env FOO=bar sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :env => { "FOO" => "bar" })
+  end
+
+  def test_env_with_symbolic_key_should_be_accepted_as_a_string
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(env FOO=bar sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :env => { :FOO => "bar" })
+  end
+
+  def test_env_as_string_should_be_substituted_in_directly
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(env HOWDY=there sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :env => "HOWDY=there")
+  end
+
+  def test_env_with_symbolic_value_should_be_accepted_as_string
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(env FOO=bar sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :env => { "FOO" => :bar })
+  end
+
+  def test_env_value_should_be_escaped
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(env FOO=(\\ \\\"bar\\\"\\ ) sh -c 'ls'))
+    end
+    open_test_channel("ls", session, :env => { "FOO" => '( "bar" )' })
+  end
+
+  def test_env_with_multiple_keys_should_chain_the_entries_together
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with do |command|
+        command =~ /^env / &&
+        command =~ /\ba=b\b/ &&
+        command =~ /\bc=d\b/ &&
+        command =~ /\be=f\b/ &&
+        command =~ / sh -c 'ls'$/
+      end
+    end
+    open_test_channel("ls", session, :env => { :a => :b, :c => :d, :e => :f })
+  end
+
+  def test_open_channel_should_set_host_key_on_channel
+    channel = nil
+    session = setup_for_extracting_channel_action { |ch| channel = ch }
+    open_test_channel("ls", session)
+    assert_equal "minestrone", channel[:host]
+  end
+
+  def test_open_channel_should_set_options_key_on_channel
+    channel = nil
+    session = setup_for_extracting_channel_action { |ch| channel = ch }
+    open_test_channel("ls", session, :data => "here we go")
+    assert_equal({ :data => 'here we go' }, channel[:options])
+  end
+
+  def test_successful_channel_should_send_command
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(sh -c 'ls'))
+    end
+    open_test_channel("ls", session)
+  end
+
+  def test_successful_channel_with_shell_option_should_send_command_via_specified_shell
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(/bin/bash -c 'ls'))
+    end
+    open_test_channel("ls", session, :shell => "/bin/bash")
+  end
+
+  def test_successful_channel_with_shell_false_should_send_command_without_shell
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(echo `hostname`))
+    end
+    open_test_channel("echo `hostname`", session, :shell => false)
+  end
+
+  def test_successful_channel_should_send_data_if_data_key_is_present
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(sh -c 'ls'))
+      ch.expects(:send_data).with("here we go")
+    end
+    open_test_channel("ls", session, :data => "here we go")
+  end
+
+  def test_unsuccessful_pty_request_should_close_channel
+    session = setup_for_extracting_channel_action(:request_pty, false) do |ch|
+      ch.expects(:close)
+    end
+    open_test_channel("ls", session, :pty => true)
+  end
+
+  def test_on_data_should_invoke_callback_as_stdout
+    session = setup_for_extracting_channel_action(:on_data, "hello")
+    called = false
+    open_test_channel("ls", session) do |ch, stream, data|
+      called = true
+      assert_equal :out, stream
+      assert_equal "hello", data
+    end
+    assert called
+  end
+
+  def test_on_extended_data_should_invoke_callback_as_stderr
+    session = setup_for_extracting_channel_action(:on_extended_data, 2, "hello")
+    called = false
+    open_test_channel("ls", session) do |ch, stream, data|
+      called = true
+      assert_equal :err, stream
+      assert_equal "hello", data
+    end
+    assert called
+  end
+
+  def test_on_request_should_record_exit_status
+    data = mock(:read_long => 5)
+    channel = nil
+    session = setup_for_extracting_channel_action([:on_request, "exit-status"], data) { |ch| channel = ch }
+    open_test_channel("ls", session)
+    assert_equal 5, channel[:status]
+  end
+
+  def test_on_request_should_log_exit_signal_if_logger_present
+    data = mock(:read_string => "TERM")
+    logger = stub_everything
+
+    session = setup_for_extracting_channel_action([:on_request, "exit-signal"], data)
+    logger.expects(:important).with("command received signal TERM", server("minestrone"))
+
+    open_test_channel("puppet", session, :logger => logger)
+  end
+
+  def test_on_close_should_set_channel_closed
+    channel = nil
+    session = setup_for_extracting_channel_action(:on_close) { |ch| channel = ch }
+    open_test_channel("ls", session)
+    assert channel[:closed]
+  end
+
+  def test_stop_should_close_open_channel
+    cmd = Minestrone::Command.new("ls", mock_session)
+    cmd.send(:open_channel, mock_session(new_channel(false)))
+    cmd.stop!
+  end
+
+  def test_process_should_return_cleanly_if_channel_has_zero_exit_status
+    cmd = Minestrone::Command.new("ls", mock_session(new_channel(true, 0)))
+    assert_nothing_raised { cmd.process! }
+  end
+
+  def test_process_should_raise_error_if_channel_has_non_zero_exit_status
+    cmd = Minestrone::Command.new("ls", mock_session(new_channel(true, 1)))
+    assert_raises(Minestrone::CommandError) { cmd.process! }
+  end
+
+  def test_command_error_should_include_accessor_with_host
+    cmd = Minestrone::Command.new("ls", mock_session(new_channel(true, 1)))
+
+    begin
+      cmd.process!
+      flunk "expected an exception to be raised"
+    rescue Minestrone::CommandError => e
+      assert e.respond_to?(:host)
+      assert_equal "minestrone", e.host.to_s
+    end
+  end
+
+  def test_process_should_loop_until_channel_is_closed
+    ch = mock("channel")
+    ch.stubs(:to_ary)
+    ch.stubs(:[]).with(:closed).returns(false, false, false, true)
+    ch.expects(:[]).with(:status).returns(0)
+    cmd = Minestrone::Command.new("ls", mock_session(ch))
+    assert_nothing_raised do
+      cmd.process!
+    end
+  end
+
+  def test_process_should_instantiate_command_and_process!
+    cmd = mock("command", :process! => nil)
+    session = mock_session
+    Minestrone::Command.expects(:new).with("ls -l", session, {:foo => "bar"}).returns(cmd)
+    Minestrone::Command.process("ls -l", session, :foo => "bar")
+  end
+
+  def test_process_with_host_placeholder_should_substitute_host_placeholder_with_each_host
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(sh -c 'echo minestrone'))
+    end
+    open_test_channel("echo $CAPISTRANO:HOST$", session)
+  end
+
+  def test_process_with_unknown_placeholder_should_not_replace_placeholder
+    session = setup_for_extracting_channel_action do |ch|
+      ch.expects(:exec).with(%(sh -c 'echo $CAPISTRANO:OTHER$'))
+    end
+    open_test_channel("echo $CAPISTRANO:OTHER$", session)
+  end
+
+  def test_input_stream_closed_when_eof_option_is_true
+    channel = nil
+    session = setup_for_extracting_channel_action { |ch| channel = ch }
+    channel.expects(:eof!)
+    open_test_channel("cat", session, :data => "here we go", :eof => true)
+    assert_equal({ :data => 'here we go', :eof => true }, channel[:options])
+  end
+
+  private
+
+  def mock_session(channel = nil)
+    stub('session',
+         :open_channel => channel,
+         :preprocess   => true,
+         :postprocess  => true,
+         :listeners    => {},
+         :xserver      => server("minestrone"))
+  end
+
+  class MockChannel < Hash
+    def close
+    end
+  end
+
+  def new_channel(closed, status = nil)
+    ch = MockChannel.new
+    ch.update({ :closed => closed, :host => "minestrone", :server => server("minestrone") })
+    ch[:status] = status if status
+    ch.expects(:close) unless closed
+    ch
+  end
+
+  def setup_for_extracting_channel_action(action = nil, *args)
+    s = server("minestrone")
+    session = mock("session", :xserver => s)
+
+    channel = FakeChannel.new
+    session.expects(:open_channel).yields(channel)
+
+    channel.stubs(:on_data)
+    channel.stubs(:on_extended_data)
+    channel.stubs(:on_request)
+    channel.stubs(:on_close)
+
+    if action
+      action = Array(action)
+      channel.expects(action.first).with(*action[1..-1]).yields(channel, *args)
+    end
+
+    yield channel if block_given?
+
+    session
+  end
+
+  def open_test_channel(command, session, options = {}, &block)
+    cmd = Minestrone::Command.new(command, session, options, &block)
+    cmd.send(:open_channel, session)
+    cmd
+  end
+end

data/test/configuration/actions/file_transfer_test.rb

@@ -0,0 +1,61 @@
+require "utils"
+require 'minestrone/configuration/actions/file_transfer'
+
+class ConfigurationActionsFileTransferTest < Test::Unit::TestCase
+  class MockConfig
+    include Minestrone::Configuration::Actions::FileTransfer
+    attr_accessor :session, :dry_run
+  end
+
+  def setup
+    @config = MockConfig.new
+    @config.stubs(:logger).returns(stub_everything)
+  end
+
+  def test_put_should_delegate_to_upload
+    @config.expects(:upload).with { |from, to, opts|
+      from.string == "some data" && to == "test.txt" && opts == { :mode => 0777 } }
+    @config.expects(:run).never
+    @config.put("some data", "test.txt", :mode => 0777)
+  end
+
+  def test_get_should_delegate_to_download
+    @config.expects(:download).with("testr.txt", "testl.txt", { :foo => "bar" })
+    @config.get("testr.txt", "testl.txt", :foo => "bar")
+  end
+
+  def test_upload_should_delegate_to_transfer
+    @config.expects(:transfer).with(:up, "testl.txt", "testr.txt", { :foo => "bar" })
+    @config.upload("testl.txt", "testr.txt", :foo => "bar")
+  end
+
+  def test_upload_without_mode_should_not_try_to_chmod
+    @config.expects(:transfer).with(:up, "testl.txt", "testr.txt", { :foo => "bar" })
+    @config.expects(:run).never
+    @config.upload("testl.txt", "testr.txt", :foo => "bar")
+  end
+
+  def test_upload_with_mode_should_try_to_chmod
+    @config.expects(:transfer).with(:up, "testl.txt", "testr.txt", { :foo => "bar" })
+    @config.expects(:run).with("chmod 775 testr.txt", {:foo => "bar"})
+    @config.upload("testl.txt", "testr.txt", :mode => 0775, :foo => "bar")
+  end
+
+  def test_upload_with_symbolic_mode_should_try_to_chmod
+    @config.expects(:transfer).with(:up, "testl.txt", "testr.txt", { :foo => "bar" })
+    @config.expects(:run).with("chmod g+w testr.txt", {:foo => "bar"})
+    @config.upload("testl.txt", "testr.txt", :mode => "g+w", :foo => "bar")
+  end
+
+  def test_download_should_delegate_to_transfer
+    @config.expects(:transfer).with(:down, "testr.txt", "testl.txt", { :foo => "bar" })
+    @config.download("testr.txt", "testl.txt", :foo => "bar")
+  end
+
+  def test_transfer_should_invoke_transfer_on_the_configured_server
+    @config.session = 1
+    @config.expects(:execute_on_server).yields
+    Minestrone::Transfer.expects(:process).with(:up, "testl.txt", "testr.txt", 1, {:foo => "bar", :logger => @config.logger})
+    @config.transfer(:up, "testl.txt", "testr.txt", :foo => "bar")
+  end
+end

data/test/configuration/actions/inspect_test.rb

@@ -0,0 +1,76 @@
+require "utils"
+require 'minestrone/configuration/actions/inspect'
+
+class ConfigurationActionsInspectTest < Test::Unit::TestCase
+  class MockConfig
+    include Minestrone::Configuration::Actions::Inspect
+  end
+
+  def setup
+    @config = MockConfig.new
+    @config.stubs(:logger).returns(stub_everything)
+    @config.stubs(:sudo).returns('sudo')
+  end
+
+  def test_stream_should_pass_options_through_to_run
+    @config.expects(:invoke_command).with("tail -f foo.log", { :once => true, :eof => true })
+    @config.stream("tail -f foo.log", :once => true)
+  end
+
+  def test_stream_with_sudo_should_avoid_closing_stdin
+    @config.expects(:invoke_command).with("sudo tail -f foo.log", { :once => true, :eof => false })
+    @config.stream("sudo tail -f foo.log", :once => true)
+  end
+
+  def test_stream_should_emit_stdout_via_puts
+    @config.expects(:invoke_command).yields(mock("channel"), :out, "something streamed")
+    @config.expects(:puts).with("something streamed")
+    @config.expects(:warn).never
+    @config.stream("tail -f foo.log")
+  end
+
+  def test_stream_should_emit_stderr_via_warn
+    ch = mock("channel")
+    ch.expects(:[]).with(:server).returns(server("minestrone"))
+    @config.expects(:invoke_command).yields(ch, :err, "something streamed")
+    @config.expects(:puts).never
+    @config.expects(:warn).with("[err :: minestrone] something streamed")
+    @config.stream("tail -f foo.log")
+  end
+
+  def test_capture_should_pass_options_merged_with_once_to_run
+    @config.expects(:invoke_command).with("hostname", { :foo => "bar", :once => true, :eof => true })
+    @config.capture("hostname", :foo => "bar")
+  end
+
+  def test_capture_with_sudo_should_avoid_closing_stdin
+    @config.expects(:invoke_command).with("sudo hostname", { :foo => "bar", :once => true, :eof => false })
+    @config.capture("sudo hostname", :foo => "bar")
+  end
+
+  def test_capture_with_stderr_should_emit_stderr_via_warn
+    ch = mock("channel")
+    ch.expects(:[]).with(:server).returns(server("minestrone"))
+    @config.expects(:invoke_command).yields(ch, :err, "boom")
+    @config.expects(:warn).with("[err :: minestrone] boom")
+    @config.capture("hostname")
+  end
+
+  def test_capture_with_stdout_should_aggregate_and_return_stdout
+    config_expects_invoke_command_to_loop_with(mock("channel"), "foo", "bar", "baz")
+    assert_equal "foobarbaz", @config.capture("hostname")
+  end
+
+  private
+
+  def config_expects_invoke_command_to_loop_with(channel, *output)
+    class <<@config
+      attr_accessor :script, :channel
+      def invoke_command(*args)
+        script.each { |item| yield channel, :out, item }
+      end
+    end
+    @config.channel = channel
+    @config.script = output
+  end
+end

data/test/configuration/actions/invocation_test.rb

@@ -0,0 +1,258 @@
+require "utils"
+require 'minestrone/configuration/actions/invocation'
+require 'minestrone/configuration/actions/file_transfer'
+
+class ConfigurationActionsInvocationTest < Test::Unit::TestCase
+  class MockConfig
+    attr_reader :options
+    attr_accessor :debug
+    attr_accessor :dry_run
+    attr_accessor :server
+    attr_accessor :session
+
+    def initialize
+      @options = {}
+      @server = nil
+      initialize_invocation
+    end
+
+    def [](*args)
+      @options[*args]
+    end
+
+    def set(name, value)
+      @options[name] = value
+    end
+
+    def fetch(*args)
+      @options.fetch(*args)
+    end
+
+    def active_server
+      @server
+    end
+
+    def execute_on_server
+      yield
+    end
+
+    include Minestrone::Configuration::Actions::Invocation
+    include Minestrone::Configuration::Actions::FileTransfer
+  end
+
+  def setup
+    @config = make_config
+    @original_io_proc = MockConfig.default_io_proc
+  end
+
+  def teardown
+    MockConfig.default_io_proc = @original_io_proc
+  end
+
+  def test_run_options_should_be_passed_to_command_process
+    ::Minestrone::Command.expects(:process).with("ls", @config.session, has_entries(:foo => "bar", :eof => true, :logger => @config.logger, :configuration => @config))
+    @config.run "ls", :foo => "bar"
+  end
+
+  def test_run_will_return_if_dry_run
+    @config.expects(:dry_run).returns(true)
+    @config.expects(:execute_on_server).never
+    @config.run "ls", :foo => "bar"
+  end
+
+  def test_put_wont_transfer_if_dry_run
+    config = make_config
+    config.dry_run = true
+    config.server = "foo"
+    config.expects(:execute_on_server).never
+    ::Minestrone::Transfer.expects(:process).never
+    config.put "foo", "bar", :mode => 0644
+  end
+
+  def test_add_default_command_options_should_return_bare_options_if_there_is_no_env_or_shell_specified
+    assert_equal({:foo => "bar"}, @config.add_default_command_options(:foo => "bar"))
+  end
+
+  def test_add_default_command_options_should_merge_default_environment_as_env
+    @config[:default_environment][:bang] = "baz"
+    assert_equal({:foo => "bar", :env => { :bang => "baz" }}, @config.add_default_command_options(:foo => "bar"))
+  end
+
+  def test_add_default_command_options_should_merge_env_with_default_environment
+    @config[:default_environment][:bang] = "baz"
+    @config[:default_environment][:bacon] = "crunchy"
+    assert_equal({:foo => "bar", :env => { :bang => "baz", :bacon => "chunky", :flip => "flop" }}, @config.add_default_command_options(:foo => "bar", :env => {:bacon => "chunky", :flip => "flop"}))
+  end
+
+  def test_add_default_command_options_should_use_default_shell_if_present
+    @config.set :default_shell, "/bin/bash"
+    assert_equal({:foo => "bar", :shell => "/bin/bash"}, @config.add_default_command_options(:foo => "bar"))
+  end
+
+  def test_add_default_command_options_should_use_default_shell_of_false_if_present
+    @config.set :default_shell, false
+    assert_equal({:foo => "bar", :shell => false}, @config.add_default_command_options(:foo => "bar"))
+  end
+
+  def test_add_default_command_options_should_use_shell_in_preference_of_default_shell
+    @config.set :default_shell, "/bin/bash"
+    assert_equal({:foo => "bar", :shell => "/bin/sh"}, @config.add_default_command_options(:foo => "bar", :shell => "/bin/sh"))
+  end
+
+  def test_default_io_proc_should_log_stdout_arguments_as_info
+    ch = { :host => "minestrone",
+           :server => server("minestrone"),
+           :options => { :logger => mock("logger") } }
+    ch[:options][:logger].expects(:info).with("data stuff", "out :: minestrone")
+    MockConfig.default_io_proc[ch, :out, "data stuff"]
+  end
+
+  def test_default_io_proc_should_log_stderr_arguments_as_important
+    ch = { :host => "minestrone",
+           :server => server("minestrone"),
+           :options => { :logger => mock("logger") } }
+    ch[:options][:logger].expects(:important).with("data stuff", "err :: minestrone")
+    MockConfig.default_io_proc[ch, :err, "data stuff"]
+  end
+
+  def test_sudo_should_default_to_sudo
+    @config.expects(:run).with("sudo -p 'sudo password: ' ls", {})
+    @config.sudo "ls"
+  end
+
+  def test_sudo_should_keep_input_stream_open
+    @config.expects(:execute_on_server)
+    @config.sudo "ls", :foo => "bar"
+  end
+
+  def test_sudo_should_use_sudo_variable_definition
+    @config.expects(:run).with("/opt/local/bin/sudo -p 'sudo password: ' ls", {})
+    @config.options[:sudo] = "/opt/local/bin/sudo"
+    @config.sudo "ls"
+  end
+
+  def test_sudo_should_interpret_as_option_as_user
+    @config.expects(:run).with("sudo -p 'sudo password: ' -u app ls", {})
+    @config.sudo "ls", :as => "app"
+  end
+
+  def test_sudo_should_pass_options_through_to_run
+    @config.expects(:run).with("sudo -p 'sudo password: ' ls", { :foo => "bar" })
+    @config.sudo "ls", :foo => "bar"
+  end
+
+  def test_sudo_should_avoid_minus_p_when_sudo_prompt_is_empty
+    @config.set :sudo_prompt, ""
+    @config.expects(:run).with("sudo ls", {})
+    @config.sudo "ls"
+  end
+
+  def test_sudo_should_interpret_sudo_prompt_variable_as_custom_prompt
+    @config.set :sudo_prompt, "give it to me: "
+    @config.expects(:run).with("sudo -p 'give it to me: ' ls", {})
+    @config.sudo "ls"
+  end
+
+  def test_sudo_behavior_callback_should_send_password_when_prompted_with_default_sudo_prompt
+    ch = mock("channel")
+    ch.expects(:send_data).with("g00b3r\n")
+    @config.options[:password] = "g00b3r"
+    @config.sudo_behavior_callback(nil)[ch, nil, "sudo password: "]
+  end
+
+  def test_sudo_behavior_callback_should_send_password_when_prompted_with_custom_sudo_prompt
+    ch = mock("channel")
+    ch.expects(:send_data).with("g00b3r\n")
+    @config.set :sudo_prompt, "give it to me: "
+    @config.options[:password] = "g00b3r"
+    @config.sudo_behavior_callback(nil)[ch, nil, "give it to me: "]
+  end
+
+  def test_sudo_behavior_callback_with_incorrect_password_on_first_prompt
+    ch = mock("channel")
+    ch.stubs(:[]).with(:host).returns("minestrone")
+    ch.stubs(:[]).with(:server).returns(server("minestrone"))
+    @config.expects(:reset!).with(:password)
+    @config.sudo_behavior_callback(nil)[ch, nil, "Sorry, try again."]
+  end
+
+  def test_sudo_behavior_callback_with_incorrect_password_on_subsequent_prompts
+    callback = @config.sudo_behavior_callback(nil)
+
+    ch = mock("channel")
+    ch.stubs(:[]).with(:host).returns("minestrone")
+    ch.stubs(:[]).with(:server).returns(server("minestrone"))
+    ch2 = mock("channel")
+    ch2.stubs(:[]).with(:host).returns("cap2")
+    ch2.stubs(:[]).with(:server).returns(server("cap2"))
+
+    @config.expects(:reset!).with(:password).times(2)
+
+    callback[ch, nil, "Sorry, try again."]
+    callback[ch2, nil, "Sorry, try again."] # shouldn't call reset!
+    callback[ch, nil, "Sorry, try again."]
+  end
+
+  def test_sudo_behavior_callback_should_reset_password_and_prompt_again_if_output_includes_both_cues
+    ch = mock("channel")
+    ch.stubs(:[]).with(:host).returns("minestrone")
+    ch.stubs(:[]).with(:server).returns(server("minestrone"))
+    ch.expects(:send_data, "password!\n").times(2)
+
+    @config.set(:password, "password!")
+    @config.expects(:reset!).with(:password)
+
+    callback = @config.sudo_behavior_callback(nil)
+    callback[ch, :out, "sudo password: "]
+    callback[ch, :out, "Sorry, try again.\nsudo password: "]
+  end
+
+  def test_sudo_behavior_callback_should_defer_to_fallback_for_other_output
+    callback = @config.sudo_behavior_callback(inspectable_proc)
+
+    a = mock("channel", :called => true)
+    b = mock("stream", :called => true)
+    c = mock("data", :called => true)
+
+    callback[a, b, c]
+  end
+
+  def test_invoke_command_should_default_to_run
+    @config.expects(:run).with("ls", { :once => true })
+    @config.invoke_command("ls", :once => true)
+  end
+
+  def test_invoke_command_should_delegate_to_method_identified_by_via
+    @config.expects(:foobar).with("ls", { :once => true })
+    @config.invoke_command("ls", :once => true, :via => :foobar)
+  end
+
+  def test_run_only_logs_once
+    @config.server = :app
+
+    logger = mock('logger')
+    logger.stubs(:debug).with("executing \"ls\"")
+    @config.stubs(:logger).returns(logger)
+
+    @config.expects(:execute_on_server)
+
+    @config.run("ls")
+  end
+
+  private
+
+  def make_config
+    config = MockConfig.new
+    config.stubs(:logger).returns(stub_everything)
+    config
+  end
+
+  def inspectable_proc
+    Proc.new do |ch, stream, data|
+      ch.called
+      stream.called
+      data.called
+    end
+  end
+
+end