From 1bb9a48ae7de267aefe6532ca51f3aab01420db3 Mon Sep 17 00:00:00 2001
From: Seth Vargo <sethvargo@gmail.com>
Date: Tue, 24 Nov 2015 16:41:13 -0500
Subject: [PATCH] Fix cap test

---
 test/unit/plugins/providers/virtualbox/cap_test.rb | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/test/unit/plugins/providers/virtualbox/cap_test.rb b/test/unit/plugins/providers/virtualbox/cap_test.rb
index baa7f610f..52eb9bc43 100644
--- a/test/unit/plugins/providers/virtualbox/cap_test.rb
+++ b/test/unit/plugins/providers/virtualbox/cap_test.rb
@@ -15,14 +15,16 @@ describe VagrantPlugins::ProviderVirtualBox::Cap do
   let(:machine) do
     iso_env.machine(iso_env.machine_names[0], :dummy).tap do |m|
       m.provider.stub(driver: driver)
+      m.stub(state: state)
     end
   end
 
   let(:driver) { double("driver") }
+  let(:state)  { double("state", id: :running) }
 
   describe "#forwarded_ports" do
     it "returns all the forwarded ports" do
-      expect(driver).to receive(:read_forwarded_ports).and_return([
+      allow(driver).to receive(:read_forwarded_ports).and_return([
         [nil, nil, 123, 456],
         [nil, nil, 245, 245],
       ])
@@ -32,5 +34,10 @@ describe VagrantPlugins::ProviderVirtualBox::Cap do
         245 => 245,
       })
     end
+
+    it "returns nil when the machine is not running" do
+      allow(machine).to receive(:state).and_return(double(:state, id: :stopped))
+      expect(described_class.forwarded_ports(machine)).to be(nil)
+    end
   end
 end