vagrant-libvirt 0.8.2 → 0.9.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c7f252cbd53f8459e02a21e94449e6fef1ce762f1d053433970015fad9d00fef
-  data.tar.gz: 54af8a7a564e8ffdd0f18d3d3eca44367ac7613fb4892af995c5d202d8e429ea
+  metadata.gz: 79027c1097a7e8136b8d56f74fd0b27932bf17c48387de84c57626c2bd348714
+  data.tar.gz: cd6acea3cf183191ea6bbee6e60d7a1fc76a5f0b556f273f60657238283841ce
 SHA512:
-  metadata.gz: 00c708fba6fb82cec02b0b68730da87975964584b8b189bfcd77857b3b386ff81609bd135423872dcd04cd7e071f10f1cf7806d97b28565ed27464d2b2021857
-  data.tar.gz: abbbe10ad322fec1a49b9f3463fa7a50c9e46bce93c8c6aa505ca3c4fbd5e6a3b4bfa248e912d0608e5aefdd3427ce819e6bff9f0807da187a187c2e9085d13e
+  metadata.gz: 6d6483a73ce398e6fd057b2fc0503392c628ad1fc228d9d954b9a5e31d9a65ab883c25f30cb9360ce5eff46554492dd010b68d741f3c8935e64228207bf2f210
+  data.tar.gz: 1a083a93899bb48d4d750aa8967bdcb8e6b1b0b9a8fb0d88c912097444cd202c52fe0330f81a1e4803dde184b06e035c8bd8d32397035c186653b978db790e12

data/lib/vagrant-libvirt/action/create_domain.rb

@@ -16,20 +16,6 @@ module VagrantPlugins
           @app = app
         end
 
-        def _disk_name(name, disk)
-          "#{name}-#{disk[:device]}.#{disk[:type]}" # disk name
-        end
-
-        def _disks_print(disks)
-          disks.collect do |x|
-            "#{x[:device]}(#{x[:type]}, #{x[:bus]}, #{x[:size]})"
-          end.join(', ')
-        end
-
-        def _cdroms_print(cdroms)
-          cdroms.collect { |x| x[:dev] }.join(', ')
-        end
-
         def call(env)
           # Get config.
           config = env[:machine].provider_config
@@ -58,8 +44,6 @@ module VagrantPlugins
           @nvram = config.nvram
           @machine_type = config.machine_type
           @machine_arch = config.machine_arch
-          @disk_bus = config.disk_bus
-          @disk_device = config.disk_device
           @disk_driver_opts = config.disk_driver_opts
           @nested = config.nested
           @memory_size = config.memory.to_i * 1024
@@ -94,9 +78,8 @@ module VagrantPlugins
 
           # Storage
           @storage_pool_name = config.storage_pool_name
-          @snapshot_pool_name = config.snapshot_pool_name
-          @domain_volumes = []
-          @disks = config.disks
+          @domain_volumes = env[:domain_volumes] || []
+          @disks = env[:disks] || []
           @cdroms = config.cdroms
 
           # Input
@@ -139,77 +122,19 @@ module VagrantPlugins
           @memballoon_pci_bus = config.memballoon_pci_bus
           @memballoon_pci_slot = config.memballoon_pci_slot
 
-          config = env[:machine].provider_config
           @domain_type = config.driver
 
           @os_type = 'hvm'
 
