Customize action

lib/vagrant/action/builtin.rb

@@ -7,6 +7,7 @@ module Vagrant
     def self.builtin!
       up = Builder.new do
         use VM::Import
+        use VM::Customize
       end
 
       register :up, up

lib/vagrant/action/vm/customize.rb

@@ -0,0 +1,21 @@
+module Vagrant
+  class Action
+    module VM
+      class Customize
+        def initialize(app, env)
+          @app = app
+        end
+
+        def call(env)
+          if !env.env.config.vm.proc_stack.empty?
+            env.logger.info "Running any VM customizations..."
+            env.env.config.vm.run_procs!(env["vm"])
+            env["vm"].save
+          end
+
+          @app.call(env)
+        end
+      end
+    end
+  end
+end

test/vagrant/action/vm/customize_test.rb

@@ -0,0 +1,26 @@
+require File.join(File.dirname(__FILE__), '..', '..', '..', 'test_helper')
+
+class CustomizeVMActionTest < Test::Unit::TestCase
+  setup do
+    @klass = Vagrant::Action::VM::Customize
+    @app, @env = mock_action_data
+    @instance = @klass.new(@app, @env)
+
+    @vm = mock("vm")
+    @env["vm"] = @vm
+  end
+
+  should "not run anything if no customize blocks exist" do
+    @vm.expects(:save).never
+    @app.expects(:call).with(@env).once
+    @instance.call(@env)
+  end
+
+  should "run the VM customization procs then save the VM" do
+    @env.env.config.vm.customize { |vm| }
+    @env.env.config.vm.expects(:run_procs!).with(@vm)
+    @vm.expects(:save).once
+    @app.expects(:call).with(@env).once
+    @instance.call(@env)
+  end
+end