beaker 0.0.0 → 1.0.0

data/spec/helpers.rb

@@ -0,0 +1,97 @@
+module TestFileHelpers
+  def create_files file_array
+    file_array.each do |f|
+      FileUtils.mkdir_p File.dirname(f)
+      FileUtils.touch f
+    end
+  end
+
+  def fog_file_contents
+    { :default => { :aws_access_key_id => "IMANACCESSKEY", 
+                    :aws_secret_access_key => "supersekritkey", 
+                    :aix_hypervisor_server => "aix_hypervisor.labs.net", 
+                    :aix_hypervisor_username => "aixer", 
+                    :aix_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-acceptance", 
+                    :solaris_hypervisor_server => "solaris_hypervisor.labs.net", 
+                    :solaris_hypervisor_username => "harness", 
+                    :solaris_hypervisor_keyfile => "/Users/user/.ssh/id_rsa-old.private", 
+                    :solaris_hypervisor_vmpath => "rpoooool/zs", 
+                    :solaris_hypervisor_snappaths => ["rpoooool/USER/z0"], 
+                    :vsphere_server => "vsphere.labs.net", 
+                    :vsphere_username => "vsphere@labs.com", 
+                    :vsphere_password => "supersekritpassword"} }
+  end
+
+end
+
+module HostHelpers
+  HOST_DEFAULTS = { :platform => 'unix',
+                    :snapshot => 'pe',
+                    :box => 'box_name',
+                    :roles => ['agent'],
+                    :snapshot => 'snap',
+                    :ip => 'default.ip.address',
+                    :box => 'default_box_name',
+                    :box_url => 'http://default.box.url',
+  }
+
+  HOST_NAME     = "vm%d"
+  HOST_SNAPSHOT = "snapshot%d"
+  HOST_IP       = "ip.address.for.%s"
+  HOST_BOX      = "%s_of_my_box"
+  HOST_BOX_URL  = "http://address.for.my.box.%s"
+  HOST_TEMPLATE = "%s_has_a_template"
+
+  def logger
+    double( 'logger' ).as_null_object
+  end
+
+  def make_opts
+     Beaker::Options::Presets.presets.merge( { :logger => logger, 
+                                               :host_config => 'sample.config',
+                                               :type => :foss,
+                                               :pooling_api => 'http://vcloud.delivery.puppetlabs.net/',
+                                               :datastore => 'instance0',
+                                               :folder => 'Delivery/Quality Assurance/Staging/Dynamic',
+                                               :resourcepool => 'delivery/Quality Assurance/Staging/Dynamic' } )
+  end
+
+  def generate_result (name, opts )
+    result = double( 'result' )
+    stdout = opts.has_key?(:stdout) ? opts[:stdout] :  name
+    stderr = opts.has_key?(:stderr) ? opts[:stderr] : name
+    exit_code = opts.has_key?(:exit_code) ? opts[:exit_code] :  0
+    result.stub( :stdout ).and_return( stdout )
+    result.stub( :stderr ).and_return( stderr )
+    result.stub( :exit_code ).and_return( exit_code )
+    result
+  end
+
+  def make_host_opts name, opts
+    make_opts.merge( { 'HOSTS' => { name => opts } } ).merge( opts )
+  end
+
+  def make_host name, opts
+    opts = HOST_DEFAULTS.merge(opts) 
+
+    host = Beaker::Host.create( name, make_host_opts(name, opts) )
+
+    host.stub( :exec ).and_return( generate_result( name, opts ) )
+    host
+  end
+
+  def make_hosts preset_opts = {}, amt = 3
+    hosts = []
+    (1..amt).each do |num|
+      name = HOST_NAME % num
+      opts = { :snapshot => HOST_SNAPSHOT % num,
+               :ip => HOST_IP % name,
+               :template => HOST_TEMPLATE % name,
+               :box => HOST_BOX % name,
+               :box_url => HOST_BOX_URL % name }.merge( preset_opts )
+      hosts << make_host(name, opts)
+    end
+    hosts
+  end
+
+end

