mirror of
https://github.com/vagrant-libvirt/vagrant-libvirt.git
synced 2025-02-25 18:55:27 -06:00
92 lines
2.6 KiB
Ruby
92 lines
2.6 KiB
Ruby
require 'spec_helper'
|
|
require 'support/sharedcontext'
|
|
require 'support/libvirt_context'
|
|
require 'vagrant-libvirt/action/shutdown_domain'
|
|
|
|
describe VagrantPlugins::ProviderLibvirt::Action::ShutdownDomain do
|
|
subject { described_class.new(app, env, target_state, current_state) }
|
|
|
|
include_context 'unit'
|
|
include_context 'libvirt'
|
|
|
|
let(:libvirt_domain) { double('libvirt_domain') }
|
|
let(:servers) { double('servers') }
|
|
let(:current_state) { :running }
|
|
let(:target_state) { :shutoff }
|
|
|
|
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)
|
|
allow(ui).to receive(:info).with('Attempting direct shutdown of domain...')
|
|
end
|
|
|
|
context "when state is shutoff" do
|
|
before { allow(domain).to receive(:state).and_return('shutoff') }
|
|
|
|
it "should not shutdown" do
|
|
expect(domain).not_to receive(:shutoff)
|
|
subject.call(env)
|
|
end
|
|
|
|
it "should not print shutdown message" do
|
|
expect(ui).not_to receive(:info)
|
|
subject.call(env)
|
|
end
|
|
|
|
it "should provide a true result" do
|
|
subject.call(env)
|
|
expect(env[:result]).to be_truthy
|
|
end
|
|
end
|
|
|
|
context "when state is running" do
|
|
before do
|
|
allow(domain).to receive(:state).and_return('running')
|
|
allow(domain).to receive(:wait_for)
|
|
allow(domain).to receive(:shutdown)
|
|
end
|
|
|
|
it "should shutdown" do
|
|
expect(domain).to receive(:shutdown)
|
|
subject.call(env)
|
|
end
|
|
|
|
it "should print shutdown message" do
|
|
expect(ui).to receive(:info).with('Attempting direct shutdown of domain...')
|
|
subject.call(env)
|
|
end
|
|
|
|
it "should wait for machine to shutdown" do
|
|
expect(domain).to receive(:wait_for)
|
|
subject.call(env)
|
|
end
|
|
|
|
context "when final state is not shutoff" do
|
|
before do
|
|
expect(domain).to receive(:state).and_return('running').exactly(4).times
|
|
end
|
|
|
|
it "should provide a false result" do
|
|
subject.call(env)
|
|
expect(env[:result]).to be_falsey
|
|
end
|
|
end
|
|
|
|
context "when final state is shutoff" do
|
|
before do
|
|
expect(domain).to receive(:state).and_return('running').exactly(3).times
|
|
expect(domain).to receive(:state).and_return('shutoff')
|
|
end
|
|
|
|
it "should provide a true result" do
|
|
subject.call(env)
|
|
expect(env[:result]).to be_truthy
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|