#
# Author:: Matthew Kent (<mkent@magoazul.com>)
# Copyright:: Copyright (c) 2011 Opscode, Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require File.expand_path(File.dirname(__FILE__) + '/../../../spec_helper.rb')

describe Ohai::System, "Linux filesystem plugin" do
  before(:each) do
    @ohai = Ohai::System.new
    @ohai[:os] = "linux"
    @ohai.stub!(:require_plugin).and_return(true)
    @ohai.extend(SimpleFromFile)

    stdout = mock("STDOUT_SKEL")
    stderr = mock("STDERR_SKEL")
    stdout.stub!(:each_line).and_yield('')
    stderr.stub!(:each_line).and_yield('')

    @df_cmd = {
      :command => "df -P",
      :timeout => 120,
      :no_status_check => true
    }
    @mount_cmd = {
      :command => "mount",
      :timeout => 120,
    }

    @ohai.stub!(:run_command).with(@df_cmd).and_return([0,stdout,stderr])
    @ohai.stub!(:run_command).with(@mount_cmd).and_return([0,stdout,stderr])
    File.stub!(:exists?).with("/bin/lsblk").and_return(false)
    @ohai.stub!(:popen4).with("blkid -s TYPE").and_return(false)
    @ohai.stub!(:popen4).with("blkid -s UUID").and_return(false)
    @ohai.stub!(:popen4).with("blkid -s LABEL").and_return(false)

    @ohai.stub!(:popen4).with("lsblk -r -o NAME,FSTYPE -n").and_return(false)
    @ohai.stub!(:popen4).with("lsblk -r -o NAME,UUID -n").and_return(false)
    @ohai.stub!(:popen4).with("lsblk -r -o NAME,LABEL -n").and_return(false)

    File.stub!(:exists?).with("/proc/mounts").and_return(false)
  end

  describe "when gathering filesystem usage data from df" do
    before(:each) do
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each_line).
        and_yield("Filesystem         1024-blocks      Used Available Capacity Mounted on").
        and_yield("/dev/mapper/sys.vg-root.lv   4805760    378716   4182924       9% /").
        and_yield("tmpfs                  2030944         0   2030944       0% /lib/init/rw").
        and_yield("udev                   2025576       228   2025348       1% /dev").
        and_yield("tmpfs                  2030944      2960   2027984       1% /dev/shm").
        and_yield("/dev/mapper/sys.vg-home.lv  97605056  53563252  44041804      55% /home").
        and_yield("/dev/mapper/sys.vg-special.lv  97605057  53563253  44041805      56% /special").
        and_yield("/dev/mapper/sys.vg-tmp.lv   1919048     46588   1774976       3% /tmp").
        and_yield("/dev/mapper/sys.vg-usr.lv  19223252   5479072  12767696      31% /usr").
        and_yield("/dev/mapper/sys.vg-var.lv  19223252   3436556  14810212      19% /var").
        and_yield("/dev/md0                960492     36388    875312       4% /boot")
    end

    it "should run df -P" do
      @ohai.should_receive(:run_command).with(@df_cmd).and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from df -P" do
      @ohai.stub!(:run_command).with(@df_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:kb_size].should be == "97605057"
    end

    it "should set kb_used to value from df -P" do
      @ohai.stub!(:run_command).with(@df_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:kb_used].should be == "53563253"
    end

    it "should set kb_available to value from df -P" do
      @ohai.stub!(:run_command).with(@df_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:kb_available].should be == "44041805"
    end

    it "should set percent_used to value from df -P" do
      @ohai.stub!(:run_command).with(@df_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:percent_used].should be == "56%"
    end

    it "should set mount to value from df -P" do
      @ohai.stub!(:run_command).with(@df_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:mount].should be == "/special"
    end
  end

  describe "when gathering mounted filesystem data from mount" do
    before(:each) do
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each_line).
        and_yield("/dev/mapper/sys.vg-root.lv on / type ext4 (rw,noatime,errors=remount-ro)").
        and_yield("tmpfs on /lib/init/rw type tmpfs (rw,nosuid,mode=0755)").
        and_yield("proc on /proc type proc (rw,noexec,nosuid,nodev)").
        and_yield("sysfs on /sys type sysfs (rw,noexec,nosuid,nodev)").
        and_yield("udev on /dev type tmpfs (rw,mode=0755)").
        and_yield("tmpfs on /dev/shm type tmpfs (rw,nosuid,nodev)").
        and_yield("devpts on /dev/pts type devpts (rw,noexec,nosuid,gid=5,mode=620)").
        and_yield("/dev/mapper/sys.vg-home.lv on /home type xfs (rw,noatime)").
        and_yield("/dev/mapper/sys.vg-special.lv on /special type xfs (ro,noatime)").
        and_yield("/dev/mapper/sys.vg-tmp.lv on /tmp type ext4 (rw,noatime)").
        and_yield("/dev/mapper/sys.vg-usr.lv on /usr type ext4 (rw,noatime)").
        and_yield("/dev/mapper/sys.vg-var.lv on /var type ext4 (rw,noatime)").
        and_yield("/dev/md0 on /boot type ext3 (rw,noatime,errors=remount-ro)").
        and_yield("fusectl on /sys/fs/fuse/connections type fusectl (rw)").
        and_yield("binfmt_misc on /proc/sys/fs/binfmt_misc type binfmt_misc (rw,noexec,nosuid,nodev)")
    end

    it "should run mount" do
      @ohai.stub!(:run_command).with(@mount_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set mount to value from mount" do
      @ohai.stub!(:run_command).with(@mount_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:mount].should be == "/special"
    end

    it "should set fs_type to value from mount" do
      @ohai.stub!(:run_command).with(@mount_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:fs_type].should be == "xfs"
    end

    it "should set mount_options to an array of values from mount" do
      @ohai.stub!(:run_command).with(@mount_cmd).and_return([@status, @stdout, @stderr])
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:mount_options].should be == [ "ro", "noatime" ]
    end
  end

  describe "when gathering filesystem type data from blkid" do
    before(:each) do
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sdb1: TYPE=\"linux_raid_member\" ").
        and_yield("/dev/sdb2: TYPE=\"linux_raid_member\" ").
        and_yield("/dev/sda1: TYPE=\"linux_raid_member\" ").
        and_yield("/dev/sda2: TYPE=\"linux_raid_member\" ").
        and_yield("/dev/md0: TYPE=\"ext3\" ").
        and_yield("/dev/md1: TYPE=\"LVM2_member\" ").
        and_yield("/dev/mapper/sys.vg-root.lv: TYPE=\"ext4\" ").
        and_yield("/dev/mapper/sys.vg-swap.lv: TYPE=\"swap\" ").
        and_yield("/dev/mapper/sys.vg-tmp.lv: TYPE=\"ext4\" ").
        and_yield("/dev/mapper/sys.vg-usr.lv: TYPE=\"ext4\" ").
        and_yield("/dev/mapper/sys.vg-var.lv: TYPE=\"ext4\" ").
        and_yield("/dev/mapper/sys.vg-home.lv: TYPE=\"xfs\" ")
    end

    it "should run blkid -s TYPE" do
      @ohai.should_receive(:popen4).with("blkid -s TYPE").and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from blkid -s TYPE" do
      @ohai.stub!(:popen4).with("blkid -s TYPE").and_yield(@pid, @stdin, @stdout, @stderr).and_return(@status)
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/md1"][:fs_type].should be == "LVM2_member"
    end
  end

  describe "when gathering filesystem type data from lsblk" do
    before(:each) do
      File.stub!(:exists?).with('/bin/lsblk').and_return(true)
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sdb1 linux_raid_member").
        and_yield("/dev/sdb2 linux_raid_member").
        and_yield("/dev/sda1 linux_raid_member").
        and_yield("/dev/sda2 linux_raid_member").
        and_yield("/dev/md0 ext3").
        and_yield("/dev/md1 LVM2_member").
        and_yield("/dev/mapper/sys.vg-root.lv ext4").
        and_yield("/dev/mapper/sys.vg-swap.lv swap").
        and_yield("/dev/mapper/sys.vg-tmp.lv ext4").
        and_yield("/dev/mapper/sys.vg-usr.lv ext4").
        and_yield("/dev/mapper/sys.vg-var.lv ext4").
        and_yield("/dev/mapper/sys.vg-home.lv xfs")
    end

    it "should run lsblk -r -o NAME,FSTYPE -n" do
      @ohai.should_receive(:popen4).with("lsblk -r -o NAME,FSTYPE -n").
        and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end
  end

  describe "when gathering filesystem uuid data from blkid" do
    before(:each) do
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sdb1: UUID=\"bd1197e0-6997-1f3a-e27e-7801388308b5\" ").
        and_yield("/dev/sdb2: UUID=\"e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa\" ").
        and_yield("/dev/sda1: UUID=\"bd1197e0-6997-1f3a-e27e-7801388308b5\" ").
        and_yield("/dev/sda2: UUID=\"e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa\" ").
        and_yield("/dev/md0: UUID=\"37b8de8e-0fe3-4b5a-b9b4-dde33e19bb32\" ").
        and_yield("/dev/md1: UUID=\"YsIe0R-fj1y-LXTd-imla-opKo-OuIe-TBoxSK\" ").
        and_yield("/dev/mapper/sys.vg-root.lv: UUID=\"7742d14b-80a3-4e97-9a32-478be9ea9aea\" ").
        and_yield("/dev/mapper/sys.vg-swap.lv: UUID=\"9bc2e515-8ddc-41c3-9f63-4eaebde9ce96\" ").
        and_yield("/dev/mapper/sys.vg-tmp.lv: UUID=\"74cf7eb9-428f-479e-9a4a-9943401e81e5\" ").
        and_yield("/dev/mapper/sys.vg-usr.lv: UUID=\"26ec33c5-d00b-4f88-a550-492def013bbc\" ").
        and_yield("/dev/mapper/sys.vg-var.lv: UUID=\"6b559c35-7847-4ae2-b512-c99012d3f5b3\" ").
        and_yield("/dev/mapper/sys.vg-home.lv: UUID=\"d6efda02-1b73-453c-8c74-7d8dee78fa5e\" ")
    end

    it "should run blkid -s UUID" do
      @ohai.should_receive(:popen4).with("blkid -s UUID").and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from blkid -s UUID" do
      @ohai.stub!(:popen4).with("blkid -s UUID").and_yield(@pid, @stdin, @stdout, @stderr).and_return(@status)
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/sda2"][:uuid].should be == "e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa"
    end
  end

  describe "when gathering filesystem uuid data from lsblk" do
    before(:each) do
      File.stub!(:exists?).with('/bin/lsblk').and_return(true)
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sdb1 bd1197e0-6997-1f3a-e27e-7801388308b5").
        and_yield("/dev/sdb2 e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa").
        and_yield("/dev/sda1 bd1197e0-6997-1f3a-e27e-7801388308b5").
        and_yield("/dev/sda2 e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa").
        and_yield("/dev/md0 37b8de8e-0fe3-4b5a-b9b4-dde33e19bb32").
        and_yield("/dev/md1 YsIe0R-fj1y-LXTd-imla-opKo-OuIe-TBoxSK").
        and_yield("/dev/mapper/sys.vg-root.lv 7742d14b-80a3-4e97-9a32-478be9ea9aea").
        and_yield("/dev/mapper/sys.vg-swap.lv 9bc2e515-8ddc-41c3-9f63-4eaebde9ce96").
        and_yield("/dev/mapper/sys.vg-tmp.lv 74cf7eb9-428f-479e-9a4a-9943401e81e5").
        and_yield("/dev/mapper/sys.vg-usr.lv 26ec33c5-d00b-4f88-a550-492def013bbc").
        and_yield("/dev/mapper/sys.vg-var.lv 6b559c35-7847-4ae2-b512-c99012d3f5b3").
        and_yield("/dev/mapper/sys.vg-home.lv d6efda02-1b73-453c-8c74-7d8dee78fa5e")
    end

    it "should run lsblk -r -o NAME,UUID -n" do
      @ohai.should_receive(:popen4).with("lsblk -r -o NAME,UUID -n").
        and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from lsblk -r -o NAME,UUID -n" do
      @ohai.stub!(:popen4).with("lsblk -r -o NAME,UUID -n").
        and_yield(@pid, @stdin, @stdout, @stderr).and_return(@status)
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/sda2"][:uuid].should be == "e36d933e-e5b9-cfe5-6845-1f84d0f7fbfa"
    end
  end

  describe "when gathering filesystem label data from blkid" do
    before(:each) do
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sda1: LABEL=\"fuego:0\" ").
        and_yield("/dev/sda2: LABEL=\"fuego:1\" ").
        and_yield("/dev/sdb1: LABEL=\"fuego:0\" ").
        and_yield("/dev/sdb2: LABEL=\"fuego:1\" ").
        and_yield("/dev/md0: LABEL=\"/boot\" ").
        and_yield("/dev/mapper/sys.vg-root.lv: LABEL=\"/\" ").
        and_yield("/dev/mapper/sys.vg-tmp.lv: LABEL=\"/tmp\" ").
        and_yield("/dev/mapper/sys.vg-usr.lv: LABEL=\"/usr\" ").
        and_yield("/dev/mapper/sys.vg-var.lv: LABEL=\"/var\" ").
        and_yield("/dev/mapper/sys.vg-home.lv: LABEL=\"/home\" ")
    end

    it "should run blkid -s LABEL" do
      @ohai.should_receive(:popen4).with("blkid -s LABEL").and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from blkid -s LABEL" do
      @ohai.stub!(:popen4).with("blkid -s LABEL").and_yield(@pid, @stdin, @stdout, @stderr).and_return(@status)
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/md0"][:label].should be == "/boot"
    end
  end

  describe "when gathering filesystem label data from lsblk" do
    before(:each) do
      File.stub!(:exists?).with('/bin/lsblk').and_return(true)
      @stdin = mock("STDIN", { :close => true })
      @pid = 10
      @stderr = mock("STDERR")
      @stdout = mock("STDOUT")
      @status = 0

      @stdout.stub!(:each).
        and_yield("/dev/sda1 fuego:0").
        and_yield("/dev/sda2 fuego:1").
        and_yield("/dev/sdb1 fuego:0").
        and_yield("/dev/sdb2 fuego:1").
        and_yield("/dev/md0 /boot").
        and_yield("/dev/mapper/sys.vg-root.lv /").
        and_yield("/dev/mapper/sys.vg-tmp.lv /tmp").
        and_yield("/dev/mapper/sys.vg-usr.lv /usr").
        and_yield("/dev/mapper/sys.vg-var.lv /var").
        and_yield("/dev/mapper/sys.vg-home.lv /home")
    end

    it "should run lsblk -r -o NAME,LABEL -n" do
      @ohai.should_receive(:popen4).with("lsblk -r -o NAME,LABEL -n").
        and_return(true)
      @ohai._require_plugin("linux::filesystem")
    end

    it "should set kb_size to value from lsblk -r -o NAME,LABEL -n" do
      @ohai.stub!(:popen4).with("lsblk -r -o NAME,LABEL -n").
        and_yield(@pid, @stdin, @stdout, @stderr).and_return(@status)
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/md0"][:label].should be == "/boot"
    end
  end


  describe "when gathering data from /proc/mounts" do
    before(:each) do
      File.stub!(:exists?).with("/proc/mounts").and_return(true)
      @mock_file = mock("/proc/mounts")
      @mock_file.stub!(:read_nonblock).and_return(@mock_file)
      @mounts = <<-MOUNTS
