net-ssh 1.1.4 → 2.0.0

data/test/service/agentforward/tc_driver.rb

@@ -1,138 +0,0 @@
-#--
-# =============================================================================
-# Copyright (c) 2007, Chris Andrews (chris@nodnol.org),
-#   Jamis Buck (jgb3@email.byu.edu)
-# 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/service/agentforward/driver'
-require 'net/ssh/util/buffer'
-require 'test/unit'
-require 'socket'
-
-class TC_AgentForward_Driver < Test::Unit::TestCase
-
-  class Net::SSH::Service::AgentForward::Driver
-    attr_reader :data
-  end
-
-  class Buffers
-    def writer
-      Net::SSH::Util::WriterBuffer.new
-    end
-  end
-
-  class Log
-    def debug?
-      true
-    end
-    def debug(msg)
-    end
-  end
-
-  class Agent
-    attr_reader :sends
-    def initialize
-      @sends = []
-    end
-    def read_raw_packet
-      "raw agent data"
-    end
-    def send_raw_packet(data)
-      @sends.push data
-    end
-  end
-
-  class MockObject
-    attr_reader :events
-    attr_reader :blocks
-    attr_reader :returns
-
-    def initialize
-      @events = []
-      @blocks = []
-      @returns = []
-    end
-    def method_missing( sym, *args, &block )
-      @blocks << block
-      @events << [ sym, args, !block.nil? ]
-      @returns << MockObject.new
-      @returns.last
-    end
-    def method( sym )
-      @events << [ :method, [ sym ], false ]
-      lambda { || }
-    end
-    def respond_to?( sym )
-      true
-    end
-  end
-
-  def setup
-    @connection = MockObject.new
-    @agent = Agent.new
-
-    @driver = Net::SSH::Service::AgentForward::Driver.new( @connection,
-      Buffers.new, Log.new, @agent )
-  end
-
-  def test_initialize
-    assert_equal [ [:add_channel_open_handler, [ "auth-agent@openssh.com" ], true] ],
-      @connection.events
-  end
-  
-  def test_request
-    @driver.request
-    assert_equal 2, @connection.events.length
-    assert_equal [:channel_request, ["auth-agent-req@openssh.com"], false],
-      @connection.events[1]
-  end
-
-  def test_do_open_channel
-    connection = MockObject.new
-    channel = MockObject.new
-    @driver.do_open_channel(connection, channel, nil)
-    assert_equal [ [:on_data, [], true] ],
-      channel.events
-  end
-
-  def test_do_data_complete_packet
-    channel = MockObject.new
-    data = "\000\000\000\001v"
-    @driver.do_data(channel, data)
-    assert_equal [ [:send_data, ["raw agent data"], false ] ],
-      channel.events
-    assert_equal [ data ],
-      @agent.sends
-    assert_equal '', @driver.data
-  end
-
-  def test_do_data_incomplete_packet
-    channel = MockObject.new
-
-    @driver.do_data(channel, "\000\000\000\001")
-    assert_equal 0, channel.events.length
-    assert_equal 0, @agent.sends.length
-    assert_equal @driver.data, "\000\000\000\001"
-
-    @driver.do_data(channel, "v")
-    assert_equal [ [:send_data, ["raw agent data"], false ] ],
-      channel.events
-    assert_equal [ "\000\000\000\001v" ],
-      @agent.sends
-    assert_equal '', @driver.data
-  end
-
-end

data/test/service/forward/tc_driver.rb

