Puppet » Facter

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

require 'facter'

require 'facter/util/virtual'

require 'facter/util/macosx'

describe "Virtual fact" do

  before do

    Facter::Util::Virtual.stubs(:zone?).returns(false)

    Facter::Util::Virtual.stubs(:openvz?).returns(false)

    Facter::Util::Virtual.stubs(:vserver?).returns(false)

    Facter::Util::Virtual.stubs(:xen?).returns(false)

    Facter::Util::Virtual.stubs(:kvm?).returns(false)

    Facter::Util::Virtual.stubs(:hpvm?).returns(false)

    Facter::Util::Virtual.stubs(:zlinux?).returns(false)

  end

  it "should be zone on Solaris when a zone" do

    Facter.fact(:kernel).stubs(:value).returns("SunOS")

    Facter::Util::Virtual.stubs(:zone?).returns(true)

    Facter::Util::Virtual.stubs(:vserver?).returns(false)

    Facter::Util::Virtual.stubs(:xen?).returns(false)

    Facter.fact(:virtual).value.should == "zone"

  end

  it "should be jail on FreeBSD when a jail in kvm" do

    Facter.fact(:kernel).stubs(:value).returns("FreeBSD")

    Facter::Util::Virtual.stubs(:jail?).returns(true)

    Facter::Util::Virtual.stubs(:kvm?).returns(true)

    Facter.fact(:virtual).value.should == "jail"

  end

  it "should be hpvm on HP-UX when in HP-VM" do

   Facter.fact(:kernel).stubs(:value).returns("HP-UX")

   Facter::Util::Virtual.stubs(:hpvm?).returns(true)

   Facter.fact(:virtual).value.should == "hpvm"

  end

  it "should be zlinux on s390x" do

   Facter.fact(:kernel).stubs(:value).returns("Linux")

   Facter.fact(:architecture).stubs(:value).returns("s390x")

   Facter::Util::Virtual.stubs(:zlinux?).returns(true)

   Facter.fact(:virtual).value.should == "zlinux"

  end

  describe "on Darwin" do

    it "should be parallels with Parallels vendor id" do

      Facter.fact(:kernel).stubs(:value).returns("Darwin")

      Facter::Util::Macosx.stubs(:profiler_data).returns({ "spdisplays_vendor-id" => "0x1ab8" })

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be parallels with Parallels vendor name" do

      Facter.fact(:kernel).stubs(:value).returns("Darwin")

      Facter::Util::Macosx.stubs(:profiler_data).returns({ "spdisplays_vendor" => "Parallels" })

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be vmware with VMWare vendor id" do

      Facter.fact(:kernel).stubs(:value).returns("Darwin")

      Facter::Util::Macosx.stubs(:profiler_data).returns({ "spdisplays_vendor-id" => "0x15ad" })

      Facter.fact(:virtual).value.should == "vmware"

    end

    it "should be vmware with VMWare vendor name" do

      Facter.fact(:kernel).stubs(:value).returns("Darwin")

      Facter::Util::Macosx.stubs(:profiler_data).returns({ "spdisplays_vendor" => "VMWare" })

      Facter.fact(:virtual).value.should == "vmware"

    end

  end

  describe "on Linux" do

    before do

    Facter::Util::Resolution.stubs(:exec).with("vmware -v").returns false

    Facter.fact(:operatingsystem).stubs(:value).returns(true)

    # Ensure the tests don't fail on Xen

    FileTest.stubs(:exists?).with("/proc/sys/xen").returns false

    FileTest.stubs(:exists?).with("/sys/bus/xen").returns false

    FileTest.stubs(:exists?).with("/proc/xen").returns false

    Facter.fact(:architecture).stubs(:value).returns(true)

    end

    it "should be parallels with Parallels vendor id from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("01:00.0 VGA compatible controller: Unknown device 1ab8:4005")

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be parallels with Parallels vendor name from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("01:00.0 VGA compatible controller: Parallels Display Adapter")

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be vmware with VMware vendor name from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("00:0f.0 VGA compatible controller: VMware Inc [VMware SVGA II] PCI Display Adapter")

      Facter.fact(:virtual).value.should == "vmware"

    end

    it "should be virtualbox with VirtualBox vendor name from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("00:02.0 VGA compatible controller: InnoTek Systemberatung GmbH VirtualBox Graphics Adapter")

      Facter.fact(:virtual).value.should == "virtualbox"

    end

    it "should be vmware with VMWare vendor name from dmidecode" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns("On Board Device 1 Information\nType: Video\nStatus: Disabled\nDescription: VMware SVGA II")

      Facter.fact(:virtual).value.should == "vmware"

    end

    it "should be xenhvm with Xen HVM vendor name from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("00:03.0 Unassigned class [ff80]: XenSource, Inc. Xen Platform Device (rev 01)")

      Facter.fact(:virtual).value.should == "xenhvm"

    end

    it "should be xenhvm with Xen HVM vendor name from dmidecode" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns("System Information\nManufacturer: Xen\nProduct Name: HVM domU")

      Facter.fact(:virtual).value.should == "xenhvm"

    end

    it "should be parallels with Parallels vendor name from dmidecode" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns("On Board Device Information\nType: Video\nStatus: Disabled\nDescription: Parallels Video Adapter")

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be virtualbox with VirtualBox vendor name from dmidecode" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns("BIOS Information\nVendor: innotek GmbH\nVersion: VirtualBox\n\nSystem Information\nManufacturer: innotek GmbH\nProduct Name: VirtualBox\nFamily: Virtual Machine")

      Facter.fact(:virtual).value.should == "virtualbox"

    end

    it "should be hyperv with Microsoft vendor name from lspci" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns("00:08.0 VGA compatible controller: Microsoft Corporation Hyper-V virtual VGA")

      Facter.fact(:virtual).value.should == "hyperv"

    end

    it "should be hyperv with Microsoft vendor name from dmidecode" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns("System Information\nManufacturer: Microsoft Corporation\nProduct Name: Virtual Machine")

      Facter.fact(:virtual).value.should == "hyperv"

    end

  end

  describe "on Solaris" do

    before(:each) do

      Facter::Util::Resolution.stubs(:exec).with("vmware -v").returns false

    end

    it "should be vmware with VMWare vendor name from prtdiag" do

      Facter.fact(:kernel).stubs(:value).returns("SunOS")

      Facter.fact(:hardwaremodel).stubs(:value).returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('prtdiag').returns("System Configuration: VMware, Inc. VMware Virtual Platform")

      Facter.fact(:virtual).value.should == "vmware"

    end

    it "should be parallels with Parallels vendor name from prtdiag" do

      Facter.fact(:kernel).stubs(:value).returns("SunOS")

      Facter.fact(:hardwaremodel).stubs(:value).returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('prtdiag').returns("System Configuration: Parallels Virtual Platform")

      Facter.fact(:virtual).value.should == "parallels"

    end

    it "should be virtualbox with VirtualBox vendor name from prtdiag" do

      Facter.fact(:kernel).stubs(:value).returns("SunOS")

      Facter.fact(:hardwaremodel).stubs(:value).returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('lspci').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('dmidecode').returns(nil)

      Facter::Util::Resolution.stubs(:exec).with('prtdiag').returns("System Configuration: innotek GmbH VirtualBox")

      Facter.fact(:virtual).value.should == "virtualbox"

    end

    it "should be xen0 with xen dom0 files in /proc" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter.fact(:operatingsystem).stubs(:value).returns("Linux")

      Facter.fact(:hardwaremodel).stubs(:value).returns("i386")

      Facter::Util::Virtual.expects(:xen?).returns(true)

      FileTest.expects(:exists?).with("/proc/xen/xsd_kva").returns(true)

      Facter.fact(:virtual).value.should == "xen0"

    end

    it "should be xenu with xen domU files in /proc" do

      Facter.fact(:kernel).stubs(:value).returns("Linux")

      Facter.fact(:operatingsystem).stubs(:value).returns("Linux")

      Facter.fact(:hardwaremodel).stubs(:value).returns("i386")

      Facter::Util::Virtual.expects(:xen?).returns(true)

      FileTest.expects(:exists?).with("/proc/xen/xsd_kva").returns(false)

      FileTest.expects(:exists?).with("/proc/xen/capabilities").returns(true)

      Facter.fact(:virtual).value.should == "xenu"

    end

  end

