mirror of
https://github.com/wader/fq.git
synced 2024-11-24 11:16:09 +03:00
58 lines
993 B
Go
58 lines
993 B
Go
package aheadreadseeker_test
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/wader/fq/internal/aheadreadseeker"
|
|
)
|
|
|
|
type readRecord struct {
|
|
pLen int
|
|
off int64
|
|
retP []byte
|
|
err error
|
|
}
|
|
|
|
type recordingReadSeeker struct {
|
|
rs io.ReadSeeker
|
|
off int64
|
|
records []readRecord
|
|
}
|
|
|
|
func (r *recordingReadSeeker) Read(p []byte) (n int, err error) {
|
|
n, err = r.rs.Read(p)
|
|
retP := make([]byte, n)
|
|
copy(retP, p)
|
|
r.records = append(r.records, readRecord{
|
|
pLen: len(p),
|
|
off: r.off,
|
|
retP: retP,
|
|
err: err,
|
|
})
|
|
return n, err
|
|
|
|
}
|
|
|
|
func (r *recordingReadSeeker) Seek(offset int64, whence int) (int64, error) {
|
|
off, err := r.rs.Seek(offset, whence)
|
|
r.off = off
|
|
return off, err
|
|
}
|
|
|
|
func TestNewReadAtCache(t *testing.T) {
|
|
rrs := &recordingReadSeeker{rs: bytes.NewReader([]byte("abc"))}
|
|
r := aheadreadseeker.New(rrs, 2)
|
|
|
|
b := make([]byte, 1)
|
|
_, _ = r.Read(b)
|
|
_, _ = r.Read(b)
|
|
_, _ = r.Read(b)
|
|
|
|
log.Printf("b: %s\n", b)
|
|
|
|
log.Printf("rrs.records: %#+v\n", rrs.records)
|
|
}
|