-          resolver = ::VagrantPlugins::ProviderLibvirt::Util::DiskDeviceResolver.new(prefix=@disk_device[0..1])
-
-          # Get path to domain image from the storage pool selected if we have a box.
-          if env[:machine].config.vm.box
-            if @snapshot_pool_name != @storage_pool_name
-                pool_name = @snapshot_pool_name
-            else
-                pool_name = @storage_pool_name
-            end
-
-            # special handling for domain volume
-            env[:box_volumes][0][:device] = env[:box_volumes][0].fetch(:device, @disk_device)
-
-            resolver.resolve!(env[:box_volumes])
-
-            @logger.debug "Search for volumes in pool: #{pool_name}"
-            env[:box_volumes].each_index do |index|
-              suffix_index = index > 0 ? "_#{index}" : ''
-              domain_volume = env[:machine].provider.driver.connection.volumes.all(
-                name: "#{@name}#{suffix_index}.img"
-              ).find { |x| x.pool_name == pool_name }
-              raise Errors::DomainVolumeExists if domain_volume.nil?
-
-              @domain_volumes.push({
-                :dev => env[:box_volumes][index][:device],
-                :cache => @domain_volume_cache,
-                :bus => @disk_bus,
-                :path => domain_volume.path,
-                :virtual_size => env[:box_volumes][index][:virtual_size]
-              })
-            end
-
+          env[:domain_volumes].each_with_index do |vol, index|
+            suffix_index = index > 0 ? "_#{index}" : ''
+            domain_volume = env[:machine].provider.driver.connection.volumes.all(
+              name: "#{@name}#{suffix_index}.img"
+            ).find { |x| x.pool_name == vol[:pool] }
+            raise Errors::NoDomainVolume if domain_volume.nil?
           end
 
-          # If we have a box, take the path from the domain volume and set our storage_prefix.
-          # If not, we dump the storage pool xml to get its defined path.
-          # the default storage prefix is typically: /var/lib/libvirt/images/
-          if env[:machine].config.vm.box
-            storage_prefix = File.dirname(@domain_volumes[0][:path]) + '/' # steal
-          else
-            storage_prefix = get_disk_storage_prefix(env, @storage_pool_name)
-          end
-
-          resolver.resolve!(@disks)
-
           @disks.each do |disk|
-            disk[:path] ||= _disk_name(@name, disk)
-
-            # On volume creation, the <path> element inside <target>
-            # is oddly ignored; instead the path is taken from the
-            # <name> element:
-            # http://www.redhat.com/archives/libvir-list/2008-August/msg00329.html
-            disk[:name] = disk[:path]
-
-            disk[:absolute_path] = storage_prefix + disk[:path]
-
-            if not disk[:pool].nil?
-              disk_pool_name = disk[:pool]
-              @logger.debug "Overriding pool name with: #{disk_pool_name}"
-              disk_storage_prefix = get_disk_storage_prefix(env, disk_pool_name)
-              disk[:absolute_path] = disk_storage_prefix + disk[:path]
-              @logger.debug "Overriding disk path with: #{disk[:absolute_path]}"
-            else
-              disk_pool_name = @storage_pool_name
-            end
-
             # make the disk. equivalent to:
             # qemu-img create -f qcow2 <path> 5g
             begin
@@ -221,13 +146,13 @@ module VagrantPlugins
                 owner: storage_uid(env),
                 group: storage_gid(env),
                 #:allocation => ?,
-                pool_name: disk_pool_name
+                pool_name: disk[:pool],
               )
             rescue Libvirt::Error => e
               # It is hard to believe that e contains just a string
               # and no useful error code!
               msg = "Call to virStorageVolCreateXML failed: " +
-                    "storage volume '#{disk[:path]}' exists already"
+                    "storage volume '#{disk[:absolute_path]}' exists already"
               if e.message == msg and disk[:allow_existing]
                 disk[:preexisting] = true
               else
@@ -300,7 +225,7 @@ module VagrantPlugins
           end
           env[:ui].info(" -- Storage pool:      #{@storage_pool_name}")
           @domain_volumes.each do |volume|
-            env[:ui].info(" -- Image(#{volume[:dev]}):        #{volume[:path]}, #{volume[:bus]}, #{volume[:virtual_size].to_GB}G")
+            env[:ui].info(" -- Image(#{volume[:device]}):        #{volume[:absolute_path]}, #{volume[:bus]}, #{volume[:virtual_size].to_GB}G")
           end
 
           if not @disk_driver_opts.empty?
