Forwarding ports action

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

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