net-ssh 0.5.0

data/test/service/process/tc_driver.rb

@@ -0,0 +1,79 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, 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 "../../../lib"
+
+require 'test/unit'
+require 'net/ssh/service/process/driver'
+
+class TC_Process_Driver < Test::Unit::TestCase
+
+  class MockObject
+    attr_reader :events
+
+    def initialize
+      @events = []
+    end
+
+    def method_missing( sym, *args, &block )
+      token = [ sym, *args ]
+      token << :with_block if block
+      @events << token
+      :return_value
+    end
+
+    undef_method :loop
+  end
+
+  def setup
+    @connection = MockObject.new
+    @log = MockObject.new
+    @handlers = { :open => MockObject.new, :popen3 => MockObject.new }
+    @driver = Net::SSH::Service::Process::Driver.new(
+      @connection, @log, @handlers )
+  end
+
+  def test_open_no_block
+    result = @driver.open( "foo" )
+    assert_equal :return_value, result
+    assert_equal [ [ :call, "foo" ] ], @handlers[:open].events
+    assert_equal [], @connection.events
+  end
+
+  def test_open_block
+    result = @driver.open( "foo" ) do |r|
+      assert_equal :return_value, r
+    end
+    assert_nil result
+    assert_equal [ [ :call, "foo" ] ], @handlers[:open].events
+    assert_equal [ [ :loop ] ], @connection.events
+  end
+
+  def test_popen3_no_block
+    result = @driver.popen3( "foo" )
+    assert_equal :return_value, result
+    assert_equal [ [ :popen3, "foo" ] ], @handlers[:popen3].events
+    assert_equal [], @connection.events
+  end
+
+  def test_popen3_no_block
+    result = @driver.popen3( "foo" ) { |f| }
+    assert_equal :return_value, result
+    assert_equal [ [ :popen3, "foo", :with_block ] ], @handlers[:popen3].events
+    assert_equal [], @connection.events
+  end
+
+end

data/test/service/process/tc_integration.rb

@@ -0,0 +1,117 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, 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 "../../../lib"
+
+if $run_integration_tests || __FILE__ == $0
+
+  require 'needle'
+  require 'test/unit'
+
+  class TC_Process_Integration < Test::Unit::TestCase
+
+    HOST = "test.host"
+    USER = "test"
+    PASSWORD = "test/unit"
+
+    def setup
+      @registry = Needle::Registry.define(
+        :logs => { :device => STDOUT, :default_level => :warn }
+      ) do |b|
+        b.require 'net/ssh/transport/services', "Net::SSH::Transport"
+        b.require 'net/ssh/userauth/services', "Net::SSH::UserAuth"
+        b.require 'net/ssh/connection/services', "Net::SSH::Connection"
+        b.require 'net/ssh/service/process/services',
+          "Net::SSH::Service::Process"
+        
+        b.crypto_backend { :ossl }
+        b.transport_host { HOST }
+      end
+
+      @registry.userauth.driver.authenticate( "ssh-connection", USER, PASSWORD )
+
+      @driver = @registry.process.driver
+    end
+
+    def teardown
+      @registry.transport.session.close
+    end
+
+    def test_open_no_block
+      script = [ "2+2", "3+2*5", "quit" ]
+      expect = [ "4\n", "13\n", :exit ]
+      results = []
+
+      process = @driver.open( "bc" )
+      process.on_success { |p| p.puts script.shift }
+      process.on_stdout do |p,d|
+        results << d
+        if script.empty?
+          p.close_input
+        else
+          p.puts script.shift
+        end
+      end
+      process.on_exit { results << :exit }
+
+      @registry.connection.driver.loop
+
+      assert_equal expect, results
+    end
+
+    def test_open_block
+      script = [ "2+2", "3+2*5", "quit" ]
+      expect = [ "4\n", "13\n", :exit ]
+      results = []
+
+      @driver.open( "bc" ) do |process|
+        process.on_success { |p| p.puts script.shift }
+        process.on_stdout do |p,d|
+          results << d
+          if script.empty?
+            p.close_input
+          else
+            p.puts script.shift
+          end
+        end
+        process.on_exit { results << :exit }
+      end
+
+      assert_equal expect, results
+    end
+
+    def test_popen3_no_block
+      stdin, stdout, stderr = @driver.popen3( "bc" )
+      stdin.puts "2+2"
+      assert_equal "4\n", stdout.read
+      stdin.puts "3+2*5"
+      assert_equal "13\n", stdout.read
+      stdin.puts "quit"
+    end
+
+    def test_popen3_block
+      @driver.popen3( "bc" ) do |stdin,stdout,stderr|
+        stdin.puts "2+2"
+        assert_equal "4\n", stdout.read
+        stdin.puts "3+2*5"
+        assert_equal "13\n", stdout.read
+        stdin.puts "quit"
+      end
+    end
+
+  end
+
+end

