fotonauts-capistrano 2.5.2

data/test/role_test.rb

@@ -0,0 +1,11 @@
+require "utils"
+require 'capistrano/role'
+
+class RoleTest < Test::Unit::TestCase
+  def test_clearing_a_populated_role_should_yield_no_servers
+    role = Capistrano::Role.new("app1.capistrano.test", lambda { |o| "app2.capistrano.test" })
+    assert_equal 2, role.servers.size
+    role.clear
+    assert role.servers.empty?
+  end
+end

data/test/server_definition_test.rb

@@ -0,0 +1,121 @@
+require "utils"
+require 'capistrano/server_definition'
+
+class ServerDefinitionTest < Test::Unit::TestCase
+  def test_new_without_credentials_or_port_should_set_values_to_defaults
+    server = Capistrano::ServerDefinition.new("www.capistrano.test")
+    assert_equal "www.capistrano.test", server.host
+    assert_nil   server.user
+    assert_nil   server.port
+  end
+
+  def test_new_with_encoded_user_should_extract_user_and_use_default_port
+    server = Capistrano::ServerDefinition.new("jamis@www.capistrano.test")
+    assert_equal "www.capistrano.test", server.host
+    assert_equal "jamis", server.user
+    assert_nil   server.port
+  end
+
+  def test_new_with_encoded_port_should_extract_port_and_use_default_user
+    server = Capistrano::ServerDefinition.new("www.capistrano.test:8080")
+    assert_equal "www.capistrano.test", server.host
+    assert_nil   server.user
+    assert_equal 8080, server.port
+  end
+
+  def test_new_with_encoded_user_and_port_should_extract_user_and_port
+    server = Capistrano::ServerDefinition.new("jamis@www.capistrano.test:8080")
+    assert_equal "www.capistrano.test", server.host
+    assert_equal "jamis", server.user
+    assert_equal 8080, server.port
+  end
+
+  def test_new_with_user_as_option_should_use_given_user
+    server = Capistrano::ServerDefinition.new("www.capistrano.test", :user => "jamis")
+    assert_equal "www.capistrano.test", server.host
+    assert_equal "jamis", server.user
+    assert_nil   server.port
+  end
+
+  def test_new_with_port_as_option_should_use_given_user
+    server = Capistrano::ServerDefinition.new("www.capistrano.test", :port => 8080)
+    assert_equal "www.capistrano.test", server.host
+    assert_nil   server.user
+    assert_equal 8080, server.port
+  end
+
+  def test_encoded_value_should_override_hash_option
+    server = Capistrano::ServerDefinition.new("jamis@www.capistrano.test:8080", :user => "david", :port => 8081)
+    assert_equal "www.capistrano.test", server.host
+    assert_equal "jamis", server.user
+    assert_equal 8080, server.port
+    assert server.options.empty?
+  end
+
+  def test_new_with_option_should_dup_option_hash
+    options = {}
+    server = Capistrano::ServerDefinition.new("www.capistrano.test", options)
+    assert_not_equal options.object_id, server.options.object_id
+  end
+
+  def test_new_with_options_should_keep_options
+    server = Capistrano::ServerDefinition.new("www.capistrano.test", :primary => true)
+    assert_equal true, server.options[:primary]
+  end
+  
+  def test_default_user_should_try_to_guess_username
+    ENV.stubs(:[]).returns(nil)
+    assert_equal "not-specified", Capistrano::ServerDefinition.default_user
+    
+    ENV.stubs(:[]).returns(nil)
+    ENV.stubs(:[]).with("USERNAME").returns("ryan")
+    assert_equal "ryan", Capistrano::ServerDefinition.default_user
+
+    ENV.stubs(:[]).returns(nil)
+    ENV.stubs(:[]).with("USER").returns("jamis")
+    assert_equal "jamis", Capistrano::ServerDefinition.default_user
+  end
+
+  def test_comparison_should_match_when_host_user_port_are_same
+    s1 = server("jamis@www.capistrano.test:8080")
+    s2 = server("www.capistrano.test", :user => "jamis", :port => 8080)
+    assert_equal s1, s2
+    assert_equal s1.hash, s2.hash
+    assert s1.eql?(s2)
+  end
+
+  def test_servers_should_be_comparable
+    s1 = server("jamis@www.capistrano.test:8080")
+    s2 = server("www.alphabet.test:1234")
+    s3 = server("jamis@www.capistrano.test:8075")
+    s4 = server("billy@www.capistrano.test:8080")
+
+    assert s2 < s1
+    assert s3 < s1
+    assert s4 < s1
+    assert s2 < s3
+    assert s2 < s4
+    assert s3 < s4
+  end
+
+  def test_comparison_should_not_match_when_any_of_host_user_port_differ
+    s1 = server("jamis@www.capistrano.test:8080")
+    s2 = server("bob@www.capistrano.test:8080")
+    s3 = server("jamis@www.capistrano.test:8081")
+    s4 = server("jamis@app.capistrano.test:8080")
+    assert_not_equal s1, s2
+    assert_not_equal s1, s3
+    assert_not_equal s1, s4
+    assert_not_equal s2, s3
+    assert_not_equal s2, s4
+    assert_not_equal s3, s4
+  end
+
+  def test_to_s
+    assert_equal "www.capistrano.test", server("www.capistrano.test").to_s
+    assert_equal "www.capistrano.test", server("www.capistrano.test:22").to_s
+    assert_equal "www.capistrano.test:1234", server("www.capistrano.test:1234").to_s
+    assert_equal "jamis@www.capistrano.test", server("jamis@www.capistrano.test").to_s
+    assert_equal "jamis@www.capistrano.test:1234", server("jamis@www.capistrano.test:1234").to_s
+  end
+end

