Function TestReader_ReadStringInto_BufferTooSmall

internal/proto/reader_test.go:320–348

(t *testing.T)

}

func TestReader_ReadStringInto_BufferTooSmall(t *testing.T) {
	// Feed two consecutive bulk-string replies. The first read uses a
	// too-small buf and must error; the second must succeed cleanly. This
	// guards the drain-on-error behaviour — without it, the unread payload
	// of the first reply would leak into the stream and the second read
	// would parse garbage.
	src := []byte("$5\r\nhello\r\n$5\r\nworld\r\n")
	r := proto.NewReader(bytes.NewReader(src))

	smallBuf := make([]byte, 3)
	n, err := r.ReadStringInto(smallBuf)
	if err == nil {
		t.Fatal("expected buffer-too-small error, got nil")
	}
	if n != 0 {
		t.Fatalf("got n=%d on error, want 0", n)
	}

	// After the drain, the reader is aligned at the start of the second
	// reply. A normal read must parse it as "world".
	bigBuf := make([]byte, 16)
	n, err = r.ReadStringInto(bigBuf)
	if err != nil {
		t.Fatalf("follow-up ReadStringInto: %v (connection left misaligned by the buffer-too-small drain)", err)
	}
	if n != 5 || string(bigBuf[:n]) != "world" {
		t.Fatalf("follow-up: got n=%d %q, want 5 \"world\"", n, bigBuf[:n])
	}
}

func TestReader_ReadStringInto_Large(t *testing.T) {
	// Payload deliberately larger than the default bufio buffer (32 KiB)