@@ -1,289 +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/service/forward/driver'
-require 'net/ssh/util/buffer'
-require 'test/unit'
-require 'socket'
-
-class TC_Forward_Driver < Test::Unit::TestCase
-
-  class Buffers
-    def writer
-      Net::SSH::Util::WriterBuffer.new
-    end
-  end
-
-  class Log
-    def debug?
-      true
-    end
-    def debug(msg)
-    end
-  end
-
-  class MockObject
-    attr_reader :events
-    attr_reader :blocks
-    attr_reader :returns
-
-    def initialize
-      @events = []
-      @blocks = []
-      @returns = []
-    end
-    def method_missing( sym, *args, &block )
-      @blocks << block
-      @events << [ sym, args, !block.nil? ]
-      @returns << MockObject.new
-      @returns.last
-    end
-    def method( sym )
-      @events << [ :method, [ sym ], false ]
-      lambda { || }
-    end
-    def respond_to?( sym )
-      true
-    end
-  end
-
-  def setup
-    @connection = MockObject.new
-    @handlers = { :local => MockObject.new, :remote => MockObject.new }
-
-    @driver = Net::SSH::Service::Forward::Driver.new( @connection,
-      Buffers.new, Log.new, @handlers )
-  end
-
-  def test_initialize
-    assert_equal 0, @driver.direct_channel_count
-    assert_equal 0, @driver.open_direct_channel_count
-    assert_equal [ [ :add_channel_open_handler, [ "forwarded-tcpip" ], true ] ],
-      @connection.events
-  end
-
-  def test_direct_channel
-    handler = MockObject.new
-    @driver.direct_channel( 1, "remote.host", 2, handler, 3, 4, 5 )
-
-    assert_equal 1, @driver.direct_channel_count
-    assert_equal 1, @driver.open_direct_channel_count
-
-    assert_equal 2, @connection.events.length
-    assert_equal [ :open_channel, [ "direct-tcpip",
-      Net::SSH::Util::WriterBuffer.new(
-        "\0\0\0\xbremote.host\0\0\0\2\0\0\0\x09127.0.0.1\0\0\0\1" ) ], true ],
-      @connection.events.last
-    assert_equal [ [ :on_confirm_failed, [], true ] ],
-      @connection.returns.last.events
-
-    channel = MockObject.new
-    @connection.blocks.last.call( channel )
-
-    assert_equal [ [ :method, [ :on_receive ], false ],
-                   [ :method, [ :on_eof ], false ],
-                   [ :confirm, [ channel, 1, "remote.host", 2, 3, 4, 5 ], false ],
-                   [ :process, [ channel ], false ] ], handler.events
-
-    assert_equal [ [ :on_data, [], true ],
-                   [ :on_eof, [], true ],
-                   [ :on_close, [], true ] ], channel.events
-
-    channel.blocks.last.call( channel )
-    assert_equal 0, @driver.open_direct_channel_count
-    assert_equal [ :on_close, [ channel ], false ], handler.events.last
-  end
-
-  def test_local_bad_parm_count
-    assert_raise( ArgumentError ) do
-      @driver.local 1, 2
-    end
-    assert_raise( ArgumentError ) do
-      @driver.local 1, 2, 3, 4, 5
-    end
-  end
-
-  [
-    [ :local_with_3, [ 12233, "test.host", 80 ] ],
-    [ :local_with_4, [ "localhost", 12233, "test.host", 80 ] ]
-  ].each do |name, args|
-    define_method "test_#{name}" do
-      assert_equal 0, @driver.open_direct_channel_count
-      assert @driver.active_locals.empty?
-
-      @driver.local(*args)
-
-      address = '127.0.0.1'
-      address = args.shift if args.first.is_a? String
-      port = args.shift
-
-      sleep 0.1
-      socket = TCPSocket.new( address, port )
-      sleep 0.1
-
-      assert_equal 1, @driver.open_direct_channel_count
-      assert !@driver.active_locals.empty?
-
-      @driver.cancel_local( port, address )
-      assert @driver.active_locals.empty?
-    end
-  end
-
-  [
-    [ :remote_with_2, [ 80 ] ],
-    [ :remote_with_zero_port, [ 0 ] ],
-    [ :remote_with_3, [ 80, 'localhost' ] ]
-  ].each do |name, args|
-    define_method "test_#{name}_success" do
-      handler = MockObject.new
-
-      port = args[0]
-      host = args.last
-      host = "127.0.0.1" unless host.is_a?( String )
-
-      assert @driver.active_remotes.empty?
-      assert_nothing_raised { @driver.remote( handler, *args ) }
-      assert @driver.active_remotes.empty?
-
-      assert_equal [ :global_request,
-        [ "tcpip-forward", Net::SSH::Util::WriterBuffer.new(
-          "\0\0\0#{host.length.chr}#{host}\0\0\0#{port.chr}" ) ], true ],
-        @connection.events.last
-
-      port = 0300 if port == 0
-      response = Net::SSH::Util::ReaderBuffer.new( "\0\0\0#{port.chr}" )
-      @connection.blocks.last.call( true, response )
-
-      assert_equal [ port ], @driver.active_remotes
-
-      assert_equal [ [ :setup, [ port ], false ] ], handler.events
-    end
-
-    define_method "test_#{name}_fail_handled" do
-      handler = MockObject.new
-
-      port = args[0]
-      host = args.last
-      host = "127.0.0.1" unless host.is_a?( String )
-
-      assert @driver.active_remotes.empty?
-      assert_nothing_raised { @driver.remote( handler, *args ) }
-      assert @driver.active_remotes.empty?
-
-      @connection.blocks.last.call( false, nil )
-
-      assert_equal [ [ :error, [ "remote port #{port} could not be forwarded to local host" ], false ] ], handler.events
-    end
-
-    define_method "test_#{name}_fail_unhandled" do
-      port = args[0]
-      host = args.last
-      host = "127.0.0.1" unless host.is_a?( String )
-
-      assert @driver.active_remotes.empty?
-      assert_nothing_raised { @driver.remote( nil, *args ) }
-      assert @driver.active_remotes.empty?
-
-      assert_raise( Net::SSH::Exception ) do
-        @connection.blocks.last.call( false, nil )
-      end
-    end
-
-    define_method "test_#{name}_duplicate" do
-      handler = MockObject.new
-      port = args[0]
-      port = 0300 if port == 0
-      @driver.remote( handler, *args )
-      response = Net::SSH::Util::ReaderBuffer.new( "\0\0\0#{port.chr}" )
-      @connection.blocks.last.call( true, response )
-
-      args = args.dup
-      args[0] = port
-      assert_raise( Net::SSH::Exception ) do
-        @driver.remote( handler, *args )
-      end
-    end
-  end
-
-  def test_remote_to
-    @driver.remote_to( 1, "test.host", 2 )
-    assert_equal [ [ :call, [ 1, "test.host" ], false ] ],
-      @handlers[:remote].events
-  end
-
-  def test_cancel_remote_success
-    @driver.remote_to( 1, "test.host", 2 )
-    @connection.blocks.last.call( true, nil )
-    assert_equal 1, @driver.active_remotes.length
-
-    @driver.cancel_remote( 2 )
-    assert_equal [ :global_request, [ "cancel-tcpip-forward",
-      Net::SSH::Util::WriterBuffer.new( "\0\0\0\x09127.0.0.1\0\0\0\2" ) ],
-      true ], @connection.events.last
-
-    @connection.blocks.last.call( true, nil )
-    assert_equal 0, @driver.active_remotes.length
-  end
-
-  def test_cancel_remote_failure
-    @driver.remote_to( 1, "test.host", 2 )
-    @connection.blocks.last.call( true, nil )
-    assert_equal 1, @driver.active_remotes.length
-
-    @driver.cancel_remote( 2 )
-    assert_raise( Net::SSH::Exception ) do
-      @connection.blocks.last.call( false, nil )
-    end
-  end
-
-  def test_do_open_channel_valid
-    handler = MockObject.new
-    @driver.remote( handler, 80 )
-    @connection.blocks.last.call( true, nil )
-    assert_equal 1, @driver.active_remotes.length
-
-    channel = MockObject.new
-    data = Net::SSH::Util::ReaderBuffer.new(
-      "\0\0\0\017123.456.789.012\0\0\0\120\0\0\0\017345.678.901.234\0\0\0\2" )
-
-    @driver.do_open_channel( @connection, channel, data )
-
-    assert_equal [
-      [ :on_open, [ channel, "123.456.789.012", 80, "345.678.901.234", 2 ], false ],
-      [ :method, [ :on_receive ], false ],
-      [ :method, [ :on_close ], false ],
-      [ :method, [ :on_eof ], false ] ], handler.events[1..-1]
-
-    assert_equal [
-      [ :on_data, [], true ],
-      [ :on_close, [], true ],
-      [ :on_eof, [], true ] ], channel.events
-  end
-
-  def test_do_open_channel_invalid
-    handler = MockObject.new
-    channel = MockObject.new
-    data = Net::SSH::Util::ReaderBuffer.new(
-      "\0\0\0\017123.456.789.012\0\0\0\120\0\0\0\017345.678.901.234\0\0\0\2" )
-
-    assert_raise( Net::SSH::Exception ) do
-      @driver.do_open_channel( @connection, channel, data )
-    end
-  end
-
-end

