From 8a78d4d8a52f758fbcd7e3c0140a0c12f28f3e30 Mon Sep 17 00:00:00 2001
From: Mitchell Hashimoto <mitchell.hashimoto@gmail.com>
Date: Sat, 4 May 2013 13:50:02 -0700
Subject: [PATCH] packer/rpc: Test Builder.Run

---
 packer/rpc/builder_test.go | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/packer/rpc/builder_test.go b/packer/rpc/builder_test.go
index 499a4fbed..46a4aa6a9 100644
--- a/packer/rpc/builder_test.go
+++ b/packer/rpc/builder_test.go
@@ -48,6 +48,21 @@ func TestBuilderRPC(t *testing.T) {
 	bClient.Prepare(config)
 	assert.True(b.prepareCalled, "prepare should be called")
 	assert.Equal(b.prepareConfig, 42, "prepare should be called with right arg")
+
+	// Test Run
+	build := &testBuild{}
+	ui := &testUi{}
+	bClient.Run(build, ui)
+	assert.True(b.runCalled, "runs hould be called")
+
+	if b.runCalled {
+		b.runBuild.Prepare()
+		assert.True(build.prepareCalled, "prepare should be called")
+
+		b.runUi.Say("format")
+		assert.True(ui.sayCalled, "say should be called")
+		assert.Equal(ui.sayFormat, "format", "format should be correct")
+	}
 }
 
 func TestBuilder_ImplementsBuild(t *testing.T) {