Class: Beaker::Vcloud

Inherits:
Hypervisor show all
Defined in:
lib/beaker/hypervisor/vcloud.rb

Instance Method Summary collapse

Methods inherited from Hypervisor

#configure, create

Constructor Details

#initialize(vcloud_hosts, options, config) ⇒ Vcloud

Returns a new instance of Vcloud.



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/beaker/hypervisor/vcloud.rb', line 4

def initialize(vcloud_hosts, options, config)
  @options = options
  @config = config['CONFIG'].dup
  @logger = options[:logger]
  @vcloud_hosts = vcloud_hosts
  require 'yaml' unless defined?(YAML)

  raise 'You must specify a datastore for vCloud instances!' unless @config['datastore']
  raise 'You must specify a resource pool for vCloud instances!' unless @config['resourcepool']
  raise 'You must specify a folder for vCloud instances!' unless @config['folder']

  vsphere_credentials = VsphereHelper.load_config

  @logger.notify "Connecting to vSphere at #{vsphere_credentials[:server]}" +
    " with credentials for #{vsphere_credentials[:user]}"

  vsphere_helper = VsphereHelper.new( vsphere_credentials )
  vsphere_vms = {}

  attempts = 10

  start = Time.now
  @vcloud_hosts.each_with_index do |h, i|
    # Generate a randomized hostname
    o = [('a'..'z'),('0'..'9')].map{|r| r.to_a}.flatten
    h['vmhostname'] = o[rand(25)] + (0...14).map{o[rand(o.length)]}.join

    if h['template'] =~ /\//
      templatefolders = h['template'].split('/')
      h['template'] = templatefolders.pop
    end

    @logger.notify "Deploying #{h['vmhostname']} (#{h.name}) to #{@config['folder']} from template '#{h['template']}'"

    vm = {}

    if templatefolders
      vm[h['template']] = vsphere_helper.find_folder(templatefolders.join('/')).find(h['template'])
    else
      vm = vsphere_helper.find_vms(h['template'])
    end

    if vm.length == 0
      raise "Unable to find template '#{h['template']}'!"
    end

    # Add VM annotation
    configSpec = RbVmomi::VIM.VirtualMachineConfigSpec(
      :annotation =>
        'Base template:  ' + h['template'] + "\n" +
        'Creation time:  ' + Time.now.strftime("%Y-%m-%d %H:%M") + "\n\n" +
        'CI build link:  ' + ( ENV['BUILD_URL'] || 'Deployed independently of CI' )
    )

    # Are we using a customization spec?
    customizationSpec = vsphere_helper.find_customization( h['template'] )

    if customizationSpec
      # Print a logger message if using a customization spec
      @logger.notify "Found customization spec for '#{h['template']}', will apply after boot"

      # Using a customization spec takes longer, set a longer timeout
      attempts = attempts * 2
    end

    # Put the VM in the specified folder and resource pool
    relocateSpec = RbVmomi::VIM.VirtualMachineRelocateSpec(
      :datastore    => vsphere_helper.find_datastore(@config['datastore']),
      :pool         => vsphere_helper.find_pool(@config['resourcepool']),
      :diskMoveType => :moveChildMostDiskBacking
    )

    # Create a clone spec
    spec = RbVmomi::VIM.VirtualMachineCloneSpec(
      :config        => configSpec,
      :location      => relocateSpec,
      :customization => customizationSpec,
      :powerOn       => true,
      :template      => false
    )

    # Deploy from specified template
    if (@vcloud_hosts.length == 1) or (i == @vcloud_hosts.length - 1)
      vm[h['template']].CloneVM_Task( :folder => vsphere_helper.find_folder(@config['folder']), :name => h['vmhostname'], :spec => spec ).wait_for_completion
    else
      vm[h['template']].CloneVM_Task( :folder => vsphere_helper.find_folder(@config['folder']), :name => h['vmhostname'], :spec => spec )
    end
  end
  @logger.notify 'Spent %.2f seconds deploying VMs' % (Time.now - start)

  start = Time.now
  @vcloud_hosts.each_with_index do |h, i|
    @logger.notify "Booting #{h['vmhostname']} (#{h.name}) and waiting for it to register with vSphere"
    try = 1
    last_wait = 0
    wait = 1
    until
      vsphere_helper.find_vms(h['vmhostname'])[h['vmhostname']].summary.guest.toolsRunningStatus == 'guestToolsRunning' and
      vsphere_helper.find_vms(h['vmhostname'])[h['vmhostname']].summary.guest.ipAddress != nil
      if try <= attempts
        sleep wait
        (last_wait, wait) = wait, last_wait + wait
        try += 1
      else
        raise "vSphere registration failed after #{wait} seconds"
      end
    end
  end
  @logger.notify "Spent %.2f seconds booting and waiting for vSphere registration" % (Time.now - start)

  start = Time.now
  @vcloud_hosts.each_with_index do |h, i|
    @logger.notify "Waiting for #{h['vmhostname']} DNS resolution"
    try = 1
    last_wait = 0
    wait = 3

    begin
      Socket.getaddrinfo(h['vmhostname'], nil)
    rescue
      if try <= attempts
        sleep wait
        (last_wait, wait) = wait, last_wait + wait
        try += 1

        retry
      else
        raise "DNS resolution failed after #{wait} seconds"
      end
    end
  end
  @logger.notify "Spent %.2f seconds waiting for DNS resolution" % (Time.now - start)

  vsphere_helper.close 
end

Instance Method Details

#cleanup ⇒ Object 



140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
 
# File 'lib/beaker/hypervisor/vcloud.rb', line 140

def cleanup
  @logger.notify "Destroying vCloud boxes"
  vsphere_credentials = VsphereHelper.load_config

  @logger.notify "Connecting to vSphere at #{vsphere_credentials[:server]}" +
    " with credentials for #{vsphere_credentials[:user]}"

  vsphere_helper = VsphereHelper.new( vsphere_credentials )

  vm_names = @vcloud_hosts.map {|h| h['vmhostname'] }.compact
  if @vcloud_hosts.length != vm_names.length
    @logger.warn "Some hosts did not have vmhostname set correctly! This likely means VM provisioning was not successful"
  end
  vms = vsphere_helper.find_vms vm_names
  vm_names.each do |name|
    unless vm = vms[name]
      raise "Couldn't find VM #{name} in vSphere!"
    end

    if vm.runtime.powerState == 'poweredOn'
      @logger.notify "Shutting down #{vm.name}"
      start = Time.now
      vm.PowerOffVM_Task.wait_for_completion
      @logger.notify "Spent %.2f seconds halting #{vm.name}" % (Time.now - start)
    end

    start = Time.now
    vm.Destroy_Task
    @logger.notify "Spent %.2f seconds destroying #{vm.name}" % (Time.now - start)
  end

  vsphere_helper.close
end