From b66446712c3064ba1b362659e260067ede5cf15f Mon Sep 17 00:00:00 2001
From: Chris Roberts <code@chrisroberts.org>
Date: Thu, 9 Mar 2017 10:27:50 -0800
Subject: [PATCH] Start adding test coverage on port collision handling action

---
 .../handle_forwarded_port_collisions_test.rb  | 109 ++++++++++++++++++
 1 file changed, 109 insertions(+)
 create mode 100644 test/unit/vagrant/action/builtin/handle_forwarded_port_collisions_test.rb

diff --git a/test/unit/vagrant/action/builtin/handle_forwarded_port_collisions_test.rb b/test/unit/vagrant/action/builtin/handle_forwarded_port_collisions_test.rb
new file mode 100644
index 000000000..8ffb08339
--- /dev/null
+++ b/test/unit/vagrant/action/builtin/handle_forwarded_port_collisions_test.rb
@@ -0,0 +1,109 @@
+require File.expand_path("../../../../base", __FILE__)
+
+require Vagrant.source_root.join("plugins/kernel_v2/config/vm")
+
+describe Vagrant::Action::Builtin::HandleForwardedPortCollisions do
+  include_context "unit"
+
+  let(:app) { lambda { |env| } }
+  let(:env) {
+    { machine: machine, ui: ui, port_collision_extra_in_use: extra_in_use,
+      port_collision_remap: collision_remap, port_collision_repair: collision_repair,
+      port_collision_port_check: collision_port_check }
+  }
+  let(:extra_in_use){ nil }
+  let(:collision_remap){ nil }
+  let(:collision_repair){ nil }
+  let(:collision_port_check){ nil }
+
+  let(:machine) do
+    double("machine").tap do |machine|
+      allow(machine).to receive(:config).and_return(machine_config)
+      allow(machine).to receive(:env).and_return(machine_env)
+    end
+  end
+
+  let(:machine_config) do
+    double("machine_config").tap do |config|
+      config.stub(vm: vm_config)
+    end
+  end
+
+  let(:data_dir){ temporary_dir }
+
+  let(:machine_env) do
+    isolated_environment.tap do |i_env|
+      allow(i_env).to receive(:data_dir).and_return(data_dir)
+      allow(i_env).to receive(:lock).and_yield
+    end
+  end
+
+  let(:vm_config) do
+    double("machine_vm_config").tap do |config|
+      allow(config).to receive(:usable_port_range).and_return(1000..2000)
+      allow(config).to receive(:networks).and_return([])
+    end
+  end
+
+  let(:ui) do
+    double("ui").tap do |result|
+      allow(result).to receive(:info)
+    end
+  end
+
+  let(:instance){ described_class.new(app, env) }
+
+  describe "#call" do
+    it "should create a lock while action runs" do
+      expect(machine_env).to receive(:lock).with("fpcollision").and_yield
+      instance.call(env)
+    end
+
+    context "with extra ports in use provided as Array type" do
+      let(:extra_in_use){ [80] }
+
+      it "should not generate an error" do
+        expect{ instance.call(env) }.not_to raise_error
+      end
+    end
+
+    context "with forwarded port defined" do
+      let(:port_options){ {guest: 80, host: 8080} }
+      before do
+        expect(vm_config).to receive(:networks).and_return([[:forwarded_port, port_options]]).twice
+      end
+
+      it "should check if host port is in use" do
+        expect(instance).to receive(:is_forwarded_already).and_return false
+        instance.call(env)
+      end
+
+      context "with forwarded port already in use" do
+        let(:extra_in_use){ [8080] }
+
+        it "should raise a port collision error" do
+          expect{ instance.call(env) }.to raise_error(Vagrant::Errors::ForwardPortCollision)
+        end
+
+        context "with auto_correct enabled" do
+          before{ port_options[:auto_correct] = true }
+
+          it "should raise a port collision error" do
+            expect{ instance.call(env) }.to raise_error(Vagrant::Errors::ForwardPortCollision)
+          end
+
+          context "with collision repair enabled" do
+            let(:collision_repair){ true }
+
+            it "should automatically correct collision" do
+              expect{ instance.call(env) }.not_to raise_error
+            end
+          end
+        end
+      end
+    end
+  end
+
+  describe "#recover" do
+  end
+end