From 8531c49e6d9b5f282ee6dfb8e6a4408deeec96cf Mon Sep 17 00:00:00 2001
From: Thibault Jouan <tj@a13.fr>
Date: Fri, 30 Aug 2013 05:45:12 +0000
Subject: [PATCH] Implement NetSSHStoryHelpers#sftp_story

---
 spec/support/net_ssh_story_helpers.rb | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/spec/support/net_ssh_story_helpers.rb b/spec/support/net_ssh_story_helpers.rb
index 949b408..bad0402 100644
--- 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