fix: Broken pipe in FormatPath (#94)

* test: enable TestFormatPath

* fix: Broken pipe in FormatPath

pkg/kclvm_runtime/proc.go

@@ -4,7 +4,6 @@ package kclvm_runtime
 
 import (
 	"bytes"
-	"errors"
 	"io"
 	"net/rpc"
 	"os/exec"
@@ -106,15 +105,22 @@ func newLimitBuffer(cap int) *limitBuffer {
 	return &limitBuffer{cap: cap}
 }
 
-func (b *limitBuffer) Write(p []byte) (n int, err error) {
-	n = b.cap - b.buf.Len()
+func (b *limitBuffer) Write(p []byte) (int, error) {
+	n := b.cap - b.buf.Len()
 	if n > 0 {
-		b.buf.Write(p[:n])
+		if n > len(p) {
+			n = len(p)
+		}
+		var err error
+		n, err = b.buf.Write(p[:n])
+		if err != nil {
+			return n, err
+		}
 	}
 	if n < len(p) {
-		err = errors.New("limitBuffer: overflow")
+		return n, io.ErrShortWrite
 	}
-	return n, err
+	return n, nil
 }
 
 func (b *limitBuffer) Read(p []byte) (n int, err error) {

pkg/tools/format/format_test.go

@@ -5,6 +5,7 @@ import (
 	"io/fs"
 	"io/ioutil"
 	"path"
+	"strings"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -37,9 +38,6 @@ func TestFormatCode(t *testing.T) {
 	}
 }
 
-// TODO: fix Broken pipe in kclvm_py, see in :
-// https://github.com/KusionStack/kclvm-go/issues/75
-/*
 func TestFormatPath(t *testing.T) {
 	successDir := "testdata/success"
 	expectedFileSuffix := ".formatted"
@@ -90,7 +88,6 @@ func TestFormatPath(t *testing.T) {
 		assert.Equal(t, expected, get, fmt.Sprintf("format path get wrong result. formatted content mismatch, file: %s, expect: %s, get: %s", actualFile, expected, get))
 	}
 }
-*/
 
 type filterFile func(fs.FileInfo) bool