@@ -466,11 +391,14 @@ module VagrantPlugins
         end
 
         private
-        def get_disk_storage_prefix(env, disk_pool_name)
-          disk_storage_pool = env[:machine].provider.driver.connection.client.lookup_storage_pool_by_name(disk_pool_name)
-          raise Errors::NoStoragePool if disk_storage_pool.nil?
-          xml = Nokogiri::XML(disk_storage_pool.xml_desc)
-          disk_storage_prefix = xml.xpath('/pool/target/path').inner_text.to_s + '/'
+        def _disks_print(disks)
+          disks.collect do |x|
+            "#{x[:device]}(#{x[:type]}, #{x[:bus]}, #{x[:size]})"
+          end.join(', ')
+        end
+
+        def _cdroms_print(cdroms)
+          cdroms.collect { |x| x[:dev] }.join(', ')
         end
       end
     end

data/lib/vagrant-libvirt/action/destroy_domain.rb

@@ -1,7 +1,12 @@
 # frozen_string_literal: true
 
 require 'log4r'
-require 'rexml'
+
+begin
+  require 'rexml'
+rescue LoadError
+  require 'rexml/rexml'
+end
 
 module VagrantPlugins
   module ProviderLibvirt
@@ -43,21 +48,24 @@ module VagrantPlugins
 
           domain = env[:machine].provider.driver.connection.servers.get(env[:machine].id.to_s)
 
+          undefine_flags = 0
+          undefine_flags |= ProviderLibvirt::Util::DomainFlags::VIR_DOMAIN_UNDEFINE_KEEP_NVRAM if env[:machine].provider_config.nvram
+
           if env[:machine].provider_config.disks.empty? &&
              env[:machine].provider_config.cdroms.empty?
             # if using default configuration of disks and cdroms
             # cdroms are consider volumes, but cannot be destroyed
-            domain.destroy(destroy_volumes: true)
+            destroy_domain(domain, destroy_volumes: true, flags: undefine_flags)
           else
             domain_xml = libvirt_domain.xml_desc(1)
             xml_descr = REXML::Document.new(domain_xml)
             disks_xml = REXML::XPath.match(xml_descr, '/domain/devices/disk[@device="disk"]')
             have_aliases = !(REXML::XPath.match(disks_xml, './alias[@name="ua-box-volume-0"]').first).nil?
             if !have_aliases
-              env[:ui].warn(I18n.t('vagrant_libvirt.destroy.obsolete_method'))
+              env[:ui].warn(I18n.t('vagrant_libvirt.domain_xml.obsolete_method'))
             end
 
-            domain.destroy(destroy_volumes: false)
+            destroy_domain(domain, destroy_volumes: false, flags: undefine_flags)
 
             volumes = domain.volumes
 
@@ -164,6 +172,14 @@ module VagrantPlugins
             libvirt_disk.destroy if libvirt_disk
           end
         end
+
+        def destroy_domain(domain, destroy_volumes:, flags:)
+          if domain.method(:destroy).parameters.first.include?(:flags)
+            domain.destroy(destroy_volumes: destroy_volumes, flags: flags)
+          else
+            domain.destroy(destroy_volumes: destroy_volumes)
+          end
+        end
       end
     end
   end

data/lib/vagrant-libvirt/action/destroy_networks.rb

@@ -68,7 +68,7 @@ module VagrantPlugins
             # Shutdown network first.
             # Undefine network.
             begin
-              libvirt_network.destroy
+              libvirt_network.destroy if libvirt_network.active?
               libvirt_network.undefine
               @logger.info 'Undefined it'
             rescue => e

data/lib/vagrant-libvirt/action/handle_box_image.rb

@@ -104,6 +104,8 @@ module VagrantPlugins
           end
           # save for use by later actions
           env[:box_volumes][0][:virtual_size] = box_virtual_size
