vagrant-libvirt 0.5.3 → 0.6.0

data/spec/unit/action/create_domain_spec.rb

@@ -1,8 +1,11 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'support/sharedcontext'
 require 'support/libvirt_context'
 
 require 'vagrant-libvirt/errors'
+require 'vagrant-libvirt/util/byte_number'
 require 'vagrant-libvirt/action/create_domain'
 
 describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
@@ -14,6 +17,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
   let(:libvirt_client) { double('libvirt_client') }
   let(:servers) { double('servers') }
   let(:volumes) { double('volumes') }
+  let(:domain_volume) { double('domain_volume') }
 
   let(:domain_xml) { File.read(File.join(File.dirname(__FILE__), File.basename(__FILE__, '.rb'), domain_xml_file)) }
   let(:storage_pool_xml) { File.read(File.join(File.dirname(__FILE__), File.basename(__FILE__, '.rb'), storage_pool_xml_file)) }
@@ -27,8 +31,15 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
 
       allow(connection).to receive(:servers).and_return(servers)
       allow(connection).to receive(:volumes).and_return(volumes)
+      allow(volumes).to receive(:all).and_return([domain_volume])
+      allow(domain_volume).to receive(:pool_name).and_return('default')
+      allow(domain_volume).to receive(:[]).with('name').and_return('vagrant-test_default.img')
+      allow(domain_volume).to receive(:path).and_return('/var/lib/libvirt/images/vagrant-test_default.img')
+      allow(machine).to receive_message_chain("box.name") { 'vagrant-libvirt/test' }
 
       allow(logger).to receive(:info)
+      allow(logger).to receive(:debug)
+      allow(ui).to receive(:info)
 
       env[:domain_name] = "vagrant-test_default"
 
@@ -36,7 +47,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
       env[:box_volumes].push({
         :path=>"/test/box.img",
         :name=>"test_vagrant_box_image_1.1.1_0.img", 
-        :virtual_size=>5
+        :virtual_size=> ByteNumber.new(5),
       })
       # should be ignored for system session and used for user session
       allow(Process).to receive(:uid).and_return(9999)
@@ -44,35 +55,44 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
     end
 
     context 'connection => qemu:///system' do
-      context 'default pool' do
-        let(:domain_xml_file) { 'default_domain.xml' }
-        let(:storage_pool_xml_file) { 'default_system_storage_pool.xml' }
+      let(:domain_xml_file) { 'default_domain.xml' }
 
+      context 'default pool' do
         it 'should execute correctly' do
-          expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
-          expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
           expect(servers).to receive(:create).with(xml: domain_xml).and_return(machine)
           expect(volumes).to_not receive(:create) # additional disks only
 
           expect(subject.call(env)).to be_nil
         end
 
-        context 'additional disks' do
+        context 'with no box' do
+          let(:storage_pool_xml_file) { 'default_system_storage_pool.xml' }
           let(:vagrantfile) do
             <<-EOF
             Vagrant.configure('2') do |config|
               config.vm.define :test
-              config.vm.provider :libvirt do |libvirt|
-                libvirt.storage :file, :size => '20G'
-              end
             end
             EOF
           end
 
+          it 'should query for the storage pool path' do
+            expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
+            expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
+            expect(servers).to receive(:create).and_return(machine)
+
+            expect(subject.call(env)).to be_nil
+          end
+        end
+
+        context 'additional disks' do
+          let(:vagrantfile_providerconfig) do
+            <<-EOF
+            libvirt.storage :file, :size => '20G'
+            EOF
+          end
+
           context 'volume create failed' do
             it 'should raise an exception' do
-              expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
-              expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
               expect(volumes).to receive(:create).and_raise(Libvirt::Error)
 
               expect{ subject.call(env) }.to raise_error(VagrantPlugins::ProviderLibvirt::Errors::FogCreateDomainVolumeError)
@@ -83,8 +103,6 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
             let(:domain_xml_file) { 'additional_disks_domain.xml' }
 
             it 'should complete' do