rootfs / rootfs rw 0 0
none /sys sysfs rw,nosuid,nodev,noexec,relatime 0 0
none /proc proc rw,nosuid,nodev,noexec,relatime 0 0
none /dev devtmpfs rw,relatime,size=2025576k,nr_inodes=506394,mode=755 0 0
none /dev/pts devpts rw,nosuid,noexec,relatime,gid=5,mode=620,ptmxmode=000 0 0
/dev/mapper/sys.vg-root.lv / ext4 rw,noatime,errors=remount-ro,barrier=1,data=ordered 0 0
tmpfs /lib/init/rw tmpfs rw,nosuid,relatime,mode=755 0 0
tmpfs /dev/shm tmpfs rw,nosuid,nodev,relatime 0 0
/dev/mapper/sys.vg-home.lv /home xfs rw,noatime,attr2,noquota 0 0
/dev/mapper/sys.vg-special.lv /special xfs ro,noatime,attr2,noquota 0 0
/dev/mapper/sys.vg-tmp.lv /tmp ext4 rw,noatime,barrier=1,data=ordered 0 0
/dev/mapper/sys.vg-usr.lv /usr ext4 rw,noatime,barrier=1,data=ordered 0 0
/dev/mapper/sys.vg-var.lv /var ext4 rw,noatime,barrier=1,data=ordered 0 0
/dev/md0 /boot ext3 rw,noatime,errors=remount-ro,data=ordered 0 0
fusectl /sys/fs/fuse/connections fusectl rw,relatime 0 0
binfmt_misc /proc/sys/fs/binfmt_misc binfmt_misc rw,nosuid,nodev,noexec,relatime 0 0
MOUNTS
      @counter = 0
      @mock_file.stub(:read_nonblock) do
        @counter += 1
        raise EOFError if @counter == 2
        @mounts
      end
      @mock_file.stub(:close)
      File.stub!(:open).with("/proc/mounts").and_return(@mock_file)
    end

    it "should set mount to value from /proc/mounts" do
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:mount].should be == "/special"
    end
  
    it "should set fs_type to value from /proc/mounts" do
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:fs_type].should be == "xfs"
    end
  
    it "should set mount_options to an array of values from /proc/mounts" do
      @ohai._require_plugin("linux::filesystem")
      @ohai[:filesystem]["/dev/mapper/sys.vg-special.lv"][:mount_options].should be == [ "ro", "noatime", "attr2", "noquota" ]
    end
  end
end