Merge pull request #283 from p0deje/test_channel_requests_pty

Add Net::SSH::Test::Channel#sends_request_pty for scripting PTY requests

3 files changed, 17 insertions, 0 deletions

diff --git a/lib/net/ssh/test/channel.rb b/lib/net/ssh/test/channel.rb
index 1b58671..2eb7a12 100644
--- a/lib/net/ssh/test/channel.rb
+++ b/lib/net/ssh/test/channel.rb
@@ -98,6 +98,13 @@ module Net; module SSH; module Test
       script.sends_channel_close(self)
     end
 
+    # Scripts the sending of a "request pty" request packet across the channel.
+    #
+    #   channel.sends_request_pty
+    def sends_request_pty
+      script.sends_channel_request_pty(self)
+    end
+
     # Scripts the reception of a channel data packet from the remote end.
     #
     #   channel.gets_data "bar"
diff --git a/lib/net/ssh/test/packet.rb b/lib/net/ssh/test/packet.rb
index 60cf4e6..8767253 100644
--- a/lib/net/ssh/test/packet.rb
+++ b/lib/net/ssh/test/packet.rb
@@ -72,6 +72,7 @@ module Net; module SSH; module Test
           case @data[1]
           when "exec", "subsystem" then parts << :string
           when "exit-status" then parts << :long
+          when "pty-req" then parts += [:string, :long, :long, :long, :long, :string]
           else raise "don't know what to do about #{@data[1]} channel request"
           end
         else raise "don't know how to parse packet type #{@type}"
diff --git a/lib/net/ssh/test/script.rb b/lib/net/ssh/test/script.rb
index 701defc..d3813c3 100644
--- a/lib/net/ssh/test/script.rb
+++ b/lib/net/ssh/test/script.rb
@@ -104,6 +104,15 @@ module Net; module SSH; module Test
       events << LocalPacket.new(:channel_close, channel.remote_id)
     end
 
+    # Scripts the sending of a channel request pty packets from the given
+    # Net::SSH::Test::Channel +channel+. This will typically be called via
+    # Net::SSH::Test::Channel#sends_request_pty.
+    def sends_channel_request_pty(channel)
+      data = ['pty-req', false]
+      data += Net::SSH::Connection::Channel::VALID_PTY_OPTIONS.merge(:modes => "\0").values
+      events << LocalPacket.new(:channel_request, channel.remote_id, *data)
+    end
+
     # Scripts the reception of a channel data packet from the remote host by
     # the given Net::SSH::Test::Channel +channel+. This will typically be
     # called via Net::SSH::Test::Channel#gets_data.