From ae6f086fadb229da8791764575ca643e7549e247 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Sat, 13 Feb 2010 12:17:59 -0800
Subject: [PATCH] Forwarding ports action

---
 lib/vagrant/actions/forward_ports.rb       | 4 ++--
 test/vagrant/actions/forward_ports_test.rb | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/vagrant/actions/forward_ports.rb b/lib/vagrant/actions/forward_ports.rb
index 767dac06d..29b59f45b 100644
--- a/lib/vagrant/actions/forward_ports.rb
+++ b/lib/vagrant/actions/forward_ports.rb
@@ -10,10 +10,10 @@ module Vagrant
           port.name = name
           port.hostport = options[:hostport]
           port.guestport = options[:guestport]
-          @vm.vm.forwarded_ports << port
+          @vm.forwarded_ports << port
         end
 
-        @vm.vm.save(true)
+        @vm.save(true)
       end
     end
   end
diff --git a/test/vagrant/actions/forward_ports_test.rb b/test/vagrant/actions/forward_ports_test.rb
index 8239fa6ce..7cbc55d1c 100644
--- a/test/vagrant/actions/forward_ports_test.rb
+++ b/test/vagrant/actions/forward_ports_test.rb
@@ -18,8 +18,8 @@ class ForwardPortsActionTest < Test::Unit::TestCase
       end
     end
 
-    @vm.expects(:forwarded_ports).returns(forwarded_ports)
-    @vm.expects(:save).with(true).once
+    @mock_vm.expects(:forwarded_ports).returns(forwarded_ports)
+    @mock_vm.expects(:save).with(true).once
     @action.execute!
   end
 end