Refine Slice read error test

This commit is contained in:
Brandon Dyck 2024-09-13 11:02:13 -06:00
parent 552b338381
commit 4761259567
2 changed files with 29 additions and 3 deletions

View File

@ -53,8 +53,7 @@ func TestSlice(t *testing.T) {
})) }))
t.Run("fails when read fails", rapid.MakeCheck(func(t *rapid.T) { t.Run("fails when read fails", rapid.MakeCheck(func(t *rapid.T) {
expectedErr := rapid.Map(rapid.StringN(0, 100, -1), errors.New).Draw(t, "err") expectedErr := rapid.Map(rapid.StringN(0, 100, -1), errors.New).Draw(t, "err")
r := ptest.ErrReaderAt(expectedErr) c := ptest.ErrCursor[byte](expectedErr)
c := cursor.NewReaderAt(r)
s := rapid.SliceOfN(rapid.Byte(), 0, 100).Draw(t, "s") s := rapid.SliceOfN(rapid.Byte(), 0, 100).Draw(t, "s")
result, err := gigaparsec.Slice(s)(gigaparsec.MakeState(c)) result, err := gigaparsec.Slice(s)(gigaparsec.MakeState(c))
failed, _, _ := result.Failed() failed, _, _ := result.Failed()

View File

@ -1,7 +1,11 @@
// Package test contains helpers for testing parsers. // Package test contains helpers for testing parsers.
package test package test
import "io" import (
"io"
"git.codemonkeysoftware.net/b/gigaparsec/cursor"
)
type errReaderAt struct { type errReaderAt struct {
err error err error
@ -15,3 +19,26 @@ func (r errReaderAt) ReadAt([]byte, int64) (int, error) {
func ErrReaderAt(err error) io.ReaderAt { func ErrReaderAt(err error) io.ReaderAt {
return errReaderAt{err: err} return errReaderAt{err: err}
} }
type errCursor[T any] struct {
err error
pos uint64
}
func (c errCursor[T]) Read([]T) (uint64, cursor.Cursor[T], error) {
return 0, c, c.err
}
func (c errCursor[T]) At(pos uint64) cursor.Cursor[T] {
c.pos = pos
return c
}
func (c errCursor[T]) Pos() uint64 {
return c.pos
}
// ErrCursor return a [cursor.Cursor] with a Read method that always returns err.
func ErrCursor[T any](err error) cursor.Cursor[T] {
return errCursor[T]{err: err}
}