98 lines
2.7 KiB
Ruby
98 lines
2.7 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(: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
|
||
|
|
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
|
||
|
|
end
|
||
|
|
end
|