1
1
mirror of https://github.com/wader/fq.git synced 2024-12-24 22:05:31 +03:00
fq/internal/ctxreadseeker/ctxreadseek_test.go
2021-09-12 13:08:50 +02:00

87 lines
1.8 KiB
Go

package ctxreadseeker_test
import (
"context"
"errors"
"fmt"
"io"
"reflect"
"sync"
"testing"
"time"
"github.com/wader/fq/internal/ctxreadseeker"
)
type rwcRecorder struct {
sync.RWMutex
log []string
}
func (r *rwcRecorder) Read(p []byte) (n int, err error) {
r.Lock()
defer r.Unlock()
r.log = append(r.log, fmt.Sprintf("read %d", len(p)))
return 0, err
}
func (r *rwcRecorder) Seek(offset int64, whence int) (n int64, err error) {
r.Lock()
defer r.Unlock()
r.log = append(r.log, fmt.Sprintf("seek %d %d", offset, whence))
return 0, err
}
func (r *rwcRecorder) Close() (err error) {
r.Lock()
defer r.Unlock()
r.log = append(r.log, "close")
return nil
}
func (r *rwcRecorder) Log() []string {
r.RLock()
defer r.RUnlock()
return append([]string(nil), r.log...)
}
func TestNormal(t *testing.T) {
r := &rwcRecorder{}
crs := ctxreadseeker.New(context.Background(), r)
_, _ = crs.Seek(1, io.SeekStart)
_, _ = crs.Read(make([]byte, 3))
_ = crs.Close()
expected := []string{"seek 1 0", "read 3", "close"}
if !reflect.DeepEqual(r.Log(), expected) {
t.Errorf("expected %v, got %v", expected, r.Log())
}
}
func TestCancel(t *testing.T) {
r := &rwcRecorder{}
c, cancelFn := context.WithCancel(context.Background())
crs := ctxreadseeker.New(c, r)
_, _ = crs.Seek(1, io.SeekStart)
_, _ = crs.Read(make([]byte, 3))
cancelFn()
time.Sleep(10 * time.Millisecond)
cancelActualReadN, cancelActualReadErr := crs.Read(make([]byte, 3))
expected := []string{"seek 1 0", "read 3", "close"}
if !reflect.DeepEqual(r.Log(), expected) {
t.Errorf("expected %v, got %v", expected, r.Log())
}
if cancelActualReadN != 0 {
t.Errorf("expected 0, got %v", cancelActualReadN)
}
if !errors.Is(cancelActualReadErr, context.Canceled) {
t.Errorf("expected cancel err, got %v", cancelActualReadErr)
}
}