end

describe "is_virtual fact" do

  it "should be virtual when running on xen" do

     Facter.fact(:kernel).stubs(:value).returns("Linux")

     Facter.fact(:virtual).stubs(:value).returns("xenu")

     Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be false when running on xen0" do

     Facter.fact(:kernel).stubs(:value).returns("Linux")

     Facter.fact(:virtual).stubs(:value).returns("xen0")

     Facter.fact(:is_virtual).value.should == "false"

  end

  it "should be true when running on xenhvm" do

     Facter.fact(:kernel).stubs(:value).returns("Linux")

     Facter.fact(:virtual).stubs(:value).returns("xenhvm")

     Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be false when running on physical" do

     Facter.fact(:kernel).stubs(:value).returns("Linux")

     Facter.fact(:virtual).stubs(:value).returns("physical")

     Facter.fact(:is_virtual).value.should == "false"

  end

  it "should be true when running on vmware" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("vmware")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on virtualbox" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("virtualbox")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on openvzve" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("openvzve")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on kvm" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("kvm")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running in jail" do

    Facter.fact(:kernel).stubs(:value).returns("FreeBSD")

    Facter.fact(:virtual).stubs(:value).returns("jail")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running in zone" do

    Facter.fact(:kernel).stubs(:value).returns("SunOS")

    Facter.fact(:virtual).stubs(:value).returns("zone")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on hp-vm" do

    Facter.fact(:kernel).stubs(:value).returns("HP-UX")

    Facter.fact(:virtual).stubs(:value).returns("hpvm")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on S390" do

    Facter.fact(:architecture).stubs(:value).returns("s390x")

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("zlinux")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be true when running on parallels" do

    Facter.fact(:kernel).stubs(:value).returns("Darwin")

    Facter.fact(:virtual).stubs(:value).returns("parallels")

    Facter.fact(:is_virtual).value.should == "true"

  end

  it "should be false on vmware_server" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("vmware_server")

    Facter.fact(:is_virtual).value.should == "false"

  end

  it "should be false on openvz host nodes" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("openvzhn")

    Facter.fact(:is_virtual).value.should == "false"

  end

  it "should be true when running on hyperv" do

    Facter.fact(:kernel).stubs(:value).returns("Linux")

    Facter.fact(:virtual).stubs(:value).returns("hyperv")

    Facter.fact(:is_virtual).value.should == "true"

  end

end