+          # special handling for domain volume
+          env[:box_volumes][0][:device] ||= config.disk_device
 
           # while inside the synchronize block take care not to call the next
           # action in the chain, as must exit this block first to prevent

data/lib/vagrant-libvirt/action/resolve_disk_settings.rb

@@ -0,0 +1,174 @@
+# frozen_string_literal: true
+
+require 'log4r'
+require 'rexml'
+
+require 'vagrant-libvirt/util/resolvers'
+
+module VagrantPlugins
+  module ProviderLibvirt
+    module Action
+      class ResolveDiskSettings
+        def initialize(app, _env)
+          @logger = Log4r::Logger.new('vagrant_libvirt::action::resolve_disk_devices')
+          @app = app
+        end
+
+        def call(env)
+          # Get config.
+          config = env[:machine].provider_config
+
+          domain_name = env[:domain_name] # only set on create
+          disk_bus = config.disk_bus
+          disk_device = config.disk_device
+          domain_volume_cache = config.volume_cache || 'default'
+
+          # Storage
+          storage_pool_name = config.storage_pool_name
+          snapshot_pool_name = config.snapshot_pool_name
+          domain_volumes = []
+          disks = config.disks.dup
+
+          resolver = ::VagrantPlugins::ProviderLibvirt::Util::DiskDeviceResolver.new(disk_device[0..1])
+
+          # Get path to domain image from the storage pool selected if we have a box.
+          if env[:machine].config.vm.box
+            pool_name = if snapshot_pool_name == storage_pool_name
+                          storage_pool_name
+                        else
+                          snapshot_pool_name
+                        end
+
+            if env[:box_volumes].nil?
+              # domain must be already created, need to read domain volumes from domain XML
+              libvirt_domain = env[:machine].provider.driver.connection.client.lookup_domain_by_uuid(
+                env[:machine].id
+              )
+              domain_xml = libvirt_domain.xml_desc(1)
+              xml_descr = REXML::Document.new(domain_xml)
+              domain_name = xml_descr.elements['domain'].elements['name'].text
+              disks_xml = REXML::XPath.match(xml_descr, '/domain/devices/disk[@device="disk"]')
+              have_aliases = !REXML::XPath.match(disks_xml, './alias[@name="ua-box-volume-0"]').first.nil?
+              env[:ui].warn(I18n.t('vagrant_libvirt.domain_xml.obsolete_method')) unless have_aliases
+
+              if have_aliases
+                REXML::XPath.match(disks_xml,
+                                   './alias[contains(@name, "ua-box-volume-")]').each_with_index do |alias_xml, idx|
+                  domain_volumes.push(volume_from_xml(alias_xml.parent, domain_name, idx))
+                end
+              else
+                # fallback to try and infer which boxes are box images, as they are listed first
+                # as soon as there is no match, can exit
+                disks_xml.each_with_index do |box_disk_xml, idx|
+                  diskname = box_disk_xml.elements['source'].attributes['file'].rpartition('/').last
+
+                  break if volume_name(domain_name, idx) != diskname
+
+                  domain_volumes.push(volume_from_xml(box_disk_xml, domain_name, idx))
+                end
+              end
+            else
+
+              @logger.debug "Search for volumes in pool: #{pool_name}"
+              env[:box_volumes].each_index do |index|
+                domain_volume = env[:machine].provider.driver.connection.volumes.all(
+                  name: volume_name(domain_name, index)
+                ).find { |x| x.pool_name == pool_name }
+                raise Errors::NoDomainVolume if domain_volume.nil?
+
+                domain_volumes.push(
+                  {
+                    name: volume_name(domain_name, index),
+                    device: env[:box_volumes][index][:device],
+                    cache: domain_volume_cache,
+                    bus: disk_bus,
+                    absolute_path: domain_volume.path,
+                    virtual_size: env[:box_volumes][index][:virtual_size],
+                    pool: pool_name,
+                  }
+                )
+              end
+            end
+
+            resolver.resolve!(domain_volumes)
+
+            # If we have a box, take the path from the domain volume and set our storage_prefix.
+            # If not, we dump the storage pool xml to get its defined path.
+            # the default storage prefix is typically: /var/lib/libvirt/images/
+            storage_prefix = "#{File.dirname(domain_volumes[0][:absolute_path])}/" # steal
+          else
+            # Ensure domain name is set for subsequent steps if restarting a machine without a box
+            libvirt_domain = env[:machine].provider.driver.connection.client.lookup_domain_by_uuid(
+              env[:machine].id
+            )
+            domain_xml = libvirt_domain.xml_desc(1)
+            xml_descr = REXML::Document.new(domain_xml)
+            domain_name = xml_descr.elements['domain'].elements['name'].text
+
+            storage_prefix = get_disk_storage_prefix(env[:machine], storage_pool_name)
+          end
+
+          resolver.resolve!(disks)
+
+          disks.each do |disk|
+            disk[:path] ||= disk_name(domain_name, disk)
+
+            # On volume creation, the <path> element inside <target>
+            # is oddly ignored; instead the path is taken from the
+            # <name> element:
+            # http://www.redhat.com/archives/libvir-list/2008-August/msg00329.html
+            disk[:name] = disk[:path]
+
+            disk[:absolute_path] = storage_prefix + disk[:path]
+
+            if disk[:pool].nil?
+              disk[:pool] = storage_pool_name
+            else
+              @logger.debug "Overriding pool name with: #{disk[:pool]}"
+              disk_storage_prefix = get_disk_storage_prefix(env[:machine], disk[:pool])
+              disk[:absolute_path] = disk_storage_prefix + disk[:path]
+              @logger.debug "Overriding disk path with: #{disk[:absolute_path]}"
+            end
+          end
+
+          env[:domain_volumes] = domain_volumes
+          env[:disks] = disks
+
+          @app.call(env)
+        end
+
+        private
+
+        def disk_name(name, disk)
+          "#{name}-#{disk[:device]}.#{disk[:type]}" # disk name
+        end
+
+        def get_disk_storage_prefix(machine, disk_pool_name)
+          disk_storage_pool = machine.provider.driver.connection.client.lookup_storage_pool_by_name(disk_pool_name)
+          raise Errors::NoStoragePool if disk_storage_pool.nil?
+
+          xml = Nokogiri::XML(disk_storage_pool.xml_desc)
+          "#{xml.xpath('/pool/target/path').inner_text}/"
+        end
+
+        def volume_name(domain_name, index)
+          domain_name + (index.zero? ? '.img' : "_#{index}.img")
+        end
+
+        def volume_from_xml(device_xml, domain_name, index)
+          driver = device_xml.elements['driver']
+          source = device_xml.elements['source']
+          target = device_xml.elements['target']
+
+          {
+            name: volume_name(domain_name, index),
+            device: target.attributes['dev'],
+            cache: driver.attributes['cache'],
+            bus: target.attributes['bus'],
+            absolute_path: source.attributes['file'],
+          }
+        end
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/action/start_domain.rb

