beaker-vmware 1.0.0 → 2.1.0

data/spec/mock_vsphere.rb

@@ -0,0 +1,286 @@
+require 'ostruct'
+
+class MockRbVmomiSnapshot
+  attr_accessor :name, :rootSnapshotList, :childSnapshotList
+
+  def initialize
+    @name = nil
+    @rootSnapshotList = []
+    @childSnapshotList = []
+  end
+
+  def print_nested_array(arg)
+    str = '[ '
+    arg.each do |arry|
+      if arry.is_a?(Array)
+        str += print_nested_array(arry)
+      elsif arry.is_a?(MockRbVmomiSnapshot)
+        str += arry.to_s + ', '
+      end
+    end
+    str + ' ]'
+  end
+
+  def to_s
+    "Snapshot(name: #{@name}, rootSnapshotlist: #{print_nested_array(@rootSnapshotList)}, childSnapshotList: #{print_nested_array(@childSnapshotList)})"
+  end
+
+  def snapshot
+    self
+  end
+end
+
+class MockRbVmomiVM
+  attr_accessor :snapshot, :name, :state
+
+  def info
+    self
+  end
+
+  def process_snaphash(snaphash)
+    shotlist = []
+    snaphash.each do |name, subsnaps|
+      new_snap = MockRbVmomiSnapshot.new
+      new_snap.name = name
+      new_snap.childSnapshotList = process_snaphash(subsnaps) if subsnaps.is_a?(Hash)
+      shotlist << new_snap
+    end
+    shotlist
+  end
+
+  def get_snapshot(name, snaplist = @snapshot.rootSnapshotList)
+    snapshot = nil
+    snaplist.each do |snap|
+      if snap.is_a?(Array)
+        snapshot = get_snapshot(snap, name)
+      elsif snap.name == name
+        snapshot = snap.snapshot
+      end
+    end
+    snapshot
+  end
+
+  def initialize(name, snaphash)
+    @name = name
+    @snapshot = MockRbVmomiSnapshot.new
+    @snapshot.name = name
+    @snapshot.rootSnapshotList = process_snaphash(snaphash)
+  end
+end
+
+class MockRbVmomiConnection
+  class CustomizationSpecManager
+    class CustomizationSpec
+      def spec
+        true
+      end
+    end
+
+    def initialize
+      @customizationspec = CustomizationSpec.new
+    end
+
+    def GetCustomizationSpec(_arg)
+      @customizationspec
+    end
+  end
+
+  class PropertyCollector
+    class Result
+      def initialize(name, object)
+        @name = name
+        @object = object
+      end
+
+      def val
+        @name
+      end
+
+      def obj
+        @object
+      end
+
+      def propSet
+        [self]
+      end
+    end
+
+    class ResultContainer
+      attr_accessor :token
+
+      def initialize
+        @results = []
+      end
+
+      def objects
+        @results
+      end
+
+      def add_object(obj)
+        @results << obj
+      end
+    end
+
+    def initialize
+      @results = ResultContainer.new
+    end
+
+    def RetrievePropertiesEx(_hash)
+      @results.token = true
+      @results
+    end
+
+    def ContinueRetrievePropertiesEx(_token)
+      @results.token = false
+      @results
+    end
+
+    def add_result(name, object)
+      @results.add_object(Result.new(name, object))
+    end
+
+    def WaitForUpdates(_arg)
+      result = OpenStruct.new
+      result.version = 'version'
+      result
+    end
+
+    def CreateFilter(_arg)
+      filter = OpenStruct.new
+      filter.DestroyPropertyFilter = true
+      filter
+    end
+  end
+
+  class ServiceInstance
+    class Datacenter
+      attr_accessor :vmFolder, :hostFolder
+
+      def initialize
+        @vmFolder = MockRbVmomi::VIM::Folder.new
+        @vmFolder.name = '/root'
+      end
+
+      def find_datastore(_arg)
+        true
+      end
+    end
+
+    def initialize
+      @datacenter = Datacenter.new
+    end
+
+    def find_datacenter(_dc)
+      @datacenter
+    end
+  end
+
+  class ServiceManager
+    class ViewManager
+      def CreateContainerView(hash)
+        @view = hash
+      end
+    end
+
+    def initialize
+      @customizationspecmanager = CustomizationSpecManager.new
+      @viewmanager = ViewManager.new
+    end
+
+    def customizationSpecManager
+      @customizationspecmanager
+    end
+
+    def viewManager
+      @viewmanager
+    end
+
+    def rootFolder
+      '/root'
+    end
+  end
+
+  def initialize(opts)
+    @host = opts[:host]
+    @user = opts[:user]
+    @password = opts[:password]
+    @insecure = opts[:insecure]
+    @serviceinstance = ServiceInstance.new
+    @servicemanager = ServiceManager.new
+    @propertycollector = PropertyCollector.new
+  end
+
+  def serviceInstance
+    @serviceinstance
+  end
+
+  def serviceContent
+    @servicemanager
+  end
+
+  def propertyCollector
+    @propertycollector
+  end
+
+  def set_info(vms)
+    vms.each do |vm|
+      @propertycollector.add_result(vm.name, vm)
+    end
+  end
+end
+
+class MockRbVmomi
+  class VIM
+    class Folder
+      attr_accessor :name
+
+      def find
+        self
+      end
+
+      def childEntity
+        self
+      end
+
+      def resourcePool
+        self
+      end
+
+      def traverse(_path, _type = Object, _create = false)
+        self
+      end
+    end
+
+    class ResourcePool
+      attr_accessor :name
+
+      def find
+        self
+      end
+
+      def resourcePool
+        self
+      end
+    end
+
+    class ClusterComputeResource
+      attr_accessor :name
+
+      def find
+        self
+      end
+
+      def resourcePool
+        self
+      end
+    end
+
+    class TraversalSpec
+      def initialize(hash); end
+    end
+
+    def self.connect(opts)
+      MockRbVmomiConnection.new(opts)
+    end
+  end
+end

