package main

import (
	"bufio"
	"strings"
	"testing"
)

func TestBytesBufferUnreadByte(t *testing.T) {
	check := func(err error) {
		if err != nil {
			t.Helper()
			t.Fatal(err)
		}
	}

	const in = "abc\n"
	t.Logf("reading %q", in)

	b := bufio.NewReader(strings.NewReader(in))

	c, err := b.ReadByte()
	check(err)
	t.Logf("read byte %q", c)

	n, err := b.Discard(1)
	check(err)
	t.Logf("discarded %v byte(s)", n)

	err = b.UnreadByte()
	t.Logf("UnreadByte: %v", err)
	var want string
	if err == nil {
		want = in[1:]
	} else {
		want = in[2:]
	}

	s, err := b.ReadString('\n')
	check(err)
	if s != want {
		t.Fatalf("buffer corrupted: read %q, expected %q", s, want)
	}
}