data/test/service/process/tc_open.rb

@@ -0,0 +1,179 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, 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 "../../../lib"
+
+require 'test/unit'
+require 'net/ssh/service/process/open'
+
+class TC_Process_Open < Test::Unit::TestCase
+
+  class Log
+    def debug?
+      true
+    end
+
+    def debug(msg)
+    end
+  end
+
+  class MockObject
+    attr_reader :events
+
+    def initialize
+      @events = []
+    end
+
+    def method_missing( sym, *args, &block )
+      token = [ sym, *args ]
+      token << true if block
+      @events << token
+    end
+  end
+
+  class Connection < MockObject
+    attr_reader :channel
+
+    def open_channel( *args, &block )
+      @events << [ :open_channel, *args ] + [ block ? true : false ]
+      @channel = MockObject.new
+    end
+  end
+
+  def create_manager( &block )
+    connection = Connection.new
+    mgr = Net::SSH::Service::Process::OpenManager.new( connection,
+      Log.new, "test", &block )
+    [ mgr, connection ]
+  end
+
+  def test_no_block
+    mgr, conn = create_manager
+    assert_equal [ [ :open_channel, "session", true ] ], conn.events
+  end
+
+  def test_with_block
+    yielded = false
+    mgr, conn = create_manager { |mgr| yielded = true }
+    assert_equal [ [ :open_channel, "session", true ],
+                   [ :loop ] ], conn.events
+    assert yielded
+  end
+
+  def test_write
+    mgr, conn = create_manager
+    mgr.write "foo"
+    assert_equal [ [ :send_data, "foo" ] ], conn.channel.events
+  end
+
+  def test_puts_nonl
+    mgr, conn = create_manager
+    mgr.puts "foo"
+    assert_equal [ [ :send_data, "foo\n" ] ], conn.channel.events
+  end
+
+  def test_puts_nl
+    mgr, conn = create_manager
+    mgr.puts "foo\n"
+    assert_equal [ [ :send_data, "foo\n" ] ], conn.channel.events
+  end
+
+  def test_close_input
+    mgr, conn = create_manager
+    mgr.close_input
+    assert_equal [ [ :send_eof ] ], conn.channel.events
+  end
+
+  def test_close
+    mgr, conn = create_manager
+    mgr.close
+    assert_equal [ [ :close ] ], conn.channel.events
+  end
+
+  def test_confirm
+    mgr, conn = create_manager
+    chan = MockObject.new
+    mgr.do_confirm chan
+    assert_equal [ [ :on_success, true ], [ :on_failure, true ],
+                   [ :exec, "test", true ] ], chan.events
+  end
+
+  def test_exec_success
+    mgr, conn = create_manager
+    chan = MockObject.new
+    called = false
+    mgr.on_success { |ch| called = true }
+    mgr.do_exec_success chan
+    assert_equal [ [ :on_data, true ], [ :on_extended_data, true ],
+                   [ :on_close, true ], [ :on_request, true ] ], chan.events
+    assert called
+  end
+
+  def test_exec_failure_no_callback
+    mgr, conn = create_manager
+    chan = MockObject.new
+    assert_raise( Net::SSH::Exception ) do
+      mgr.do_exec_failure chan
+    end
+  end
+
+  def test_exec_failure_callback
+    mgr, conn = create_manager
+    chan = MockObject.new
+    called = false
+    mgr.on_failure { |ch,msg| called = true }
+    mgr.do_exec_failure chan
+    assert called
+  end
+
+  def test_stdout
+    mgr, conn = create_manager
+    chan = MockObject.new
+    data = ""
+    mgr.on_stdout { |ch,d| data = d }
+    mgr.do_data chan, "foo"
+    assert_equal "foo", data
+  end
+
+  def test_extended_data_not_stderr
+    mgr, conn = create_manager
+    chan = MockObject.new
+    data = ""
+    mgr.on_stderr { |ch,d| data = d }
+    mgr.do_extended_data chan, 0, "foo"
+    assert_equal "", data
+  end
+
+  def test_extended_data_stderr
+    mgr, conn = create_manager
+    chan = MockObject.new
+    data = ""
+    mgr.on_stderr { |ch,d| data = d }
+    mgr.do_extended_data chan, 1, "foo"
+    assert_equal "foo", data
+  end
+
+  def test_close
+    mgr, conn = create_manager
+    chan = MockObject.new
+    called = false
+    mgr.on_exit { |ch,status| called = true }
+    mgr.do_close chan
+    assert called
+    assert_equal [], chan.events
+  end
+
+end

