test/paths/sftp_test.rb in iostreams-1.0.0.beta3 vs test/paths/sftp_test.rb in iostreams-1.0.0.beta4

@@ -3,30 +3,26 @@
 module Paths
   class SFTPTest < Minitest::Test
     describe IOStreams::Paths::SFTP do
       before do
         unless ENV["SFTP_HOSTNAME"]
-          skip "Supply environment variables to test SFTP paths: SFTP_HOSTNAME, SFTP_USERNAME, SFTP_PASSWORD, and optional SFTP_DIR"
+          skip "Supply environment variables to test SFTP paths: SFTP_HOSTNAME, SFTP_USERNAME, SFTP_PASSWORD, and optional SFTP_DIR, SFTP_IDENTITY_FILE"
         end
       end
 
       let(:host_name) { ENV["SFTP_HOSTNAME"] }
       let(:username) { ENV["SFTP_USERNAME"] }
       let(:password) { ENV["SFTP_PASSWORD"] }
-      let(:ftp_dir) { ENV["SFTP_DIR"] || "iostreams_test"}
+      let(:ftp_dir) { ENV["SFTP_DIR"] || "iostreams_test" }
+
       let(:url) { File.join("sftp://", host_name, ftp_dir) }
 
-      let :file_name do
-        File.join(File.dirname(__FILE__), '..', 'files', 'text.txt')
-      end
+      let(:file_name) { File.join(File.dirname(__FILE__), '..', 'files', 'text.txt') }
+      let(:raw) { File.read(file_name) }
 
-      let :raw do
-        File.read(file_name)
-      end
+      let(:root_path) { IOStreams::Paths::SFTP.new(url, username: username, password: password) }
 
-      let(:root_path) { IOStreams::Paths::SFTP.new(url, username: username, password: password, ruby: false) }
-
       let :existing_path do
         path = root_path.join('test.txt')
         path.write(raw)
         path
       end
@@ -68,9 +64,34 @@
         end
 
         it 'fails when the directory does not exist' do
           assert_raises IOStreams::Errors::CommunicationsFailure do
             missing_path.write("Bad path")
+          end
+        end
+
+        describe 'use identity file instead of password' do
+          let :root_path do
+            IOStreams::Paths::SFTP.new(url, username: username, ssh_options: {'IdentityFile' => ENV["SFTP_IDENTITY_FILE"]} )
+          end
+
+          it 'writes' do
+            skip "No identity file env var set: SFTP_IDENTITY_FILE" unless ENV["SFTP_IDENTITY_FILE"]
+            assert_equal raw.size, write_path.writer { |io| io.write(raw) }
+            assert_equal raw, write_path.read
+          end
+        end
+
+        describe 'use identity key instead of password' do
+          let :root_path do
+            key = File.open(ENV["SFTP_IDENTITY_FILE"], 'rb', &:read)
+            IOStreams::Paths::SFTP.new(url, username: username, ssh_options: {'IdentityKey' => key})
+          end
+
+          it 'writes' do
+            skip "No identity file env var set: SFTP_IDENTITY_FILE" unless ENV["SFTP_IDENTITY_FILE"]
+            assert_equal raw.size, write_path.writer { |io| io.write(raw) }
+            assert_equal raw, write_path.read
           end
         end
       end
     end
   end