package util import ( "github.com/stretchr/testify/require" "io" "strings" "testing" ) func TestPeak_LimitReached(t *testing.T) { underlying := io.NopCloser(strings.NewReader("1234567890")) peaked, err := Peak(underlying, 5) if err != nil { t.Fatal(err) } require.Equal(t, []byte("12345"), peaked.PeakedBytes) require.Equal(t, true, peaked.LimitReached) all, err := io.ReadAll(peaked) if err != nil { t.Fatal(err) } require.Equal(t, []byte("1234567890"), all) require.Equal(t, []byte("12345"), peaked.PeakedBytes) require.Equal(t, true, peaked.LimitReached) } func TestPeak_LimitNotReached(t *testing.T) { underlying := io.NopCloser(strings.NewReader("1234567890")) peaked, err := Peak(underlying, 15) if err != nil { t.Fatal(err) } all, err := io.ReadAll(peaked) if err != nil { t.Fatal(err) } require.Equal(t, []byte("1234567890"), all) require.Equal(t, []byte("1234567890"), peaked.PeakedBytes) require.Equal(t, false, peaked.LimitReached) } func TestPeak_Nil(t *testing.T) { peaked, err := Peak(nil, 15) if err != nil { t.Fatal(err) } all, err := io.ReadAll(peaked) if err != nil { t.Fatal(err) } require.Equal(t, []byte(""), all) require.Equal(t, []byte(""), peaked.PeakedBytes) require.Equal(t, false, peaked.LimitReached) }