@@ -375,10 +375,50 @@ module VagrantPlugins
                 end
               end
 
+              loader = REXML::XPath.first(xml_descr, '/domain/os/loader')
+              if config.loader
+                if loader.nil?
+                  descr_changed = true
+                  loader = REXML::Element.new('loader')
+                  REXML::XPath.first(xml_descr, '/domain/os').insert_after('//type', loader)
+                  loader.text = config.loader
+                else
+                  if (loader.text or '').strip != config.loader
+                    descr_changed = true
+                    loader.text = config.loader
+                  end
+                end
+                loader.attributes['type'] = config.nvram ? 'pflash' : 'rom'
+              elsif !loader.nil?
+                descr_changed = true
+                loader.parent.delete_element(loader)
+              end
+
+              undefine_flags = 0
+              nvram = REXML::XPath.first(xml_descr, '/domain/os/nvram')
+              if config.nvram
+                if nvram.nil?
+                  descr_changed = true
+                  nvram = REXML::Element.new('nvram')
+                  REXML::XPath.first(xml_descr, '/domain/os').insert_after(loader, nvram)
+                  nvram.text = config.nvram
+                else
+                  if (nvram.text or '').strip != config.nvram
+                    descr_changed = true
+                    nvram.text = config.nvram
+                  end
+                  undefine_flags |= ProviderLibvirt::Util::DomainFlags::VIR_DOMAIN_UNDEFINE_KEEP_NVRAM
+                end
+              elsif !nvram.nil?
+                descr_changed = true
+                undefine_flags |= ProviderLibvirt::Util::DomainFlags::VIR_DOMAIN_UNDEFINE_NVRAM
+                nvram.parent.delete_element(nvram)
+              end
+
               # Apply
               if descr_changed
                 begin