-              expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
-              expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
               expect(volumes).to receive(:create).with(
                 hash_including(
                   :path        => "/var/lib/libvirt/images/vagrant-test_default-vdb.qcow2",
@@ -102,6 +120,14 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
       end
 
       context 'no default pool' do
+        let(:vagrantfile) do
+          <<-EOF
+          Vagrant.configure('2') do |config|
+            config.vm.define :test
+          end
+          EOF
+        end
+
         it 'should raise an exception' do
           expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(nil)
 
@@ -111,45 +137,51 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomain do
     end
 
     context 'connection => qemu:///session' do
-      let(:vagrantfile) do
+      let(:vagrantfile_providerconfig) do
         <<-EOF
-        Vagrant.configure('2') do |config|
-          config.vm.define :test
-          config.vm.provider :libvirt do |libvirt|
-            libvirt.qemu_use_session = true
-          end
-        end
+        libvirt.qemu_use_session = true
         EOF
       end
 
       context 'default pool' do
-        let(:storage_pool_xml_file) { 'default_user_storage_pool.xml' }
-
         it 'should execute correctly' do
-          expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
-          expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
           expect(servers).to receive(:create).and_return(machine)
 
           expect(subject.call(env)).to be_nil
         end
 
-        context 'additional disks' do
+        context 'with no box' do
+          let(:storage_pool_xml_file) { 'default_user_storage_pool.xml' }
           let(:vagrantfile) do
             <<-EOF
             Vagrant.configure('2') do |config|
               config.vm.define :test
               config.vm.provider :libvirt do |libvirt|
-                libvirt.qemu_use_session = true
-                libvirt.storage :file, :size => '20G'
+                #{vagrantfile_providerconfig}
               end
             end
             EOF
           end
 
+          it 'should query for the storage pool path' do
+            expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
+            expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
+            expect(servers).to receive(:create).and_return(machine)
+
+            expect(subject.call(env)).to be_nil
+          end
+        end
+
+        context 'additional disks' do
+          let(:vagrantfile_providerconfig) do
+            <<-EOF
+            libvirt.qemu_use_session = true
+            libvirt.storage :file, :size => '20G'
+            EOF
+          end
+
           context 'volume create succeeded' do
             it 'should complete' do
-              expect(libvirt_client).to receive(:lookup_storage_pool_by_name).and_return(libvirt_storage_pool)
-              expect(libvirt_storage_pool).to receive(:xml_desc).and_return(storage_pool_xml)
               expect(volumes).to receive(:create).with(
                 hash_including(
                   :path        => "/var/lib/libvirt/images/vagrant-test_default-vdb.qcow2",

data/spec/unit/action/create_domain_volume_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'support/sharedcontext'
 require 'support/libvirt_context'
@@ -32,6 +34,8 @@ describe VagrantPlugins::ProviderLibvirt::Action::CreateDomainVolume do
       allow(all).to receive(:first).and_return(box_volume)
       allow(box_volume).to receive(:id).and_return(nil)
       env[:domain_name] = 'test'
+
+      allow(logger).to receive(:debug)
     end
 
     context 'when one disk' do

data/spec/unit/action/destroy_domain_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'support/sharedcontext'
 require 'support/libvirt_context'
@@ -10,14 +12,18 @@ describe VagrantPlugins::ProviderLibvirt::Action::DestroyDomain do
   include_context 'unit'
   include_context 'libvirt'
 
+  let(:driver) { double('driver') }
   let(:libvirt_domain) { double('libvirt_domain') }
   let(:libvirt_client) { double('libvirt_client') }
   let(:servers) { double('servers') }
 
+  before do
+    allow(machine.provider).to receive('driver').and_return(driver)
+    allow(driver).to receive(:connection).and_return(connection)
+  end
+
   describe '#call' do
     before do
-      allow_any_instance_of(VagrantPlugins::ProviderLibvirt::Driver)
-        .to receive(:connection).and_return(connection)
       allow(connection).to receive(:client).and_return(libvirt_client)
       allow(libvirt_client).to receive(:lookup_domain_by_uuid)
         .and_return(libvirt_domain)

data/spec/unit/action/forward_ports_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'support/sharedcontext'
 require 'support/libvirt_context'

data/spec/unit/action/halt_domain_spec.rb

@@ -1,7 +1,9 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'support/sharedcontext'
 require 'support/libvirt_context'
-require 'vagrant-libvirt/action/destroy_domain'
+require 'vagrant-libvirt/action/halt_domain'
 
 describe VagrantPlugins::ProviderLibvirt::Action::HaltDomain do
   subject { described_class.new(app, env) }
@@ -9,81 +11,52 @@ describe VagrantPlugins::ProviderLibvirt::Action::HaltDomain do
   include_context 'unit'
   include_context 'libvirt'
 
+  let(:driver) { double('driver') }
   let(:libvirt_domain) { double('libvirt_domain') }
   let(:servers) { double('servers') }
 
+  before do
+    allow(machine.provider).to receive('driver').and_return(driver)
+    allow(driver).to receive(:created?).and_return(true)
+    allow(driver).to receive(:connection).and_return(connection)
+  end
+
   describe '#call' do
     before do
-      allow_any_instance_of(VagrantPlugins::ProviderLibvirt::Driver)
-        .to receive(:connection).and_return(connection)
       allow(connection).to receive(:servers).and_return(servers)
       allow(servers).to receive(:get).and_return(domain)
-      # always see this at the start of #call
-      expect(ui).to receive(:info).with('Halting domain...')
+      allow(ui).to receive(:info).with('Halting domain...')
     end
 
-    context 'with graceful timeout' do
-      it "should shutdown" do
-        expect(guest).to receive(:capability).with(:halt).and_return(true)
-        expect(domain).to receive(:wait_for).with(60).and_return(false)
-        expect(subject.call(env)).to be_nil
-      end
-
-      context 'when halt fails' do
-        before do
-          expect(logger).to receive(:info).with('Trying Libvirt graceful shutdown.')
-          expect(guest).to receive(:capability).with(:halt).and_raise(IOError)
-          expect(domain).to receive(:state).and_return('running')
-        end
+    context "when state is not running" do
+      before { expect(driver).to receive(:state).at_least(1).
+          and_return(:not_created) }
 
-        it "should call shutdown" do
-          expect(domain).to receive(:shutdown)
-          expect(domain).to receive(:wait_for).with(60).and_return(false)
-          expect(subject.call(env)).to be_nil
-        end
-
-        context 'when shutdown fails' do
-          it "should call power off" do
-            expect(logger).to receive(:error).with('Failed to shutdown cleanly. Calling force poweroff.')
-            expect(domain).to receive(:shutdown).and_raise(IOError)
-            expect(domain).to receive(:poweroff)
-            expect(subject.call(env)).to be_nil
-          end
-        end
-
-        context 'when shutdown exceeds the timeout' do
-          it "should call poweroff" do
-            expect(logger).to receive(:info).with('VM is still running. Calling force poweroff.')
-            expect(domain).to receive(:shutdown).and_raise(Timeout::Error)
-            expect(domain).to receive(:poweroff)
-            expect(subject.call(env)).to be_nil
-          end
-        end
+      it "should not poweroff when state is not running" do
+        expect(domain).not_to receive(:poweroff)
+        subject.call(env)
       end
 
-      context 'when halt exceeds the timeout' do
-        before do
-          expect(logger).to_not receive(:info).with('Trying Libvirt graceful shutdown.')
-          expect(guest).to receive(:capability).with(:halt).and_raise(Timeout::Error)
-        end
-
-        it "should call poweroff" do
-          expect(logger).to receive(:info).with('VM is still running. Calling force poweroff.')
-          expect(domain).to receive(:poweroff)
-          expect(subject.call(env)).to be_nil
-        end
+      it "should not print halting message" do
+        expect(ui).not_to receive(:info)
+        subject.call(env)
       end
     end
 
-    context 'with force halt enabled' do
+    context "when state is running" do
       before do
-        allow(env).to receive(:[]).and_call_original
-        expect(env).to receive(:[]).with(:force_halt).and_return(true)
+        expect(driver).to receive(:state).and_return(:running)
       end
 
-      it "should just call poweroff" do
+      it "should poweroff" do
         expect(domain).to receive(:poweroff)
-        expect(subject.call(env)).to be_nil
+        subject.call(env)
+      end
+
+      it "should print halting message" do
+        allow(domain).to receive(:poweroff)
+        expect(ui).to receive(:info).with('Halting domain...')
+        subject.call(env)
       end
     end
   end

data/spec/unit/action/handle_box_image_spec.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'spec_helper'
 require 'json'
 require 'support/sharedcontext'
@@ -77,7 +79,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::HandleBoxImage do
           ]
         }
         allow(env[:machine]).to receive_message_chain("box.directory.join") do |arg|
-          '/test/'.concat(arg.to_s)
+          '/test/' + arg.to_s
         end
       end
 
@@ -197,7 +199,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::HandleBoxImage do
           ],
         ]}
         allow(env[:machine]).to receive_message_chain("box.directory.join") do |arg|
-          '/test/'.concat(arg.to_s)
+          '/test/' + arg.to_s
         end
         allow(status).to receive(:success?).and_return(true)
         allow(Open3).to receive(:capture3).with('qemu-img', 'info', '--output=json', '/test/box.img').and_return([
@@ -334,7 +336,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::HandleBoxImage do
           ]
         }
         allow(env[:machine]).to receive_message_chain("box.directory.join") do |arg|
-          '/test/'.concat(arg.to_s)
+          '/test/' + arg.to_s
         end
       end
 
@@ -354,7 +356,7 @@ describe VagrantPlugins::ProviderLibvirt::Action::HandleBoxImage do
         allow(env[:machine]).to receive_message_chain("box.version") { '1.1.1' }
         allow(env[:machine]).to receive_message_chain("box.metadata") { box_metadata }
         allow(env[:machine]).to receive_message_chain("box.directory.join") do |arg|
-          '/test/'.concat(arg.to_s)
+          '/test/' + arg.to_s
         end
         allow(status).to receive(:success?).and_return(true)
         allow(Open3).to receive(:capture3).with('qemu-img', 'info', "--output=json", '/test/box.img').and_return([