data/spec/matchers.rb

@@ -0,0 +1,39 @@
+RSpec::Matchers.define :execute_commands_matching do |pattern|
+
+  match do |actual|
+    raise(RuntimeError, "Expected #{actual} to be a FakeHost") unless actual.kind_of?(FakeHost)
+    @found_count = actual.command_strings.grep(pattern).size
+    @times.nil? ?
+      @found_count > 0 :
+      @found_count == @times
+  end
+
+  chain :exactly do |times|
+    @times = times
+  end
+
+  chain :times do
+    # clarity only
+  end
+
+  chain :once do
+    @times = 1
+    # clarity only
+  end
+
+  def message(actual, pattern, times, found_count)
+      msg = times == 1 ?
+        "#{pattern} once" :
+        "#{pattern} #{times} times"
+      msg += " but instead found a count of #{found_count}" if found_count != times
+      msg + " in:\n #{actual.command_strings.pretty_inspect}"
+  end
+
+  failure_message_for_should do |actual|
+    "Expected to find #{message(actual, pattern, @times, @found_count)}"
+  end
+
+  failure_message_for_should_not do |actual|
+    "Unexpectedly found #{message(actual, pattern, @times, @found_count)}"
+  end
+end

data/spec/mock_blimpy.rb

@@ -0,0 +1,48 @@
+class MockShip 
+  attr_accessor :name, :ports, :image_id, :flavor, :region, :username, :dns
+
+  def initialize
+    @dns = "my.ip.address"
+  end
+
+  def wait_for_sshd
+    @dns = "#{@name}.my.ip"
+    return true
+  end
+
+end
+
+class MockFleet
+  attr_accessor :ships
+  @@attempts = 0
+
+  def initialize
+    @ships = []
+  end
+
+  def add type
+    @ships << MockShip.new
+    yield(@ships[-1])
+    @ships[-1]
+  end
+
+  def start
+    if @@attempts < 1
+      @@attempts += 1
+      raise Fog::Errors::Error
+    end
+  end
+
+  def destroy
+  end
+
+end
+
+class MockBlimpy 
+  @@fleet = nil
+  def self.fleet 
+    yield(@@fleet = MockFleet.new)
+    @@fleet
+  end
+
+end

data/spec/mock_fission.rb

@@ -0,0 +1,60 @@
+class Response
+  attr_accessor :code, :message, :data
+  def initialize(code = 0, message = '', data = nil)
+    @code = code
+    @message = message
+    @data = data
+  end
+end
+
+class MockFissionVM 
+  attr_accessor :name
+  @@snaps = []
+  def initialize name
+    @name = name
+    @running = true
+  end
+
+  def self.set_snapshots snaps
+    @@snaps = snaps
+  end
+
+  def snapshots
+    Response.new(0, '', @@snaps)
+  end
+
+  def revert_to_snapshot name
+    @running = false
+  end
+
+  def running?
+    Response.new(0, '', @running) 
+  end
+
+  def start opt
+    @running = true 
+  end
+
+  def exists?
+    true
+  end
+end
+
+class MockFission 
+  @@vms = []
+  def self.presets hosts
+    snaps = []
+    hosts.each do |host|
+      @@vms << MockFissionVM.new( host.name )
+      snaps << host[ :snapshot ]
+    end
+    MockFissionVM.set_snapshots(snaps)
+  end
+
+  def self.all
+    Response.new(0, '', @@vms)
+  end
+  def self.new name
+    MockFissionVM.new(name)
+  end
+end

data/spec/mock_vsphere.rb

@@ -0,0 +1,310 @@
+class MockRbVmomiSnapshot
+  attr_accessor :name
+  attr_accessor :rootSnapshotList
+  attr_accessor :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
+      if subsnaps.is_a?(Hash) 
+        new_snap.childSnapshotList = process_snaphash( subsnaps )
+      end
+      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
+      attr_accessor :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
+      @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
+
+    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