data/test/service/forward/tc_local_network_handler.rb

@@ -1,123 +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/service/forward/local-network-handler'
-require 'test/unit'
-
-class TC_LocalNetworkHandler < Test::Unit::TestCase
-
-  class Log
-    def debug?
-      true
-    end
-
-    def debug( msg )
-    end
-
-    def error( msg )
-    end
-  end
-
-  class Client
-    attr_reader :events
-    attr_reader :script
-
-    def initialize
-      @events = []
-      @script = []
-    end
-
-    def send( data, code )
-      @events << [ :send, data, code ]
-    end
-
-    def recv( block_size )
-      @events << [ :recv, block_size ]
-      @script.shift
-    end
-  end
-
-  class Channel
-    attr_reader :events
-
-    def initialize
-      @values = Hash.new
-      @events = []
-    end
-
-    def []( name )
-      @values[name]
-    end
-
-    def []=( name, value )
-      @values[name] = value
-    end
-
-    def close
-      @events << :close
-    end
-
-    def send_data( data )
-      @events << data
-    end
-  end
-
-  def setup
-    @client = Client.new
-    @channel = Channel.new
-    @handler = Net::SSH::Service::Forward::LocalNetworkHandler.new(
-      Log.new, 16, @client )
-  end
-
-  def test_on_receive
-    @handler.on_receive( @channel, "hello" )
-    assert_equal [ [ :send, "hello", 0 ] ], @client.events
-  end
-
-  def test_on_eof
-    assert !@channel[:eof]
-    @handler.on_eof( @channel )
-    assert @channel[:eof]
-  end
-
-  def test_process_no_eof
-    @client.script << "part 1"
-    @client.script << "part 2"
-    @client.script << "part 3"
-
-    @handler.process @channel
-
-    assert_equal [ "part 1", "part 2", "part 3", :close ], @channel.events
-    assert_equal [ [ :recv, 16 ], [ :recv, 16 ], [ :recv, 16 ], [ :recv, 16 ] ],
-      @client.events
-  end
-
-  def test_process_eof
-    @channel[:eof] = true
-    @handler.process @channel
-    assert_equal [ :close ], @channel.events
-    assert_equal [], @client.events
-  end
-
-  def test_process_error
-    assert_nothing_raised do
-      @handler.process nil
-    end
-  end
-
-end