base: improve coverage

pkg/base/body_test.go

@@ -3,11 +3,25 @@ package base
 import (
 	"bufio"
 	"bytes"
+	"fmt"
 	"testing"
 
 	"github.com/stretchr/testify/require"
 )
 
+type limitedBuffer struct {
+	cap int
+	n   int
+}
+
+func (b *limitedBuffer) Write(p []byte) (int, error) {
+	b.n += len(p)
+	if b.n > b.cap {
+		return 0, fmt.Errorf("capacity reached")
+	}
+	return len(p), nil
+}
+
 var casesBody = []struct {
 	name string
 	h    Header
@@ -90,3 +104,9 @@ func TestBodyReadErrors(t *testing.T) {
 		})
 	}
 }
+
+func TestBodyWriteErrors(t *testing.T) {
+	bw := bufio.NewWriterSize(&limitedBuffer{cap: 3}, 1)
+	err := body([]byte("1234")).write(bw)
+	require.Equal(t, "capacity reached", err.Error())
+}