fog/fog-vsphere

module Fog
  module Vsphere
    class Compute
      class Real
        def vm_reboot(options = {})
          options = { 'force' => false }.merge(options)
          raise ArgumentError, 'instance_uuid is a required parameter' unless options.key? 'instance_uuid'

          search_filter = { :uuid => options['instance_uuid'], 'vmSearch' => true, 'instanceUuid' => true }
          vm_mob_ref = connection.searchIndex.FindAllByUuid(search_filter).first

          if options['force']
            task = vm_mob_ref.ResetVM_Task
            task.wait_for_completion
            { 'task_state' => task.info.result, 'reboot_type' => 'reset_power' }
          else
            vm_mob_ref.RebootGuest
            { 'task_state' => 'running', 'reboot_type' => 'reboot_guest' }
          end
        end
      end

      class Mock
        def vm_reboot(options = {})
          raise ArgumentError, 'instance_uuid is a required parameter' unless options.key? 'instance_uuid'
          { 'task_state' => 'running', 'reboot_type' => options['force'] ? 'reset_power' : 'reboot_guest' }
        end
      end
    end
  end
end