net-ssh 1.1.4 → 2.0.0

data/test/userauth/tc_driver.rb

@@ -1,190 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# All rights reserved.
-#
-# This source file is distributed as part of the Net::SSH Secure Shell Client
-# library for Ruby. This file (and the library as a whole) may be used only as
-# allowed by either the BSD license, or the Ruby license (or, by association
-# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
-# distribution for the texts of these licenses.
-# -----------------------------------------------------------------------------
-# net-ssh website : http://net-ssh.rubyforge.org
-# project website: http://rubyforge.org/projects/net-ssh
-# =============================================================================
-#++
-
-$:.unshift "#{File.dirname(__FILE__)}/../../lib"
-
-require 'net/ssh/userauth/driver'
-require 'net/ssh/util/buffer'
-require 'test/unit'
-
-class TC_UserAuth_Driver < Test::Unit::TestCase
-
-  class Log
-    def debug?
-      false
-    end
-  end
-
-  class Buffers
-    def writer
-      Net::SSH::Util::WriterBuffer.new
-    end
-  end
-
-  class KeyManager
-    attr_reader :events
-
-    def initialize
-      @events = []
-    end
-
-    def clear!
-      @events << :clear!
-    end
-
-    def clear_host!
-      @events << :clear_host!
-    end
-
-    def finish
-      @events << :finish
-    end
-
-    def add( file )
-      @events << { :name => :add, :file => file }
-    end
-    alias :<< :add
-
-    def add_host_key( file )
-      @events << { :name => :add_host_key, :file => file }
-    end
-  end
-
-  class Session
-    attr_reader :script
-    attr_reader :messages
-
-    def initialize
-      @script = []
-      @messages = []
-    end
-
-    def send_message( message )
-      @messages << message
-    end
-
-    def wait_for_message
-      type, data = @script.shift
-      [ type, Net::SSH::Util::ReaderBuffer.new( data ) ]
-    end
-  end
-
-  class AuthMethod
-    attr_reader :next_service
-    attr_reader :username
-    attr_reader :data
-
-    def initialize( good )
-      @good = good
-    end
-
-    def authenticate( next_service, username, data={} )
-      @next_service = next_service
-      @username = username
-      @data = data
-      @good
-    end
-  end
-
-  def setup
-    @methods = { :test_fail => AuthMethod.new( false ),
-                :test_succeed => AuthMethod.new( true ) }
-    order = [ "test-fail", "test-succeed" ]
-    @driver = Net::SSH::UserAuth::Driver.new( Log.new,
-      Buffers.new, @methods, order )
-    @driver.session = @session = Session.new
-    @driver.key_manager = @manager = KeyManager.new
-  end
-
-  def test_set_key_files
-    @driver.set_key_files [ :one, :two ]
-    assert_equal [ :clear!,
-                   { :name => :add, :file => :one },
-                   { :name => :add, :file => :two } ], @manager.events
-  end
-
-  def test_set_host_key_files
-    @driver.set_host_key_files [ :one, :two ]
-    assert_equal [ :clear_host!,
-                   { :name => :add_host_key, :file => :one },
-                   { :name => :add_host_key, :file => :two } ], @manager.events
-  end
-
-  def test_order
-    assert_equal [ "test-fail", "test-succeed" ], @driver.order
-    original = [ "one", "two", "three" ]
-    @driver.set_auth_method_order( original )
-    @driver.order << "four"
-    assert_equal [ "one", "two", "three" ], original
-    assert_equal [ "one", "two", "three", "four" ], @driver.order
-    @driver.set_auth_method_order(*original)
-    assert_equal [ "one", "two", "three" ], @driver.order
-  end
-
-  def test_on_banner
-    msg = lang = nil
-    @driver.on_banner { |m,l| msg, lang = m, l }
-    @session.script << [ 53, "\0\0\0\15hello, world!\0\0\0\2en" ]
-    @session.script << [ 51, "\0\0\0\15howdy, earth!\0" ]
-    result = @driver.wait_for_message
-
-    assert_equal "hello, world!", msg
-    assert_equal "en", lang
-    assert_equal 51, result.message_type
-  end
-
-  def test_send_message
-    @driver.send_message "konnichi wa"
-    assert_equal [ "konnichi wa" ], @session.messages
-  end
-
-  def test_authenticate_bad_reply
-    @session.script << [ 0, "\0\0\0\0" ]
-    assert_raise( Net::SSH::Exception ) do
-      @driver.authenticate( "service", "test_user", "passwd" )
-    end
-  end
-
-  def test_authenticate_unexpected_reply
-    @session.script << [ 60, "\0\0\0\0" ]
-    assert_raise( Net::SSH::Exception ) do
-      @driver.authenticate( "service", "test_user", "passwd" )
-    end
-  end
-
-  def test_authenticate
-    @session.script << [ 6, "\0\0\0\14ssh-userauth" ]
-    assert @driver.authenticate( "service", "test_user", "passwd" )
-
-    @methods.each_value do |method|
-      assert_equal "service", method.next_service
-      assert_equal "test_user", method.username
-      assert_equal( { :password => "passwd", :key_manager => @manager }, method.data )
-    end
-
-    assert_equal :finish, @manager.events.last
-  end
-
-  def test_authenticate_not_implemented
-    @session.script << [ 6, "\0\0\0\14ssh-userauth" ]
-    @driver.set_auth_method_order "bogus"
-    assert_raise( NotImplementedError ) do
-      @driver.authenticate( "service", "test_user", "passwd" )
-    end
-    assert_equal :finish, @manager.events.last
-  end
-
-end

