require 'spec_helper'

describe LinuxAdmin::LogicalVolume do
  before(:each) do
    LinuxAdmin::Distro.stub(:local => LinuxAdmin::Distros::Test.new)

    @logical_volumes = <<eos
  /dev/vg_foobar/lv_swap:vg_foobar:3:1:-1:2:4128768:63:-1:0:-1:253:0
  /dev/vg_foobar/lv_root:vg_foobar:3:1:-1:1:19988480:305:-1:0:-1:253:1
eos

    @groups = <<eos
  vg_foobar:r/w:772:-1:0:2:2:-1:0:1:1:12058624:32768:368:368:0:tILZUF-IspH-H90I-pT5j-vVFl-b76L-zWx3CW
eos
  end

  after(:each) do
    # reset local copies of volumes / groups
    described_class.instance_variable_set(:@lvs, nil)
    LinuxAdmin::PhysicalVolume.instance_variable_set(:@pvs, nil)
    LinuxAdmin::VolumeGroup.instance_variable_set(:@vgs, nil)
  end

  describe "#extend_with" do
    it "uses lvextend" do
      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
      lv = described_class.new :name => 'lv', :volume_group => vg
      lv.should_receive(:run!).
         with(vg.cmd(:lvextend),
              :params => ['lv', 'vg'])
      lv.extend_with(vg)
    end

    it "returns self" do
      vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
      lv = described_class.new :name => 'lv', :volume_group => vg
      lv.stub(:run!)
      lv.extend_with(vg).should == lv
    end
  end

  describe "#create" do
    before(:each) do
      @vg = LinuxAdmin::VolumeGroup.new :name => 'vg'
    end

    it "uses lvcreate" do
      described_class.instance_variable_set(:@lvs, [])
      described_class.should_receive(:run!).
                                with(LinuxAdmin.cmd(:lvcreate),
                                     :params => { '-n' => 'lv',
                                                   nil => 'vg',
                                                  '-L' => '256G' })
      described_class.create 'lv', @vg, 256.gigabytes
    end

    context "size is specified" do
      it "passes -L option to lvcreate" do
        described_class.instance_variable_set(:@lvs, [])
        described_class.should_receive(:run!).
                                  with(LinuxAdmin.cmd(:lvcreate),
                                       :params => { '-n' => 'lv',
                                                     nil => 'vg',
                                                    '-L' => '256G' })
        described_class.create 'lv', @vg, 256.gigabytes
      end
    end

    context "extents is specified" do
      it "passes -l option to lvcreate" do
        described_class.instance_variable_set(:@lvs, [])
        described_class.should_receive(:run!).
                                  with(LinuxAdmin.cmd(:lvcreate),
                                       :params => { '-n' => 'lv',
                                                     nil => 'vg',
                                                    '-l' => '100%FREE' })
        described_class.create 'lv', @vg, 100
      end
    end

    it "returns new logical volume" do
      LinuxAdmin::VolumeGroup.stub(:run! => double(:output => ""))
      described_class.stub(:run! => double(:output => ""))
      lv = described_class.create 'lv', @vg, 256.gigabytes
      lv.should be_an_instance_of(described_class)
      lv.name.should == 'lv'
    end

    context "name is specified" do
      it "sets path under volume group" do
        LinuxAdmin::VolumeGroup.stub(:run! => double(:output => ""))
        described_class.stub(:run! => double(:output => ""))
        lv = described_class.create 'lv', @vg, 256.gigabytes
        lv.path.to_s.should == "#{described_class::DEVICE_PATH}#{@vg.name}/lv"
      end
    end

    context "path is specified" do
      it "sets name" do
        LinuxAdmin::VolumeGroup.stub(:run! => double(:output => ""))
        described_class.stub(:run! => double(:output => ""))
        lv = described_class.create '/dev/lv', @vg, 256.gigabytes
        lv.name.should == "lv"
      end
    end

    it "adds logical volume to local registry" do
      LinuxAdmin::VolumeGroup.stub(:run! => double(:output => ""))
      described_class.stub(:run! => double(:output => ""))
      lv = described_class.create 'lv', @vg, 256.gigabytes
      described_class.scan.should include(lv)
    end
  end

  describe "#scan" do
    it "uses lvdisplay" do
      described_class.should_receive(:run!).
                                with(LinuxAdmin.cmd(:lvdisplay),
                                     :params => { '-c' => nil}).
                                and_return(double(:output => @logical_volumes))
      LinuxAdmin::VolumeGroup.should_receive(:run!).and_return(double(:output => @groups)) # stub out call to vgdisplay
      described_class.scan
    end

    it "returns local logical volumes" do
      described_class.should_receive(:run!).and_return(double(:output => @logical_volumes))
      LinuxAdmin::VolumeGroup.should_receive(:run!).and_return(double(:output => @groups))
      lvs = described_class.scan

      lvs[0].should be_an_instance_of(described_class)
      lvs[0].path.should == '/dev/vg_foobar/lv_swap'
      lvs[0].name.should == 'lv_swap'
      lvs[0].sectors.should == 4128768

      lvs[1].should be_an_instance_of(described_class)
      lvs[1].path.should == '/dev/vg_foobar/lv_root'
      lvs[1].name.should == 'lv_root'
      lvs[1].sectors.should == 19988480
    end

    it "resolves volume group references" do
      described_class.should_receive(:run!).and_return(double(:output => @logical_volumes))
      LinuxAdmin::VolumeGroup.should_receive(:run!).and_return(double(:output => @groups))
      lvs = described_class.scan
      lvs[0].volume_group.should be_an_instance_of(LinuxAdmin::VolumeGroup)
      lvs[0].volume_group.name.should == 'vg_foobar'
      lvs[1].volume_group.should be_an_instance_of(LinuxAdmin::VolumeGroup)
      lvs[1].volume_group.name.should == 'vg_foobar'
    end
  end
end