Module: Net::SSH::Test

Defined in:
lib/net/ssh/test.rb,
lib/net/ssh/test/kex.rb,
lib/net/ssh/test/packet.rb,
lib/net/ssh/test/script.rb,
lib/net/ssh/test/socket.rb,
lib/net/ssh/test/channel.rb,
lib/net/ssh/test/extensions.rb,
lib/net/ssh/test/local_packet.rb,
lib/net/ssh/test/remote_packet.rb

Overview

This module may be used in unit tests, for when you want to test that your SSH state machines are really doing what you expect they are doing. You will typically include this module in your unit test class, and then build a "story" of expected sends and receives:

require 'minitest/autorun'
require 'net/ssh/test'

class MyTest < Minitest::Test
include Net::SSH::Test

def test_exec_via_channel_works
  story do |session|
    channel = session.opens_channel
    channel.sends_exec "ls"
    channel.gets_data "result of ls"
    channel.gets_close
    channel.sends_close
  end

  assert_scripted do
    result = nil

    connection.open_channel do |ch|
      ch.exec("ls") do |success|
        ch.on_data { |c, data| result = data }
        ch.on_close { |c| c.close }
      end
    end

    connection.loop
    assert_equal "result of ls", result
  end
end
end

See Net::SSH::Test::Channel and Net::SSH::Test::Script for more options.

Note that the Net::SSH::Test system is rather finicky yet, and can be kind of frustrating to get working. Any suggestions for improvement will be welcome!

Defined Under Namespace

Modules: Extensions Classes: Channel, Kex, LocalPacket, Packet, RemotePacket, Script, Socket

Instance Method Summary collapse

Instance Method Details

#assert_scripted ⇒ Object

First asserts that a story has been described (see #story). Then yields, and then asserts that all items described in the script have been processed. Typically, this is called immediately after a story has been built, and the SSH commands being tested are then executed within the block passed to this assertion.



85
86
87
88
89
90
91
 
# File 'lib/net/ssh/test.rb', line 85

def assert_scripted
  raise "there is no script to be processed" if socket.script.events.empty?

  Net::SSH::Test::Extensions::IO.with_test_extension { yield }
  assert socket.script.events.empty?, "there should not be any remaining scripted events, but there are still" \
                                      "#{socket.script.events.length} pending"
end

#connection(options = {}) ⇒ Object

Returns the connection session (Net::SSH::Connection::Session) for use in these tests. It is a fully functional SSH session, operating over a mock socket (#socket).



66
67
68
 
# File 'lib/net/ssh/test.rb', line 66

def connection(options = {})
  @connection ||= Net::SSH::Connection::Session.new(transport(options), options)
end

#socket(options = {}) ⇒ Object

Returns the test socket instance to use for these tests (see Net::SSH::Test::Socket).



59
60
61
 
# File 'lib/net/ssh/test.rb', line 59

def socket(options = {})
  @socket ||= Net::SSH::Test::Socket.new
end

#story ⇒ Object

If a block is given, yields the script for the test socket (#socket). Otherwise, simply returns the socket's script. See Net::SSH::Test::Script.



52
53
54
55
 
# File 'lib/net/ssh/test.rb', line 52

def story
  Net::SSH::Test::Extensions::IO.with_test_extension { yield socket.script if block_given? }
  return socket.script
end

#transport(options = {}) ⇒ Object

Returns the transport session (Net::SSH::Transport::Session) for use in these tests. It is a fully functional SSH transport session, operating over a mock socket (#socket).



73
74
75
76
77
78
 
# File 'lib/net/ssh/test.rb', line 73

def transport(options = {})
  @transport ||= Net::SSH::Transport::Session.new(
    options[:host] || "localhost",
    options.merge(kex: "test", host_key: "ssh-rsa", append_all_supported_algorithms: true, verify_host_key: :never, proxy: socket(options))
  )
end