data/test/userauth/tc_integration.rb

@@ -1,97 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# All rights reserved.
-#
-# This source file is distributed as part of the Net::SSH Secure Shell Client
-# library for Ruby. This file (and the library as a whole) may be used only as
-# allowed by either the BSD license, or the Ruby license (or, by association
-# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
-# distribution for the texts of these licenses.
-# -----------------------------------------------------------------------------
-# net-ssh website : http://net-ssh.rubyforge.org
-# project website: http://rubyforge.org/projects/net-ssh
-# =============================================================================
-#++
-
-$:.unshift "#{File.dirname(__FILE__)}/../../lib"
-
-if $run_integration_tests || __FILE__ == $0
-
-  require 'needle'
-  require 'net/ssh/null-host-key-verifier'
-  require 'net/ssh/transport/services'
-  require 'net/ssh/userauth/services'
-  require 'test/unit'
-
-  class TC_UserAuth_Integration < Test::Unit::TestCase
-
-    HOST = "test.host"
-    USER = "test"
-    PASSWORD = "test/unit"
-    SERVICE = "ssh-connection"
-
-    def setup
-      @registry = Needle::Registry.new(
-        :logs => { :device=>STDOUT, :default_level => :WARN }
-      )
-
-      Net::SSH::Transport.register_services( @registry )
-      Net::SSH::UserAuth.register_services( @registry )
-
-      @registry.define do |b|
-        b.crypto_backend { :ossl }
-        b.transport_host { HOST }
-        b.host_key_verifier { Net::SSH::NullHostKeyVerifier.new }
-      end
-
-      @userauth = @registry[:userauth][:driver]
-    end
-
-    def teardown
-      @registry[:transport][:session].close
-      @registry.logs.close
-    end
-
-    def test_keyboard_interactive
-      @userauth.set_auth_method_order "keyboard-interactive"
-
-      called = 0
-      @registry.userauth[:methods].define.keyboard_interactive_callback do |c,p|
-        proc do |req|
-          called += 1
-          if req.prompts.length > 0
-            [ req.password ]
-          else
-            []
-          end
-        end
-      end
-
-      assert @userauth.authenticate( SERVICE, USER, PASSWORD )
-      assert_equal 2, called
-    end
-
-    def test_password
-      @userauth.set_auth_method_order "password"
-      assert @userauth.authenticate( SERVICE, USER, PASSWORD )
-    end
-
-    def test_password_bad
-      @userauth.set_auth_method_order "password"
-      assert !@userauth.authenticate( SERVICE, USER, PASSWORD + 'K' )
-    end
-
-    def test_publickey_bad
-      @userauth.set_auth_method_order "publickey"
-      assert !@userauth.authenticate( SERVICE, USER )
-    end
-
-    def test_hostbased_bad
-      @userauth.set_auth_method_order "hostbased"
-      assert !@userauth.authenticate( SERVICE, USER )
-    end
-
-  end
-
-end

