Implement NetSSHStoryHelpers#sftp_story

2013-08-30 05:45:12 +00:00 · 2013-08-30 05:45:12 +00:00 · 8531c49e6d
commit 8531c49e6d
parent 0529fab3c2
1 changed files with 16 additions and 0 deletions
--- a/spec/support/net_ssh_story_helpers.rb
+++ b/spec/support/net_ssh_story_helpers.rb
@ -32,4 +32,20 @@ module NetSSHStoryHelpers
  def story_completed?
    socket.script.events.empty?
  end
+
+  def sftp_story
+    story do |session|
+      ch = session.opens_channel
+      ch.sends_subsystem('sftp')
+      ch.sends_packet(
+        Net::SFTP::Constants::PacketTypes::FXP_INIT, :long,
+        Net::SFTP::Session::HIGHEST_PROTOCOL_VERSION_SUPPORTED
+      )
+      ch.gets_packet(
+        Net::SFTP::Constants::PacketTypes::FXP_VERSION, :long,
+        Net::SFTP::Session::HIGHEST_PROTOCOL_VERSION_SUPPORTED
+      )
+      yield ch if block_given?
+    end
+  end
 end