require "helper"
require "train/file/local/unix"
require "train/transports/mock"
require "train/transports/local"

class FileTester < Train::File::Local::Unix
  def type
    :file
  end
end

describe Train::File::Local::Unix do
  let(:cls) { Train::File::Local::Unix }

  let(:backend) do
    backend = Train::Transports::Mock.new.connection
    backend.mock_os({ family: "linux" })
    backend
  end

  it "checks a mounted path" do
    backend.mock_command("mount | grep -- ' on /mount/path '", rand.to_s)
    _(cls.new(backend, "/mount/path").mounted?).must_equal true
  end

  describe "file metadata" do
    before do
      # something about the way meta_stub works blows out on windows
      skip "not on windows" if windows?
    end

    # there is zero need to instantiate this OVER and over, so just do it once.
    transport  = Train::Transports::Local.new
    connection = transport.connection

    let(:stat) { Struct.new(:mode, :size, :mtime, :uid, :gid) }
    let(:uid) { rand }
    let(:gid) { rand }
    let(:statres) { stat.new(00140755, rand, (rand * 100).to_i, uid, gid) }

    def meta_stub(method, param, &block)
      pwres = Struct.new(:name)
      Etc.stub :getpwuid, pwres.new("owner") do
        Etc.stub :getgrgid, pwres.new("group") do
          File.stub method, param do
            yield
          end
        end
      end
    end

    it "recognizes type" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:type]).must_equal :socket
      end
    end

    it "recognizes mode" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:mode]).must_equal 00755
      end
    end

    it "recognizes mtime" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:mtime]).must_equal statres.mtime
      end
    end

    it "recognizes size" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:size]).must_equal statres.size
      end
    end

    it "recognizes uid" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:uid]).must_equal uid
      end
    end

    it "recognizes gid" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).stat[:gid]).must_equal gid
      end
    end

    it "recognizes owner" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).owner).must_equal "owner"
      end
    end

    it "recognizes group" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).group).must_equal "group"
      end
    end

    it "grouped_into" do
      meta_stub :stat, statres do
        _(connection.file(rand.to_s).grouped_into?("group")).must_equal true
      end
    end

    it "recognizes selinux label" do
      meta_stub :stat, statres do
        label = rand.to_s
        res = Train::Extras::CommandResult.new(label, nil, 0)
        connection.stub :run_command, res do
          _(connection.file(rand.to_s).selinux_label).must_equal label
        end
      end
    end

    it "recognizes source selinux label" do
      meta_stub :lstat, statres do
        label = rand.to_s
        res = Train::Extras::CommandResult.new(label, nil, 0)
        connection.stub :run_command, res do
          _(connection.file(rand.to_s).source.selinux_label).must_equal label
        end
      end
    end
  end

  describe "#unix_mode_mask" do
    let(:file_tester) do
      FileTester.new(nil, nil, false)
    end

    it "check owner mode calculation" do
      _(file_tester.unix_mode_mask("owner", "x")).must_equal 0100
      _(file_tester.unix_mode_mask("owner", "w")).must_equal 0200
      _(file_tester.unix_mode_mask("owner", "r")).must_equal 0400
    end

    it "check group mode calculation" do
      _(file_tester.unix_mode_mask("group", "x")).must_equal 0010
      _(file_tester.unix_mode_mask("group", "w")).must_equal 0020
      _(file_tester.unix_mode_mask("group", "r")).must_equal 0040
    end

    it "check other mode calculation" do
      _(file_tester.unix_mode_mask("other", "x")).must_equal 0001
      _(file_tester.unix_mode_mask("other", "w")).must_equal 0002
      _(file_tester.unix_mode_mask("other", "r")).must_equal 0004
    end

    it "check all mode calculation" do
      _(file_tester.unix_mode_mask("all", "x")).must_equal 0111
      _(file_tester.unix_mode_mask("all", "w")).must_equal 0222
      _(file_tester.unix_mode_mask("all", "r")).must_equal 0444
    end
  end

  describe "#md5sum" do
    let(:md5_checksum) { "57d4c6f9d15313fd5651317e588c035d" }

    let(:ruby_md5_mock) do
      checksum_mock = mock
      checksum_mock.expects(:update).returns("")
      checksum_mock.expects(:hexdigest).returns(md5_checksum)
      checksum_mock
    end

    it "defaults to a Ruby based checksum if other methods fail" do
      backend.mock_command("md5sum /tmp/testfile", "", "", 1)
      Digest::MD5.expects(:new).returns(ruby_md5_mock)
      _(cls.new(backend, "/tmp/testfile").md5sum).must_equal md5_checksum
    end

    it "calculates the correct md5sum on the `linux` platform family" do
      output = "#{md5_checksum} /tmp/testfile"
      backend.mock_command("md5sum /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").md5sum).must_equal md5_checksum
    end

    it "calculates the correct md5sum on the `darwin` platform family" do
      output = "#{md5_checksum} /tmp/testfile"
      backend.mock_os(family: "darwin")
      backend.mock_command("md5 -r /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").md5sum).must_equal md5_checksum
    end

    it "calculates the correct md5sum on the `solaris` platform family" do
      # The `digest` command doesn't output the filename by default
      output = "#{md5_checksum}"
      backend.mock_os(family: "solaris")
      backend.mock_command("digest -a md5 /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").md5sum).must_equal md5_checksum
    end
  end

  describe "#sha256sum" do
    let(:sha256_checksum) do
      "491260aaa6638d4a64c714a17828c3d82bad6ca600c9149b3b3350e91bcd283d"
    end

    let(:ruby_sha256_mock) do
      checksum_mock = mock
      checksum_mock.expects(:update).returns("")
      checksum_mock.expects(:hexdigest).returns(sha256_checksum)
      checksum_mock
    end

    it "defaults to a Ruby based checksum if other methods fail" do
      backend.mock_command("sha256sum /tmp/testfile", "", "", 1)
      Digest::SHA256.expects(:new).returns(ruby_sha256_mock)
      _(cls.new(backend, "/tmp/testfile").sha256sum).must_equal sha256_checksum
    end

    it "calculates the correct sha256sum on the `linux` platform family" do
      output = "#{sha256_checksum} /tmp/testfile"
      backend.mock_command("sha256sum /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").sha256sum).must_equal sha256_checksum
    end

    it "calculates the correct sha256sum on the `darwin` platform family" do
      output = "#{sha256_checksum} /tmp/testfile"
      backend.mock_os(family: "darwin")
      backend.mock_command("shasum -a 256 /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").sha256sum).must_equal sha256_checksum
    end

    it "calculates the correct sha256sum on the `solaris` platform family" do
      # The `digest` command doesn't output the filename by default
      output = "#{sha256_checksum}"
      backend.mock_os(family: "solaris")
      backend.mock_command("digest -a sha256 /tmp/testfile", output)
      _(cls.new(backend, "/tmp/testfile").sha256sum).must_equal sha256_checksum
    end
  end
end