data/test/userauth/tc_userkeys.rb

@@ -1,265 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
-# All rights reserved.
-#
-# This source file is distributed as part of the Net::SSH Secure Shell Client
-# library for Ruby. This file (and the library as a whole) may be used only as
-# allowed by either the BSD license, or the Ruby license (or, by association
-# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
-# distribution for the texts of these licenses.
-# -----------------------------------------------------------------------------
-# net-ssh website : http://net-ssh.rubyforge.org
-# project website: http://rubyforge.org/projects/net-ssh
-# =============================================================================
-#++
-
-$:.unshift "#{File.dirname(__FILE__)}/../../lib"
-
-require 'net/ssh/userauth/userkeys'
-require 'net/ssh/util/buffer'
-require 'test/unit'
-require 'logger'
-require 'stringio'
-
-class TC_UserKeyManager < Test::Unit::TestCase
-
-  class Buffers
-    def writer
-      Net::SSH::Util::WriterBuffer.new
-    end
-  end
-
-  class Keys
-    def load_public_key( file )
-      o = Object.new
-      singleton = class << o; self; end
-      singleton.send( :define_method, :ssh_type ) { "test" }
-      o
-    end
-
-    def load_private_key( file )
-      count = 0
-      o = Object.new
-      singleton = class << o; self; end
-      singleton.send( :define_method, :ssh_do_sign ) { |data| count += 1; "<#{data}:#{count}>" }
-      o
-    end
-  end
-
-  class AgentFactory
-    attr_reader :state
-
-    def initialize
-      @state = :new
-    end
-
-    def open
-      @state = :opened
-      self
-    end
-
-    def sign( identity, data )
-      "from the agent (#{identity.inspect}, #{data.inspect})"
-    end
-
-    def identities
-      [ :one, :two, :three ]
-    end
-
-    def close
-      @state = :closed
-    end
-  end
-
-  class ExistenceTester
-    def readable?( file )
-      true
-    end
-  end
-
-  def setup
-    @userkeys = Net::SSH::UserAuth::UserKeyManager.new
-    @userkeys.agent_factory = @agent_factory = AgentFactory.new
-    @userkeys.keys = Keys.new
-    @userkeys.log = @log = Logger.new( StringIO.new )
-    @userkeys.buffers = Buffers.new
-    @userkeys.key_existence_tester = ExistenceTester.new
-  end
-
-  def test_initialize
-    assert_equal 0, @userkeys.key_files.length
-    assert_equal 0, @userkeys.host_key_files.length
-    assert @userkeys.use_agent?
-  end
-
-  def test_add
-    assert_equal 0, @userkeys.key_files.length
-    @userkeys.add "hello"
-    assert_equal 1, @userkeys.key_files.length
-    assert_equal "hello", @userkeys.key_files.first
-    @userkeys.add "world"
-    assert_equal 2, @userkeys.key_files.length
-    @userkeys.add "hello"
-    assert_equal 2, @userkeys.key_files.length
-  end
-
-  def test_add_host_key
-    assert_equal 0, @userkeys.host_key_files.length
-    @userkeys.add_host_key "hello"
-    assert_equal 1, @userkeys.host_key_files.length
-    assert_equal "hello", @userkeys.host_key_files.first
-    @userkeys.add_host_key "world"
-    assert_equal 2, @userkeys.host_key_files.length
-    @userkeys.add_host_key "hello"
-    assert_equal 2, @userkeys.host_key_files.length
-  end
-
-  def test_clear!
-    @userkeys.add "hello"
-    @userkeys.add "howdy"
-    assert_equal 2, @userkeys.key_files.length
-    @userkeys.clear!
-    assert_equal 0, @userkeys.key_files.length
-  end
-
-  def test_clear_host!
-    @userkeys.add_host_key "hello"
-    @userkeys.add_host_key "howdy"
-    assert_equal 2, @userkeys.host_key_files.length
-    @userkeys.clear_host!
-    assert_equal 0, @userkeys.host_key_files.length
-  end
-
-  def test_clear_and_clear_host!
-    @userkeys.add "hello"
-    @userkeys.add "howdy"
-    @userkeys.add_host_key "hello"
-    @userkeys.add_host_key "howdy"
-    assert_equal 2, @userkeys.key_files.length
-    assert_equal 2, @userkeys.host_key_files.length
-    @userkeys.clear!
-    assert_equal 0, @userkeys.key_files.length
-    assert_equal 2, @userkeys.host_key_files.length
-    @userkeys.add "hello"
-    @userkeys.add "howdy"
-    assert_equal 2, @userkeys.key_files.length
-    assert_equal 2, @userkeys.host_key_files.length
-    @userkeys.clear_host!
-    assert_equal 2, @userkeys.key_files.length
-    assert_equal 0, @userkeys.host_key_files.length
-  end
-
-  def test_finish_use_agent_unopened
-    @userkeys.use_agent = true
-    assert_equal :new, @agent_factory.state
-    @userkeys.finish
-    assert_equal :new, @agent_factory.state
-  end
-
-  def test_finish_unopened_no_use_agent
-    @userkeys.use_agent = false
-    assert_equal :new, @agent_factory.state
-    @userkeys.finish
-    assert_equal :new, @agent_factory.state
-  end
-
-  def test_finish_use_agent_opened
-    @userkeys.use_agent = true
-    @userkeys.identities
-    assert_equal :opened, @agent_factory.state
-    @userkeys.finish
-    assert_equal :closed, @agent_factory.state
-  end
-
-  def test_finish_no_use_agent_opened
-    @userkeys.use_agent = false
-    @userkeys.identities
-    assert_equal :new, @agent_factory.state
-    @userkeys.finish
-    assert_equal :new, @agent_factory.state
-  end
-
-  def test_identities_no_use_agent_no_files
-    @userkeys.use_agent = false
-    ids = @userkeys.identities
-    assert_equal 0, ids.length
-  end
-
-  def test_identities_use_agent_no_files
-    @userkeys.use_agent = true
-    ids = @userkeys.identities
-    assert_equal 3, ids.length
-  end
-
-  def test_identities_no_use_agent_files
-    @userkeys.use_agent = false
-    @userkeys.add "one"
-    @userkeys.add "two"
-    @userkeys.add "three"
-    ids = @userkeys.identities
-    assert_equal 3, ids.length
-  end
-
-  def test_identities_use_agent_files
-    @userkeys.use_agent = true
-    @userkeys.add "one"
-    @userkeys.add "two"
-    @userkeys.add "three"
-    ids = @userkeys.identities
-    assert_equal 6, ids.length
-  end
-
-  def test_host_identities_no_files
-    ids = @userkeys.host_identities
-    assert_equal 0, ids.length
-  end
-
-  def test_host_identities_files
-    @userkeys.add_host_key "one"
-    @userkeys.add_host_key "two"
-    @userkeys.add_host_key "three"
-    ids = @userkeys.host_identities
-    assert_equal 3, ids.length
-  end
-
-  def test_sign_from_file
-    @userkeys.use_agent = false
-    @userkeys.add "one"
-    ids = @userkeys.identities
-    assert_equal 1, ids.length
-    data = @userkeys.sign( ids.first, "hello" )
-    assert_equal "\0\0\0\4test\0\0\0\11<hello:1>", data
-  end
-
-  def test_sign_from_agent
-    @userkeys.use_agent = true
-    ids = @userkeys.identities
-    assert_equal 3, ids.length
-    data = @userkeys.sign( ids.first, "hello" )
-    assert_equal %q{from the agent (:one, "hello")}, data
-  end
-
-  def test_sign_from_key
-    @userkeys.use_agent = false
-    @userkeys.add "one"
-    ids = @userkeys.identities
-    assert_equal 1, ids.length
-    @userkeys.sign( ids.first, "hello" )
-    data = @userkeys.sign( ids.first, "hello" )
-    assert_equal "\0\0\0\4test\0\0\0\11<hello:2>", data
-  end
-
-  def test_use_agent
-    assert @userkeys.use_agent?
-    @userkeys.identities
-    assert_equal :opened, @agent_factory.state
-    @userkeys.use_agent = false
-    assert_equal :closed, @agent_factory.state
-    @userkeys.use_agent = true
-    assert_equal :closed, @agent_factory.state
-    @userkeys.identities
-    assert_equal :opened, @agent_factory.state
-  end
-
-end