data/test/service/process/tc_popen3.rb

@@ -0,0 +1,164 @@
+#--
+# =============================================================================
+# Copyright (c) 2004, 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 "../../../lib"
+
+require 'test/unit'
+require 'net/ssh/service/process/popen3'
+
+class TC_Process_POpen3 < Test::Unit::TestCase
+
+  class MockObject
+    attr_reader :events
+
+    def initialize
+      @events = []
+    end
+
+    def method_missing( sym, *args, &block )
+      token = [ sym, *args ]
+      token << :with_block if block
+      @events << token
+    end
+
+    undef_method :loop
+  end
+
+  class Channel < MockObject
+    attr_reader :connection
+
+    def connection
+      @connection ||= MockObject.new
+    end
+  end
+
+  class TC_SSHStdinPipe < Test::Unit::TestCase
+    def setup
+      @channel = MockObject.new
+      @pipe = Net::SSH::Service::Process::POpen3Manager::SSHStdinPipe.new(
+        @channel )
+    end
+
+    def test_write
+      @pipe.write "foo"
+      assert_equal [ [ :send_data, "foo" ] ], @channel.events
+    end
+
+    def test_puts_nonl
+      @pipe.puts "foo"
+      assert_equal [ [ :send_data, "foo\n" ] ], @channel.events
+    end
+
+    def test_puts_nl
+      @pipe.puts "foo\n"
+      assert_equal [ [ :send_data, "foo\n" ] ], @channel.events
+    end
+  end
+
+  class TC_SSHStdoutPipe < Test::Unit::TestCase
+    def setup
+      @channel = Channel.new
+      @pipe = Net::SSH::Service::Process::POpen3Manager::SSHStdoutPipe.new(
+        @channel )
+    end
+
+    def test_read_empty
+      s = @pipe.read
+      assert_equal "", s
+      assert_equal [ [ :loop, :with_block ] ], @channel.connection.events
+    end
+
+    def test_read_full
+      @pipe.do_data @channel, "foo"
+      s = @pipe.read
+      assert_equal "foo", s
+      assert_equal [], @channel.connection.events
+    end
+  end
+
+  class TC_SSHStderrPipe < Test::Unit::TestCase
+    def setup
+      @channel = Channel.new
+      @pipe = Net::SSH::Service::Process::POpen3Manager::SSHStderrPipe.new(
+        @channel )
+    end
+
+    def test_read_empty
+      s = @pipe.read
+      assert_equal "", s
+      assert_equal [ [ :loop, :with_block ] ], @channel.connection.events
+    end
+
+    def test_read_non_stderr
+      @pipe.do_data @channel, 0, "foo"
+      s = @pipe.read
+      assert_equal "", s
+      assert_equal [ [ :loop, :with_block ] ], @channel.connection.events
+    end
+
+    def test_read_stderr
+      @pipe.do_data @channel, 1, "foo"
+      s = @pipe.read
+      assert_equal "foo", s
+      assert_equal [], @channel.connection.events
+    end
+  end
+
+  class Connection < MockObject
+    class Channel < MockObject
+      attr_reader :success
+      attr_reader :failure
+
+      def on_success( &block )
+        super
+        @success = block
+      end
+
+      def on_failure( &block )
+        super
+        @failure = block
+      end
+    end
+
+    attr_reader :channel
+    def open_channel( type )
+      super
+      yield @channel = Channel.new
+    end
+  end
+
+  def test_popen3
+    conn = Connection.new
+    mgr = Net::SSH::Service::Process::POpen3Manager.new( conn, nil )
+    mgr.popen3( "foo" ) do |a,b,c|
+      assert_instance_of(
+        Net::SSH::Service::Process::POpen3Manager::SSHStdinPipe, a )
+      assert_instance_of(
+        Net::SSH::Service::Process::POpen3Manager::SSHStdoutPipe, b )
+      assert_instance_of(
+        Net::SSH::Service::Process::POpen3Manager::SSHStderrPipe, c )
+    end
+
+    assert_equal [ [ :open_channel, "session", :with_block ],
+      [ :loop ] ], conn.events
+    chan = conn.channel
+    assert_equal [ [ :on_success, :with_block ],
+      [ :on_failure, :with_block ], [ :exec, "foo", true ] ], chan.events
+
+    chan.success.call( chan )
+  end
+
+end