You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
671 B
33 lines
671 B
package page_writer
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestWriteChunkedStream(t *testing.T) {
|
|
x := NewChunkedStreamWriter(20)
|
|
defer x.Destroy()
|
|
y := NewChunkedFileWriter(os.TempDir(), 12)
|
|
defer y.Destroy()
|
|
|
|
batchSize := 4
|
|
buf := make([]byte, batchSize)
|
|
for i := 0; i < 256; i++ {
|
|
for x := 0; x < batchSize; x++ {
|
|
buf[x] = byte(i)
|
|
}
|
|
x.WriteAt(buf, int64(i*batchSize))
|
|
y.WriteAt(buf, int64((255-i)*batchSize))
|
|
}
|
|
|
|
a := make([]byte, 1)
|
|
b := make([]byte, 1)
|
|
for i := 0; i < 256*batchSize; i++ {
|
|
x.ReadDataAt(a, int64(i))
|
|
y.ReadDataAt(b, int64(256*batchSize-1-i))
|
|
assert.Equal(t, a[0], b[0], "same read")
|
|
}
|
|
|
|
}
|