packer/rpc: Implement Builder.Cancel

packer/rpc/builder.go

@@ -93,7 +93,9 @@ func (b *builder) Run(ui packer.Ui, hook packer.Hook) packer.Artifact {
 }
 
 func (b *builder) Cancel() {
-
+	if err := b.client.Call("Builder.Cancel", new(interface{}), new(interface{})); err != nil {
+		panic(err)
+	}
 }
 
 func (b *BuilderServer) Prepare(args *BuilderPrepareArgs, reply *error) error {
@@ -135,3 +137,8 @@ func (b *BuilderServer) Run(args *BuilderRunArgs, reply *interface{}) error {
 
 	return nil
 }
+
+func (b *BuilderServer) Cancel(args *interface{}, reply*interface{}) error {
+	b.builder.Cancel()
+	return nil
+}

packer/rpc/builder_test.go

@@ -73,6 +73,10 @@ func TestBuilderRPC(t *testing.T) {
 
 		assert.Equal(artifact.Id(), testBuilderArtifact.Id(), "should have artifact Id")
 	}
+
+	// Test Cancel
+	bClient.Cancel()
+	assert.True(b.cancelCalled, "cancel should be called")
 }
 
 func TestBuilder_ImplementsBuilder(t *testing.T) {

packer/rpc/server.go

@@ -66,7 +66,7 @@ func serveSingleConn(s *rpc.Server) string {
 		defer l.Close()
 		conn, err := l.Accept()
 		if err != nil {
-			return
+			panic(err)
 		}
 
 		s.ServeConn(conn)