data/test/shell_test.rb

@@ -0,0 +1,90 @@
+require "utils"
+require 'capistrano/configuration'
+require 'capistrano/shell'
+
+class ShellTest < Test::Unit::TestCase
+  def setup
+    @config = Capistrano::Configuration.new
+    @shell  = Capistrano::Shell.new(@config)
+    @shell.stubs(:puts)
+  end
+
+  def test_readline_fallback_prompt_should_write_to_stdout_and_read_from_stdin
+    STDOUT.expects(:print).with("prompt> ")
+    STDOUT.expects(:flush)
+    STDIN.expects(:gets).returns("hi\n")
+    assert_equal "hi\n", Capistrano::Shell::ReadlineFallback.readline("prompt> ")
+  end
+
+  def test_question_mark_as_input_should_trigger_help
+    @shell.expects(:read_line).returns("?")
+    @shell.expects(:help)
+    assert @shell.read_and_execute
+  end
+
+  def test_help_as_input_should_trigger_help
+    @shell.expects(:read_line).returns("help")
+    @shell.expects(:help)
+    assert @shell.read_and_execute
+  end
+
+  def test_quit_as_input_should_cause_read_and_execute_to_return_false
+    @shell.expects(:read_line).returns("quit")
+    assert !@shell.read_and_execute
+  end
+
+  def test_exit_as_input_should_cause_read_and_execute_to_return_false
+    @shell.expects(:read_line).returns("exit")
+    assert !@shell.read_and_execute
+  end
+
+  def test_set_should_parse_flag_and_value_and_call_set_option
+    @shell.expects(:read_line).returns("set -v 5")
+    @shell.expects(:set_option).with("v", "5")
+    assert @shell.read_and_execute
+  end
+
+  def test_text_without_with_or_on_gets_processed_verbatim
+    @shell.expects(:read_line).returns("hello world")
+    @shell.expects(:process_command).with(nil, nil, "hello world")
+    assert @shell.read_and_execute
+  end
+
+  def test_text_with_with_gets_processed_with_with # lol
+    @shell.expects(:read_line).returns("with app,db hello world")
+    @shell.expects(:process_command).with("with", "app,db", "hello world")
+    assert @shell.read_and_execute
+  end
+
+  def test_text_with_on_gets_processed_with_on
+    @shell.expects(:read_line).returns("on app,db hello world")
+    @shell.expects(:process_command).with("on", "app,db", "hello world")
+    assert @shell.read_and_execute
+  end
+  
+  def test_task_command_with_bang_gets_processed_by_exec_tasks
+    while_testing_post_exec_commands do
+      @shell.expects(:read_line).returns("!deploy")
+      @shell.expects(:exec_tasks).with(["deploy"])
+      assert @shell.read_and_execute
+    end
+  end
+  
+  def test_normal_command_gets_processed_by_exec_command
+    while_testing_post_exec_commands do
+      @shell.expects(:read_line).returns("uptime")
+      @shell.expects(:exec_command).with("uptime",nil)
+      @shell.expects(:connect)
+      assert @shell.read_and_execute
+    end
+  end
+  
+  
+  private
+  
+  def while_testing_post_exec_commands(&block)
+    @shell.instance_variable_set(:@mutex,Mutex.new)
+    yield
+  end
+  
+end

