b218e62ffc
This can be redone manually with `pre-commit run --all` While the pre-commit hook could be merged to run locally, it is much cleaner to align all the files to best-practice syntax in a single commit. It is also required for server-side validation.
56 lines
1.3 KiB
Go
56 lines
1.3 KiB
Go
package util
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPeak_LimitReached(t *testing.T) {
|
|
underlying := io.NopCloser(strings.NewReader("1234567890"))
|
|
peaked, err := Peek(underlying, 5)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
require.Equal(t, []byte("12345"), peaked.PeekedBytes)
|
|
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.PeekedBytes)
|
|
require.Equal(t, true, peaked.LimitReached)
|
|
}
|
|
|
|
func TestPeak_LimitNotReached(t *testing.T) {
|
|
underlying := io.NopCloser(strings.NewReader("1234567890"))
|
|
peaked, err := Peek(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.PeekedBytes)
|
|
require.Equal(t, false, peaked.LimitReached)
|
|
}
|
|
|
|
func TestPeak_Nil(t *testing.T) {
|
|
peaked, err := Peek(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.PeekedBytes)
|
|
require.Equal(t, false, peaked.LimitReached)
|
|
}
|