Add tests for prune command

test/unit/plugins/commands/box/command/prune_test.rb

@@ -0,0 +1,194 @@
+require File.expand_path("../../../../../base", __FILE__)
+
+require Vagrant.source_root.join("plugins/commands/box/command/prune")
+
+describe VagrantPlugins::CommandBox::Command::Prune do
+  include_context "unit"
+  include_context "command plugin helpers"
+
+  let(:entry_klass) { Vagrant::MachineIndex::Entry }
+
+  let(:iso_env) do
+    # We have to create a Vagrantfile so there is a root path
+    isolated_environment.tap do |env|
+      env.vagrantfile("")
+    end
+  end
+
+  let(:iso_vagrant_env) { iso_env.create_vagrant_env }
+
+  let(:argv) { [] }
+
+  # Seems this way of providing a box version triggers box in use.
+  def new_entry(name, box_name, box_provider, version)
+    entry_klass.new.tap do |e|
+      e.name = name
+      e.vagrantfile_path = "/bar"
+      e.extra_data["box"] = {
+          "name" => box_name,
+          "provider" => box_provider,
+          "version" => version,
+      }
+    end
+  end
+
+  subject { described_class.new(argv, iso_vagrant_env) }
+
+  describe "execute" do
+    context "with no args" do
+      it "removes the old version and keeps the current one" do
+
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("foobox", "1.1", :virtualbox);
+        iso_env.box3("barbox", "1.0", :vmware);
+        iso_env.box3("barbox", "1.1", :vmware);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << data
+        end
+        expect(iso_vagrant_env.boxes.all.count).to eq(4)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(2)
+
+        expect(output).to include("barbox (vmware, 1.1)")
+        expect(output).to include("Removing box 'barbox' (v1.0) with provider 'vmware'...")
+        expect(output).to include("foobox (virtualbox, 1.1)")
+        expect(output).to include("Removing box 'foobox' (v1.0) with provider 'virtualbox'...")
+      end
+
+      it "removes nothing" do
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("barbox", "1.0", :vmware);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << data
+        end
+        expect(iso_vagrant_env.boxes.all.count).to eq(2)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(2)
+
+        expect(output).to include("No old versions of boxes to remove...")
+
+      end
+    end
+
+    context "with --provider" do
+      let(:argv) { ["--provider", "virtualbox"] }
+
+      it "removes the old versions of the specified provider" do
+
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("foobox", "1.1", :virtualbox);
+        iso_env.box3("barbox", "1.0", :vmware);
+        iso_env.box3("barbox", "1.1", :vmware);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << "\n" + data
+        end
+
+        expect(iso_vagrant_env.boxes.all.count).to eq(4)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(3)
+
+        expect(output).to include("foobox (virtualbox, 1.1)")
+        expect(output).to include("Removing box 'foobox' (v1.0) with provider 'virtualbox'...")
+
+      end
+    end
+
+    context "with --dry-run" do
+      let(:argv) { ["--dry-run"] }
+
+      it "removes the old versions of the specified provider" do
+
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("foobox", "1.1", :virtualbox);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << "\n" + data
+        end
+
+        expect(iso_vagrant_env.boxes.all.count).to eq(2)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(2)
+
+
+        expect(output).to include("foobox (virtualbox, 1.1)")
+        expect(output).to include("Would remove foobox virtualbox 1.0")
+
+
+      end
+    end
+
+    context "with --name" do
+      let(:argv) { ["--name", "barbox"] }
+
+      it "removes the old versions of the specified provider" do
+
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("foobox", "1.1", :virtualbox);
+        iso_env.box3("barbox", "1.0", :vmware);
+        iso_env.box3("barbox", "1.1", :vmware);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << "\n" + data
+        end
+
+        expect(iso_vagrant_env.boxes.all.count).to eq(4)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(3)
+
+        expect(output).to include("barbox (vmware, 1.1)")
+        expect(output).to include("Removing box 'barbox' (v1.0) with provider 'vmware'...")
+      end
+    end
+
+
+    context "with --name and --provider" do
+      let(:argv) { ["--name", "foobox", "--provider", "virtualbox"] }
+
+      it "removed the old versions of that name and provider only" do
+        # Let's put some things in the index
+        iso_env.box3("foobox", "1.0", :virtualbox);
+        iso_env.box3("foobox", "1.1", :virtualbox);
+        iso_env.box3("foobox", "1.0", :vmware);
+        iso_env.box3("foobox", "1.1", :vmware);
+        iso_env.box3("barbox", "1.0", :vmware);
+        iso_env.box3("barbox", "1.1", :vmware);
+
+        iso_vagrant_env.machine_index.set(new_entry("foo", "foobox", "virtualbox", 1))
+
+        output = ""
+        allow(iso_vagrant_env.ui).to receive(:info) do |data|
+          output << "\n" + data
+        end
+
+        expect(iso_vagrant_env.boxes.all.count).to eq(6)
+        expect(subject.execute).to eq(0)
+        expect(iso_vagrant_env.boxes.all.count).to eq(5)
+
+        expect(output).to include("Removing box 'foobox' (v1.0) with provider 'virtualbox'...")
+      end
+    end
+  end
+end