Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Fetching contributors…

Cannot retrieve contributors at this time

98 lines (82 sloc) 5.478 kB
require "#{File.dirname(__FILE__)}/utils"
require 'capistrano/ssh'
class SSHTest < Test::Unit::TestCase
def setup
@options = { :username => nil,
:password => nil,
:port => 22,
: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, @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, @options).raises(Net::SSH::AuthenticationFailed)
Net::SSH.expects(:start).with(@server.host, @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, @options).raises(Net::SSH::AuthenticationFailed)
Net::SSH.expects(:start).with(@server.host, @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, @options.merge(:username => "jamis")).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, @options.merge(:username => "jamis")).raises(Net::SSH::AuthenticationFailed)
Net::SSH.expects(:start).with(@server.host, @options.merge(:username => "jamis", :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, @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, @options.merge(:username => "jamis")).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, @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, @options.merge(:username => "jamis", :port => 1235)).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, @options.merge(:username => "JamisMan", :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, @options.merge(:username => "jamis", :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, @options.merge(:username => "jamis", :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, @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, @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
Jump to Line
Something went wrong with that request. Please try again.