diff --git a/packer/rpc/builder_test.go b/packer/rpc/builder_test.go index 402cb6920..fa20a4b85 100644 --- a/packer/rpc/builder_test.go +++ b/packer/rpc/builder_test.go @@ -30,7 +30,14 @@ func TestBuilderPrepare(t *testing.T) { // Test Prepare config := 42 - bClient.Prepare(config) + warnings, err := bClient.Prepare(config) + if err != nil { + t.Fatalf("bad: %s", err) + } + if len(warnings) > 0 { + t.Fatalf("bad: %#v", warnings) + } + if !b.PrepareCalled { t.Fatal("should be called") } @@ -40,6 +47,22 @@ func TestBuilderPrepare(t *testing.T) { } } +func TestBuilderPrepare_Warnings(t *testing.T) { + b, bClient := builderRPCClient(t) + + expected := []string{"foo"} + b.PrepareWarnings = expected + + // Test Prepare + warnings, err := bClient.Prepare(nil) + if err != nil { + t.Fatalf("bad: %s", err) + } + if !reflect.DeepEqual(warnings, expected) { + t.Fatalf("bad: %#v", warnings) + } +} + func TestBuilderRun(t *testing.T) { b, bClient := builderRPCClient(t)