-                  libvirt_domain.undefine
+                  libvirt_domain.undefine(undefine_flags)
                   new_descr = String.new
                   xml_descr.write new_descr
                   env[:machine].provider.driver.connection.servers.create(xml: new_descr)

data/lib/vagrant-libvirt/action.rb

@@ -35,6 +35,7 @@ module VagrantPlugins
       autoload :ReadMacAddresses, action_root.join('read_mac_addresses')
       autoload :RemoveLibvirtImage, action_root.join('remove_libvirt_image')
       autoload :RemoveStaleVolume, action_root.join('remove_stale_volume')
+      autoload :ResolveDiskSettings, action_root.join('resolve_disk_settings')
       autoload :ResumeDomain, action_root.join('resume_domain')
       autoload :SetNameOfDomain, action_root.join('set_name_of_domain')
       autoload :SetBootOrder, action_root.join('set_boot_order')
@@ -81,6 +82,7 @@ module VagrantPlugins
               b2.use SetNameOfDomain
 
               if !env[:machine].config.vm.box
+                b2.use ResolveDiskSettings
                 b2.use CreateDomain
                 b2.use CreateNetworks
                 b2.use CreateNetworkInterfaces
@@ -94,6 +96,7 @@ module VagrantPlugins
                 b2.use HandleBox
                 b2.use HandleBoxImage
                 b2.use CreateDomainVolume
+                b2.use ResolveDiskSettings
                 b2.use CreateDomain
                 b2.use CreateNetworks
                 b2.use CreateNetworkInterfaces
@@ -104,6 +107,7 @@ module VagrantPlugins
               end
             else
               env[:halt_on_error] = true
+              b2.use ResolveDiskSettings
               b2.use CreateNetworks
               b2.use action_start
             end

data/lib/vagrant-libvirt/config.rb

@@ -994,6 +994,10 @@ module VagrantPlugins
           errors << "libvirt.qemu_use_agent must be a boolean."
         end
 
+        if !@nvram.nil? && @loader.nil?
+          errors << "use of 'nvram' requires a 'loader' to be specified, please add one to the configuration"
+        end
+
         if @qemu_use_agent == true
           # if qemu agent is used to optain domain ip configuration, at least
           # one qemu channel has to be configured. As there are various options,

data/lib/vagrant-libvirt/templates/domain.xml.erb

@@ -117,9 +117,9 @@
             @disk_driver_opts.reject { |k,v| v.nil? }
                              .map { |k,v| "#{k}='#{v}'"}
                              .join(' ') -%>/>
-      <source file='<%= volume[:path] %>'/>
+      <source file='<%= volume[:absolute_path] %>'/>
       <%# we need to ensure a unique target dev -%>
-      <target dev='<%= volume[:dev] %>' bus='<%= volume[:bus] %>'/>
+      <target dev='<%= volume[:device] %>' bus='<%= volume[:bus] %>'/>
     </disk>
 <%- end -%>
 <%# additional disks -%>