data/spec/mock_vsphere_helper.rb

@@ -0,0 +1,167 @@
+class MockVsphereSnapshot
+  attr_accessor :name
+
+  def RevertToSnapshot_Task
+    self
+  end
+
+  def wait_for_completion
+    true
+  end
+end
+
+class MockVsphereVM
+  attr_accessor :name, :powerState, :snapshot, :template, :toolsRunningStatus
+
+  def initialize
+    @powerState = 'poweredOff'
+    @toolsRunningStatus = 'guestToolsStopped'
+    @first = true
+  end
+
+  def runtime
+    self
+  end
+
+  attr_reader :powerState
+
+  def toolsRunningStatus
+    prev = @toolsRunningStatus
+    @toolsRunningStatus = 'guestToolsRunning'
+    prev
+  end
+
+  def PowerOnVM_Task
+    @powerState = 'poweredOn'
+    self
+  end
+
+  def PowerOffVM_Task
+    @powerState = 'poweredOff'
+    self
+  end
+
+  def summary
+    self
+  end
+
+  def guest
+    self
+  end
+
+  def ipAddress
+    @toolsRunningStatus = 'guestToolsRunning'
+    "#{@name}.ip.address"
+  end
+
+  def CloneVM_Task(opts)
+    clone = MockVsphereVM.new
+    clone.name = opts[:name]
+    clone.snapshot = snapshot
+    clone.template = template
+    clone.PowerOnVM_Task
+    MockVsphereHelper.add_vm(opts[:name], clone)
+    clone
+  end
+
+  def wait_for_completion
+    true
+  end
+
+  def Destroy_Task
+    true
+  end
+end
+
+class MockVsphereHelper
+  @@fog_file = {}
+  @@vms = {}
+
+  def initialize(arg); end
+
+  def self.add_vm(name, vm)
+    @@vms[name] = vm
+  end
+
+  def self.set_config(conf)
+    @@fog_file = conf
+  end
+
+  def self.powerOn
+    @@vms.each_value do |vm|
+      vm.powerState = 'poweredOn'
+    end
+  end
+
+  def self.powerOff
+    @@vms.each_value do |vm|
+      vm.powerState = 'poweredOff'
+    end
+  end
+
+  def self.set_vms(hosts)
+    @@vms = {}
+    hosts.each do |host|
+      vm = MockVsphereVM.new
+      vm.name = host.name
+      vm.snapshot = MockVsphereSnapshot.new
+      vm.snapshot.name = host[:snapshot]
+      @@vms[host.name] = vm
+      template = MockVsphereVM.new
+      template.name = host[:template]
+      template.snapshot = MockVsphereSnapshot.new
+      template.snapshot.name = host[:snapshot]
+      @@vms[host[:template]] = template
+    end
+  end
+
+  def self.load_config(_file)
+    @@fog_file
+  end
+
+  def find_vms(keys)
+    found = {}
+    keys = ([] << keys)
+    keys.flatten!
+    keys.each do |key|
+      found[key] = @@vms[key] if @@vms.has_key?(key)
+    end
+    found
+  end
+
+  def self.find_vm(key)
+    return unless @@vms.has_key?(key)
+
+    @@vms[key]
+  end
+
+  def find_snapshot(vm, snap)
+    return unless @@vms[vm.name].snapshot.name == snap
+
+    @@vms[vm.name].snapshot
+  end
+
+  def find_customization(_template)
+    nil
+  end
+
+  def find_datastore(_dc, datastore)
+    datastore
+  end
+
+  def find_pool(_dc, pool)
+    pool
+  end
+
+  def find_folder(_dc, folder)
+    folder
+  end
+
+  def wait_for_tasks(_tasks, _try, _attempts)
+    true
+  end
+
+  def close
+    true
+  end
+end

data/spec/spec_helper.rb

@@ -1,19 +1,24 @@
+require 'beaker'
+require 'mock_fission'
+require 'mock_vsphere'
+require 'mock_vsphere_helper'
+
 begin
   require 'simplecov'
   require 'simplecov-console'
   require 'codecov'
 rescue LoadError
+  # Do nothing if no required gem installed
 else
   SimpleCov.start do
     track_files 'lib/**/*.rb'
 
     add_filter '/spec'
-
-    enable_coverage :branch
-
     # do not track vendored files
     add_filter '/vendor'
     add_filter '/.vendor'
+
+    enable_coverage :branch
   end
 
   SimpleCov.formatters = [
@@ -22,10 +27,7 @@ else
   ]
 end
 
-require 'rspec/its'
-require 'beaker'
-
-Dir.glob(Dir.pwd + '/lib/beaker/hypervisor/*.rb') {|file| require file}
+Dir['./lib/beaker/hypervisor/*.rb'].sort.each { |file| require file }
 
 # setup & require beaker's spec_helper.rb
 beaker_gem_spec = Gem::Specification.find_by_name('beaker')