packer/rpc/ui_test.go: test progress bar too

packer/rpc/ui_test.go

@@ -3,20 +3,23 @@ package rpc
 import (
 	"reflect"
 	"testing"
+
+	"github.com/hashicorp/packer/packer"
 )
 
 type testUi struct {
-	askCalled      bool
+	askCalled         bool
-	askQuery       string
+	askQuery          string
-	errorCalled    bool
+	errorCalled       bool
-	errorMessage   string
+	errorMessage      string
-	machineCalled  bool
+	machineCalled     bool
-	machineType    string
+	machineType       string
-	machineArgs    []string
+	machineArgs       []string
-	messageCalled  bool
+	messageCalled     bool
-	messageMessage string
+	messageMessage    string
-	sayCalled      bool
+	sayCalled         bool
-	sayMessage     string
+	sayMessage        string
+	progressBarCalled bool
 }
 
 func (u *testUi) Ask(query string) (string, error) {
@@ -46,6 +49,11 @@ func (u *testUi) Say(message string) {
 	u.sayMessage = message
 }
 
+func (u *testUi) ProgressBar() packer.ProgressBar {
+	u.progressBarCalled = true
+	return new(packer.NoopProgressBar)
+}
+
 func TestUiRPC(t *testing.T) {
 	// Create the UI to test
 	ui := new(testUi)
@@ -87,6 +95,10 @@ func TestUiRPC(t *testing.T) {
 	if ui.sayMessage != "message" {
 		t.Fatalf("bad: %#v", ui.errorMessage)
 	}
+	uiClient.ProgressBar()
+	if ui.progressBarCalled != true {
+		t.Fatalf("ProgressBar not called.")
+	}
 
 	uiClient.Machine("foo", "bar", "baz")
 	if !ui.machineCalled {