c999e7c3d4
Both of these commands failed to default the options disabling the provisioning from ignoring the sentinel file. This resulted in different behavior than what was seen with the `up` and `resume` commands which would only provision items with run set to "always". This defaults the options to proper match the behavior of `up` and `resume` to be consistent. This also adds an extra `--no-start` flag to allow users to restore a snapshot but not start the restored guest immediately. Fixes #6752
131 lines
3.6 KiB
Ruby
131 lines
3.6 KiB
Ruby
require File.expand_path("../../../../../base", __FILE__)
|
|
|
|
require Vagrant.source_root.join("plugins/commands/snapshot/command/restore")
|
|
|
|
describe VagrantPlugins::CommandSnapshot::Command::Restore do
|
|
include_context "unit"
|
|
|
|
let(:iso_env) do
|
|
# We have to create a Vagrantfile so there is a root path
|
|
env = isolated_environment
|
|
env.vagrantfile("")
|
|
env.create_vagrant_env
|
|
end
|
|
|
|
let(:snapshot_name) { "snapshot_name" }
|
|
let(:guest) { double("guest") }
|
|
let(:host) { double("host") }
|
|
let(:machine) { iso_env.machine(iso_env.machine_names[0], :dummy) }
|
|
|
|
let(:argv) { [] }
|
|
|
|
subject { described_class.new(argv, iso_env) }
|
|
|
|
before do
|
|
allow(machine.provider).to receive(:capability).with(:snapshot_list).
|
|
and_return([])
|
|
|
|
allow(machine.provider).to receive(:capability?).with(:snapshot_list).
|
|
and_return(true)
|
|
|
|
allow(subject).to receive(:with_target_vms) { |&block| block.call machine }
|
|
end
|
|
|
|
describe "execute" do
|
|
before do
|
|
allow(machine.provider).to receive(:capability).
|
|
with(:snapshot_list).and_return([snapshot_name])
|
|
end
|
|
|
|
context "with no arguments" do
|
|
it "shows help" do
|
|
expect { subject.execute }.
|
|
to raise_error(Vagrant::Errors::CLIInvalidUsage)
|
|
end
|
|
end
|
|
|
|
context "with an unsupported provider" do
|
|
let(:argv) { ["test"] }
|
|
|
|
before do
|
|
allow(machine.provider).to receive(:capability?).with(:snapshot_list).
|
|
and_return(false)
|
|
end
|
|
|
|
it "raises an exception" do
|
|
machine.id = "foo"
|
|
expect { subject.execute }.
|
|
to raise_error(Vagrant::Errors::SnapshotNotSupported)
|
|
end
|
|
end
|
|
|
|
context "with a snapshot name given" do
|
|
let(:argv) { ["test"] }
|
|
it "calls snapshot_delete with a snapshot name" do
|
|
machine.id = "foo"
|
|
|
|
allow(machine.provider).to receive(:capability).with(:snapshot_list).
|
|
and_return(["test"])
|
|
|
|
expect(machine).to receive(:action) do |name, opts|
|
|
expect(name).to eq(:snapshot_restore)
|
|
expect(opts[:snapshot_name]).to eq("test")
|
|
end
|
|
|
|
expect(subject.execute).to eq(0)
|
|
end
|
|
|
|
it "doesn't delete a snapshot on a non-existent machine" do
|
|
machine.id = nil
|
|
|
|
expect(subject).to receive(:with_target_vms){}
|
|
|
|
expect(machine).to_not receive(:action)
|
|
expect(subject.execute).to eq(0)
|
|
end
|
|
end
|
|
|
|
context "with a snapshot name that doesn't exist" do
|
|
let(:argv) { ["nopetest"] }
|
|
|
|
it "fails to take a snapshot and prints a warning to the user" do
|
|
machine.id = "foo"
|
|
|
|
allow(machine.provider).to receive(:capability).with(:snapshot_list).
|
|
and_return(["test"])
|
|
|
|
expect(machine).to_not receive(:action)
|
|
expect { subject.execute }.
|
|
to raise_error(Vagrant::Errors::SnapshotNotFound)
|
|
end
|
|
end
|
|
|
|
it "should disable ignoring sentinel file for provisioning" do
|
|
argv << snapshot_name
|
|
expect(machine).to receive(:action) do |_, opts|
|
|
expect(opts[:provision_ignore_sentinel]).to eq(false)
|
|
end
|
|
subject.execute
|
|
end
|
|
|
|
it "should start the snapshot" do
|
|
argv << snapshot_name
|
|
expect(machine).to receive(:action) do |_, opts|
|
|
expect(opts[:snapshot_start]).to eq(true)
|
|
end
|
|
subject.execute
|
|
end
|
|
|
|
context "when --no-start flag is provided" do
|
|
let(:argv) { [snapshot_name, "--no-start"] }
|
|
|
|
it "should not start the snapshot" do
|
|
expect(machine).to receive(:action) do |_, opts|
|
|
expect(opts[:snapshot_start]).to eq(false)
|
|
end
|
|
subject.execute
|
|
end
|
|
end
|
|
end
|
|
end
|