fix: update

Signed-off-by: Junjie Gao <[email protected]>

internal/io/limitedwriter.go

@@ -16,7 +16,12 @@
 
 package io
 
-import "io"
+import (
+	"errors"
+	"io"
+)
+
+var ErrLimitExceeded = errors.New("write limit exceeded")
 
 // LimitedWriter is a writer that writes to an underlying writer up to a limit.
 type LimitedWriter struct {
@@ -36,7 +41,7 @@ func LimitWriter(w io.Writer, limit int64) *LimitedWriter {
 // Write writes p to the underlying writer up to the limit.
 func (l *LimitedWriter) Write(p []byte) (int, error) {
 	if l.N <= 0 {
-		return 0, io.ErrShortWrite
+		return 0, ErrLimitExceeded
 	}
 	if int64(len(p)) > l.N {
 		p = p[:l.N]

internal/io/limitedwriter_test.go

@@ -15,7 +15,7 @@ package io
 
 import (
 	"bytes"
-	"io"
+	"errors"
 	"testing"
 )
 
@@ -60,8 +60,8 @@ func TestLimitWriterFailed(t *testing.T) {
 		t.Fatalf("unexpected error: %v", err)
 	}
 	_, err = lw.Write([]byte(longString))
-	expectedErr := io.ErrShortWrite
-	if err != expectedErr {
+	expectedErr := errors.New("write limit exceeded")
+	if err.Error() != expectedErr.Error() {
 		t.Errorf("expected error %v, got %v", expectedErr, err)
 	}
 }

plugin/plugin.go

@@ -222,6 +222,8 @@ func (c execCommander) Output(ctx context.Context, name string, command plugin.C
 	var stdout, stderr bytes.Buffer
 	cmd := exec.CommandContext(ctx, name, string(command))
 	cmd.Stdin = bytes.NewReader(req)
+	// The limit writer will be handled by the caller in run() by comparing the
+	// bytes written with the expected length of the bytes.
 	cmd.Stderr = io.LimitWriter(&stderr, maxPluginOutputSize)
 	cmd.Stdout = io.LimitWriter(&stdout, maxPluginOutputSize)
 	err := cmd.Run()