module Fog
  module Compute
    class Vsphere
      class Real

        def vm_migrate(options = {})
          #priority is the only required option, and it has a sane default option.
          priority = options['priority'].nil? ? 'defaultPriority' : options["priority"]
          raise ArgumentError, "instance_uuid is a required parameter" unless options.has_key? 'instance_uuid'

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

          unless vm_mob_ref.kind_of? RbVmomi::VIM::VirtualMachine
            raise Fog::Vsphere::Errors::NotFound,
              "Could not find VirtualMachine with instance uuid #{options['instance_uuid']}"
          end
          task = vm_mob_ref.MigrateVM_Task(:pool => options['pool'], :host => options['host'], :priority => "#{priority}", :state => options['state'] )
          task.wait_for_completion
          { 'task_state' => task.info.state }
        end

      end

      class Mock

        def vm_migrate(options = {})
          priority = options['priority'].nil? ? 'defaultPriority' : options["priority"]
          raise ArgumentError, "instance_uuid is a required parameter" unless options.has_key? 'instance_uuid'
          { 'task_state' => 'success' }
        end

      end
    end
  end
end