data/lib/vagrant-libvirt/util/domain_flags.rb

@@ -0,0 +1,15 @@
+ # Ripped from https://libvirt.org/html/libvirt-libvirt-domain.html#types
+module VagrantPlugins
+  module ProviderLibvirt
+    module Util
+      module DomainFlags
+        # virDomainUndefineFlagsValues
+        VIR_DOMAIN_UNDEFINE_MANAGED_SAVE = 1 # Also remove any managed save
+        VIR_DOMAIN_UNDEFINE_SNAPSHOTS_METADATA = 2 # If last use of domain, then also remove any snapshot metadata
+        VIR_DOMAIN_UNDEFINE_NVRAM = 4 # Also remove any nvram file
+        VIR_DOMAIN_UNDEFINE_KEEP_NVRAM = 8 # Keep nvram file
+        VIR_DOMAIN_UNDEFINE_CHECKPOINTS_METADATA = 16 # If last use of domain, then also remove any checkpoint metadata Future undefine control flags should come here.
+      end
+    end
+  end
+end

data/lib/vagrant-libvirt/util.rb

@@ -9,6 +9,7 @@ module VagrantPlugins
       autoload :NetworkUtil, 'vagrant-libvirt/util/network_util'
       autoload :StorageUtil, 'vagrant-libvirt/util/storage_util'
       autoload :ErrorCodes, 'vagrant-libvirt/util/error_codes'
+      autoload :DomainFlags, 'vagrant-libvirt/util/domain_flags'
       autoload :Ui, 'vagrant-libvirt/util/ui'
     end
   end

data/lib/vagrant-libvirt/version

@@ -1 +1 @@
-0.8.2
+0.9.0

data/locales/en.yml

@@ -58,9 +58,11 @@ en:
     remove_stale_volume: |-
       Remove stale volume...
 
-    destroy:
+    domain_xml:
       obsolete_method: |
-        Destroying machine that was originally created without device aliases (pre 0.6.0), using fallback approach.
+        Machine that was originally created without device aliases (pre 0.6.0), using fall-back approach for device identification.
+
+    destroy:
       unexpected_volumes: |
         Unexpected number of volumes detected, possibly additional volumes attached outside of vagrant-libvirt.
         Attempting to handle this correctly, however may experience unexpected behaviour in the domain destroy.

data/spec/support/libvirt_context.rb

@@ -10,6 +10,7 @@ shared_context 'libvirt' do
   let(:libvirt_context) { true                      }
   let(:id)              { 'dummy-vagrant_dummy'     }
   let(:connection)      { double('connection') }
+  let(:driver)          { instance_double(VagrantPlugins::ProviderLibvirt::Driver) }
   let(:domain)          { instance_double(::Fog::Libvirt::Compute::Server) }
   let(:libvirt_client)  { instance_double(::Libvirt::Connect) }
   let(:libvirt_domain)  { instance_double(::Libvirt::Domain) }
@@ -33,5 +34,8 @@ shared_context 'libvirt' do
 
     allow(machine).to receive(:id).and_return(id)
     allow(Log4r::Logger).to receive(:new).and_return(logger)
+
+    allow(machine.provider).to receive('driver').and_return(driver)
+    allow(driver).to receive(:connection).and_return(connection)
   end
 end

data/spec/unit/action/cleanup_on_failure_spec.rb

@@ -34,8 +34,6 @@ describe VagrantPlugins::ProviderLibvirt::Action::CleanupOnFailure do
   let(:state) { double('state') }
 
   before do
-    allow_any_instance_of(VagrantPlugins::ProviderLibvirt::Driver)
-      .to receive(:connection).and_return(connection)
     allow(machine).to receive(:id).and_return('test-machine-id')
     allow(machine).to receive(:state).and_return(state)