data/test/ssh_test.rb

@@ -0,0 +1,102 @@
+require "utils"
+require 'capistrano/ssh'
+
+class SSHTest < Test::Unit::TestCase
+  def setup
+    Capistrano::ServerDefinition.stubs(:default_user).returns("default-user")
+    @options = { :password => nil,
+                 :auth_methods => %w(publickey hostbased) }
+    @server = server("capistrano")
+  end
+
+  def test_connect_with_bare_server_without_options_or_config_with_public_key_succeeding_should_only_loop_once
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server)
+  end
+
+  def test_connect_with_bare_server_without_options_with_public_key_failing_should_try_password
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options).raises(Net::SSH::AuthenticationFailed)
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options.merge(:password => "f4b13n", :auth_methods => %w(password keyboard-interactive))).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, :password => "f4b13n")
+  end
+
+  def test_connect_with_bare_server_without_options_public_key_and_password_failing_should_raise_error
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options).raises(Net::SSH::AuthenticationFailed)
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options.merge(:password => "f4b13n", :auth_methods => %w(password keyboard-interactive))).raises(Net::SSH::AuthenticationFailed)
+    assert_raises(Net::SSH::AuthenticationFailed) do
+      Capistrano::SSH.connect(@server, :password => "f4b13n")
+    end
+  end
+
+  def test_connect_with_bare_server_and_user_via_public_key_should_pass_user_to_net_ssh
+    Net::SSH.expects(:start).with(@server.host, "jamis", @options).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, :user => "jamis")
+  end
+
+  def test_connect_with_bare_server_and_user_via_password_should_pass_user_to_net_ssh
+    Net::SSH.expects(:start).with(@server.host, "jamis", @options).raises(Net::SSH::AuthenticationFailed)
+    Net::SSH.expects(:start).with(@server.host, "jamis", @options.merge(:password => "f4b13n", :auth_methods => %w(password keyboard-interactive))).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, :user => "jamis", :password => "f4b13n")
+  end
+
+  def test_connect_with_bare_server_with_explicit_port_should_pass_port_to_net_ssh
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options.merge(:port => 1234)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, :port => 1234)
+  end
+
+  def test_connect_with_server_with_user_should_pass_user_to_net_ssh
+    server = server("jamis@capistrano")
+    Net::SSH.expects(:start).with(server.host, "jamis", @options).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(server)
+  end
+
+  def test_connect_with_server_with_port_should_pass_port_to_net_ssh
+    server = server("capistrano:1235")
+    Net::SSH.expects(:start).with(server.host, "default-user", @options.merge(:port => 1235)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(server)
+  end
+
+  def test_connect_with_server_with_user_and_port_should_pass_user_and_port_to_net_ssh
+    server = server("jamis@capistrano:1235")
+    Net::SSH.expects(:start).with(server.host, "jamis", @options.merge(:port => 1235)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(server)
+  end
+
+  def test_connect_with_server_with_other_ssh_options_should_pass_ssh_options_to_net_ssh
+    server = server("jamis@capistrano:1235", :ssh_options => { :keys => %w(some_valid_key), :auth_methods => %w(a_method), :hmac => 'none' })
+    Net::SSH.expects(:start).with(server.host, "jamis", @options.merge(:port => 1235, :keys => %w(some_valid_key), :auth_methods => %w(a_method), :hmac => 'none' )).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(server)
+  end
+
+  def test_connect_with_ssh_options_should_use_ssh_options
+    ssh_options = { :username => "JamisMan", :port => 8125 }
+    Net::SSH.expects(:start).with(@server.host, "JamisMan", @options.merge(:port => 8125)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, {:ssh_options => ssh_options})
+  end
+
+  def test_connect_with_options_and_ssh_options_should_see_options_override_ssh_options
+    ssh_options = { :username => "JamisMan", :port => 8125, :forward_agent => true }
+    Net::SSH.expects(:start).with(@server.host, "jamis", @options.merge(:port => 1235, :forward_agent => true)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server, :ssh_options => ssh_options, :user => "jamis", :port => 1235)
+  end
+
+  def test_connect_with_ssh_options_should_see_server_options_override_ssh_options
+    ssh_options = { :username => "JamisMan", :port => 8125, :forward_agent => true }
+    server = server("jamis@capistrano:1235")
+    Net::SSH.expects(:start).with(server.host, "jamis", @options.merge(:port => 1235, :forward_agent => true)).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(server, {:ssh_options => ssh_options})
+  end
+
+  def test_connect_should_add_xserver_accessor_to_connection
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options).returns(success = Object.new)
+    assert_equal success, Capistrano::SSH.connect(@server)
+    assert success.respond_to?(:xserver)
+    assert success.respond_to?(:xserver)
+    assert_equal success.xserver, @server
+  end
+
+  def test_connect_should_not_retry_if_custom_auth_methods_are_given
+    Net::SSH.expects(:start).with(@server.host, "default-user", @options.merge(:auth_methods => %w(publickey))).raises(Net::SSH::AuthenticationFailed)
+    assert_raises(Net::SSH::AuthenticationFailed) { Capistrano::SSH.connect(@server, :ssh_options => { :auth_methods => %w(publickey) }) }
+  end
+end

data/test/task_definition_test.rb

@@ -0,0 +1,101 @@
+require "utils"
+require 'capistrano/task_definition'
+
+class TaskDefinitionTest < Test::Unit::TestCase
+  def setup
+    @namespace = namespace
+  end
+
+  def test_fqn_at_top_level_should_be_task_name
+    task = new_task(:testing)
+    assert_equal "testing", task.fully_qualified_name
+  end
+
+  def test_fqn_in_namespace_should_include_namespace_fqn
+    ns = namespace("outer:inner")
+    task = new_task(:testing, ns)
+    assert_equal "outer:inner:testing", task.fully_qualified_name
+  end
+
+  def test_fqn_at_top_level_when_default_should_be_default
+    task = new_task(:default)
+    assert_equal "default", task.fully_qualified_name
+  end
+
+  def test_fqn_in_namespace_when_default_should_be_namespace_fqn
+    ns = namespace("outer:inner")
+    task = new_task(:default, ns)
+    ns.stubs(:default_task => task)
+    assert_equal "outer:inner", task.fully_qualified_name
+  end
+
+  def test_task_should_require_block
+    assert_raises(ArgumentError) do
+      Capistrano::TaskDefinition.new(:testing, @namespace)
+    end
+  end
+
+  def test_description_should_return_empty_string_if_not_given
+    assert_equal "", new_task(:testing).description
+  end
+
+  def test_description_should_return_desc_attribute
+    assert_equal "something", new_task(:testing, @namespace, :desc => "something").description
+  end
+
+  def test_description_should_strip_leading_and_trailing_whitespace
+    assert_equal "something", new_task(:testing, @namespace, :desc => "   something   ").description
+  end
+
+  def test_description_should_normalize_newlines
+    assert_equal "a\nb\nc", new_task(:testing, @namespace, :desc => "a\nb\r\nc").description
+  end
+
+  def test_description_should_detect_and_remove_indentation
+    desc = <<-DESC
+      Here is some indented text \
+      and I want all of this to \
+      run together on a single line, \
+      without any extraneous spaces.
+
+        additional indentation will
+        be preserved.
+    DESC
+
+    task = new_task(:testing, @namespace, :desc => desc)
+    assert_equal "Here is some indented text and I want all of this to run together on a single line, without any extraneous spaces.\n\n  additional indentation will\n  be preserved.", task.description
+  end
+
+  def test_description_munging_should_be_sensitive_to_code_blocks
+    desc = <<-DESC
+      Here is a line \
+      wrapped      with spacing in it.
+
+        foo         bar
+        baz         bang
+    DESC
+
+    task = new_task(:testing, @namespace, :desc => desc)
+    assert_equal "Here is a line wrapped with spacing in it.\n\n  foo         bar\n  baz         bang", task.description
+  end
+
+  def test_brief_description_should_return_first_sentence_in_description
+    desc = "This is the task. It does all kinds of things."
+    task = new_task(:testing, @namespace, :desc => desc)
+    assert_equal "This is the task.", task.brief_description
+  end
+
+  def test_brief_description_should_truncate_if_length_given
+    desc = "This is the task that does all kinds of things. And then some."
+    task = new_task(:testing, @namespace, :desc => desc)
+    assert_equal "This is the task ...", task.brief_description(20)
+  end
+
+  def test_brief_description_should_not_break_at_period_in_middle_of_sentence
+    task = new_task(:testing, @namespace, :desc => "Take file.txt and copy it.")
+    assert_equal "Take file.txt and copy it.", task.brief_description
+
+    task = new_task(:testing, @namespace, :desc => "Take file.txt and copy it. Then do something else.")
+    assert_equal "Take file.txt and copy it.", task.brief_description
+  end
+end

data/test/transfer_test.rb

@@ -0,0 +1,160 @@
+require 'utils'
+require 'capistrano/transfer'
+
+class TransferTest < Test::Unit::TestCase
+  def test_class_process_should_delegate_to_instance_process
+    Capistrano::Transfer.expects(:new).with(:up, "from", "to", %w(a b c), {}).returns(mock('transfer', :process! => nil)).yields
+    yielded = false
+    Capistrano::Transfer.process(:up, "from", "to", %w(a b c), {}) { yielded = true }
+    assert yielded
+  end
+
+  def test_default_transport_is_sftp
+    transfer = Capistrano::Transfer.new(:up, "from", "to", [])
+    assert_equal :sftp, transfer.transport
+  end
+
+  def test_active_is_true_when_any_sftp_transfers_are_active
+    returns = [false, false, true]
+    sessions = [session('app1', :sftp), session('app2', :sftp), session('app3', :sftp)].each { |s| s.xsftp.expects(:upload).returns(stub('operation', :active? => returns.shift)) }
+    transfer = Capistrano::Transfer.new(:up, "from", "to", sessions, :via => :sftp)
+    assert_equal true, transfer.active?
+  end
+
+  def test_active_is_false_when_all_sftp_transfers_are_not_active
+    sessions = [session('app1', :sftp), session('app2', :sftp)].each { |s| s.xsftp.expects(:upload).returns(stub('operation', :active? => false)) }
+    transfer = Capistrano::Transfer.new(:up, "from", "to", sessions, :via => :sftp)
+    assert_equal false, transfer.active?
+  end
+
+  def test_active_is_true_when_any_scp_transfers_are_active
+    returns = [false, false, true]
+    sessions = [session('app1', :scp), session('app2', :scp), session('app3', :scp)].each do |s|
+      channel = stub('channel', :[]= => nil, :active? => returns.shift)
+      s.scp.expects(:upload).returns(channel)
+    end
+    transfer = Capistrano::Transfer.new(:up, "from", "to", sessions, :via => :scp)
+    assert_equal true, transfer.active?
+  end
+
+  def test_active_is_false_when_all_scp_transfers_are_not_active
+    sessions = [session('app1', :scp), session('app2', :scp), session('app3', :scp)].each do |s|
+      channel = stub('channel', :[]= => nil, :active? => false)
+      s.scp.expects(:upload).returns(channel)
+    end
+    transfer = Capistrano::Transfer.new(:up, "from", "to", sessions, :via => :scp)
+    assert_equal false, transfer.active?
+  end
+
+  [:up, :down].each do |direction|
+    define_method("test_sftp_#{direction}load_from_file_to_file_should_normalize_from_and_to") do
+      sessions = [session('app1', :sftp), session('app2', :sftp)]
+
+      sessions.each do |session|
+        session.xsftp.expects("#{direction}load".to_sym).with("from-#{session.xserver.host}", "to-#{session.xserver.host}",
+          :properties => { :server => session.xserver, :host => session.xserver.host })
+      end
+
+      transfer = Capistrano::Transfer.new(direction, "from-$CAPISTRANO:HOST$", "to-$CAPISTRANO:HOST$", sessions)
+    end
+
+    define_method("test_scp_#{direction}load_from_file_to_file_should_normalize_from_and_to") do
+      sessions = [session('app1', :scp), session('app2', :scp)]
+
+      sessions.each do |session|
+        session.scp.expects("#{direction}load".to_sym).returns({}).with("from-#{session.xserver.host}", "to-#{session.xserver.host}", :via => :scp)
+      end
+
+      transfer = Capistrano::Transfer.new(direction, "from-$CAPISTRANO:HOST$", "to-$CAPISTRANO:HOST$", sessions, :via => :scp)
+    end
+  end
+
+  def test_sftp_upload_from_IO_to_file_should_clone_the_IO_for_each_connection
+    sessions = [session('app1', :sftp), session('app2', :sftp)]
+    io = StringIO.new("from here")
+
+    sessions.each do |session|
+      session.xsftp.expects(:upload).with do |from, to, opts|
+        from != io && from.is_a?(StringIO) && from.string == io.string &&
+        to == "/to/here-#{session.xserver.host}" &&
+        opts[:properties][:server] == session.xserver &&
+        opts[:properties][:host] == session.xserver.host
+      end
+    end
+
+    transfer = Capistrano::Transfer.new(:up, StringIO.new("from here"), "/to/here-$CAPISTRANO:HOST$", sessions)
+  end
+
+  def test_scp_upload_from_IO_to_file_should_clone_the_IO_for_each_connection
+    sessions = [session('app1', :scp), session('app2', :scp)]
+    io = StringIO.new("from here")
+
+    sessions.each do |session|
+      channel = mock('channel')
+      channel.expects(:[]=).with(:server, session.xserver)
+      channel.expects(:[]=).with(:host, session.xserver.host)
+      session.scp.expects(:upload).returns(channel).with do |from, to, opts|
+        from != io && from.is_a?(StringIO) && from.string == io.string &&
+        to == "/to/here-#{session.xserver.host}"
+      end
+    end
+
+    transfer = Capistrano::Transfer.new(:up, StringIO.new("from here"), "/to/here-$CAPISTRANO:HOST$", sessions, :via => :scp)
+  end
+
+  def test_process_should_block_until_transfer_is_no_longer_active
+    transfer = Capistrano::Transfer.new(:up, "from", "to", [])
+    transfer.expects(:process_iteration).times(4).yields.returns(true,true,true,false)
+    transfer.expects(:active?).times(4)
+    transfer.process!
+  end
+
+  def test_errors_raised_for_a_sftp_session_should_abort_session_and_continue_with_remaining_sessions
+    s = session('app1')
+    error = ExceptionWithSession.new(s)
+    transfer = Capistrano::Transfer.new(:up, "from", "to", [])
+    transfer.expects(:process_iteration).raises(error).times(3).returns(true, false)
+    txfr = mock('transfer', :abort! => true)
+    txfr.expects(:[]=).with(:failed, true)
+    txfr.expects(:[]=).with(:error, error)
+    transfer.expects(:session_map).returns(s => txfr)
+    transfer.process!
+  end
+
+  def test_errors_raised_for_a_scp_session_should_abort_session_and_continue_with_remaining_sessions
+    s = session('app1')
+    error = ExceptionWithSession.new(s)
+    transfer = Capistrano::Transfer.new(:up, "from", "to", [], :via => :scp)
+    transfer.expects(:process_iteration).raises(error).times(3).returns(true, false)
+    txfr = mock('channel', :close => true)
+    txfr.expects(:[]=).with(:failed, true)
+    txfr.expects(:[]=).with(:error, error)
+    transfer.expects(:session_map).returns(s => txfr)
+    transfer.process!
+  end
+
+  private
+
+    class ExceptionWithSession < ::Exception
+      attr_reader :session
+
+      def initialize(session)
+        @session = session
+        super()
+      end
+    end
+
+    def session(host, mode=nil)
+      session = stub('session', :xserver => stub('server', :host => host))
+      case mode
+      when :sftp
+        sftp = stub('sftp')
+        session.expects(:sftp).with(false).returns(sftp)
+        sftp.expects(:connect).yields(sftp).returns(sftp)
+        session.stubs(:xsftp).returns(sftp)
+      when :scp
+        session.stubs(:scp).returns(stub('scp'))
+      end
+      session
+    end
+end