112 lines
3.1 KiB
Ruby
112 lines
3.1 KiB
Ruby
require File.expand_path("../../../../base", __FILE__)
|
|
require Vagrant.source_root.join("plugins/commands/suspend/command")
|
|
|
|
describe VagrantPlugins::CommandSuspend::Command do
|
|
include_context "unit"
|
|
|
|
let(:entry_klass) { Vagrant::MachineIndex::Entry }
|
|
let(:argv) { [] }
|
|
let(:vagrantfile_content){ "" }
|
|
let(:iso_env) do
|
|
env = isolated_environment
|
|
env.vagrantfile(vagrantfile_content)
|
|
env.create_vagrant_env
|
|
end
|
|
|
|
subject { described_class.new(argv, iso_env) }
|
|
|
|
let(:action_runner) { double("action_runner") }
|
|
let(:machine) { iso_env.machine(iso_env.machine_names[0], :dummy) }
|
|
let(:machine2) { iso_env.machine(iso_env.machine_names[0], :dummy) }
|
|
|
|
def new_entry(name)
|
|
entry_klass.new.tap do |e|
|
|
e.name = name
|
|
e.vagrantfile_path = "/bar"
|
|
end
|
|
end
|
|
|
|
before do
|
|
allow(iso_env).to receive(:action_runner).and_return(action_runner)
|
|
allow(subject).to receive(:with_target_vms) { |&block| block.call machine }
|
|
end
|
|
|
|
context "with no argument" do
|
|
let(:vagrantfile_content) do
|
|
<<-VF
|
|
Vagrant.configure("2") do |config|
|
|
config.vm.define "app"
|
|
config.vm.define "db"
|
|
end
|
|
VF
|
|
end
|
|
|
|
it "should suspend all vms" do
|
|
allow(subject).to receive(:with_target_vms) { |&block|
|
|
block.call machine
|
|
block.call machine2
|
|
}
|
|
expect(machine).to receive(:action) do |name, opts|
|
|
expect(name).to eq(:suspend)
|
|
end
|
|
expect(machine2).to receive(:action) do |name, opts|
|
|
expect(name).to eq(:suspend)
|
|
end
|
|
|
|
expect(subject.execute).to eq(0)
|
|
end
|
|
end
|
|
|
|
context "with an argument" do
|
|
let(:vagrantfile_content) do
|
|
<<-VF
|
|
Vagrant.configure("2") do |config|
|
|
config.vm.define "app"
|
|
config.vm.define "db"
|
|
end
|
|
VF
|
|
end
|
|
let(:argv) { ["app"] }
|
|
|
|
it "should suspend a vm" do
|
|
expect(machine).to receive(:action) do |name, opts|
|
|
expect(name).to eq(:suspend)
|
|
end
|
|
|
|
expect(subject.execute).to eq(0)
|
|
end
|
|
end
|
|
|
|
context "with the global all flag" do
|
|
let(:argv){ ["--all-global"] }
|
|
|
|
it "should suspend all vms globally" do
|
|
global_env = isolated_environment
|
|
global_env.vagrantfile("Vagrant.configure(2){|config| config.vm.box = 'dummy'}")
|
|
global_venv = global_env.create_vagrant_env
|
|
global_machine = global_venv.machine(global_venv.machine_names[0], :dummy)
|
|
global_machine.id = "1234"
|
|
global = new_entry(global_machine.name)
|
|
global.provider = "dummy"
|
|
global.vagrantfile_path = global_env.workdir
|
|
locked = iso_env.machine_index.set(global)
|
|
iso_env.machine_index.release(locked)
|
|
|
|
allow(subject).to receive(:with_target_vms) { |&block| block.call global_machine }
|
|
expect(global_machine).to receive(:action) do |name, opts|
|
|
expect(name).to eq(:suspend)
|
|
end
|
|
|
|
expect(subject.execute).to eq(0)
|
|
end
|
|
|
|
context "with an argument is used" do
|
|
let(:argv){ ["machine", "--all-global"] }
|
|
|
|
it "errors out" do
|
|
expect{subject.execute}.to raise_error(Vagrant::Errors::CommandSuspendAllArgs)
|
|
end
|
|
end
|
|
end
|
|
end
|