require_relative "../test_helper"

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, 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(:identity_username) { ENV["SFTP_IDENTITY_USERNAME"] || username }

      let(:url) { File.join("sftp://", host_name, ftp_dir) }

      let(:file_name) { File.join(File.dirname(__FILE__), "..", "files", "text file.txt") }
      let(:raw) { File.read(file_name) }

      let(:root_path) do
        if ENV["SFTP_HOST_KEY"]
          IOStreams::Paths::SFTP.new(url, username: username, password: password, ssh_options: {"HostKey" => ENV["SFTP_HOST_KEY"]})
        else
          IOStreams::Paths::SFTP.new(url, username: username, password: password)
        end
      end

      let :existing_path do
        path = root_path.join("test.txt")
        path.write(raw)
        path
      end

      let :missing_path do
        root_path.join("unknown_path", "test_file.txt")
      end

      let :missing_file_path do
        root_path.join("test_file.txt")
      end

      let :write_path do
        root_path.join("writer_test.txt")
      end

      describe "#reader" do
        it "reads" do
          assert_equal raw, existing_path.read
        end

        it "fails when the file does not exist" do
          assert_raises IOStreams::Errors::CommunicationsFailure do
            missing_file_path.read
          end
        end

        it "fails when the directory does not exist" do
          assert_raises IOStreams::Errors::CommunicationsFailure do
            missing_path.read
          end
        end
      end

      describe "#writer" do
        it "writes" do
          assert_equal(raw.size, write_path.writer { |io| io.write(raw) })
          assert_equal raw, write_path.read
        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: identity_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: identity_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
end