Merge pull request #10784 from briancain/fixup-snapshot-list-virtualbox

Ensure non-existent machines do not attempt to list snapshots
This commit is contained in:
Brian Cain 2019-04-09 15:03:50 -07:00 committed by GitHub
commit 8109433885
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 19 additions and 1 deletions

View File

@ -23,8 +23,10 @@ module VagrantPlugins
opts = OptionParser.new do |o| opts = OptionParser.new do |o|
o.banner = "Usage: vagrant snapshot pop [options] [vm-name]" o.banner = "Usage: vagrant snapshot pop [options] [vm-name]"
o.separator "" o.separator ""
o.separator "Restore state that was pushed onto the snapshot stack"
o.separator "with `vagrant snapshot push`."
o.separator ""
build_start_options(o, options) build_start_options(o, options)
o.separator "Restore state that was pushed with `vagrant snapshot push`."
o.on("--no-delete", "Don't delete the snapshot after the restore") do o.on("--no-delete", "Don't delete the snapshot after the restore") do
options[:snapshot_delete] = false options[:snapshot_delete] = false

View File

@ -29,6 +29,7 @@ module VagrantPlugins
# #
# @return [Array<String>] Snapshot Name # @return [Array<String>] Snapshot Name
def self.snapshot_list(machine) def self.snapshot_list(machine)
return [] if machine.id.nil?
machine.provider.driver.list_snapshots(machine.id) machine.provider.driver.list_snapshots(machine.id)
end end
end end

View File

@ -40,4 +40,19 @@ describe VagrantPlugins::ProviderVirtualBox::Cap do
expect(described_class.forwarded_ports(machine)).to be(nil) expect(described_class.forwarded_ports(machine)).to be(nil)
end end
end end
describe "#snapshot_list" do
it "returns all the snapshots" do
allow(machine).to receive(:id).and_return("1234")
allow(driver).to receive(:list_snapshots).with(machine.id).
and_return(["backup", "old"])
expect(described_class.snapshot_list(machine)).to eq(["backup", "old"])
end
it "returns empty array when the machine is does not exist" do
allow(machine).to receive(:id).and_return(nil)
expect(described_class.snapshot_list(machine)).to eq([])
end
end
end end