cerc-io/lotus
16
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

fr32: Fix tests

Browse Source
This commit is contained in:
Łukasz Magiera 2020-05-29 19:06:44 +02:00
parent 2184f21dd2
commit 8b9dbc22e5
3 changed files with 34 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fr32/fr32_ffi_cmp_test.go
View File

@ -1,7 +1,8 @@
package fr32
package fr32_test
import (
"bytes"
"github.com/filecoin-project/sector-storage/fr32"
"io"
"io/ioutil"
"os"
@ -56,10 +57,10 @@ func TestWriteTwoPcs(t *testing.T) {
}
outBytes := make([]byte, int(paddedSize)*n)
Pad(rawBytes, outBytes)
fr32.Pad(rawBytes, outBytes)
require.Equal(t, ffiBytes, outBytes)
unpadBytes := make([]byte, int(paddedSize.Unpadded())*n)
Unpad(ffiBytes, unpadBytes)
fr32.Unpad(ffiBytes, unpadBytes)
require.Equal(t, rawBytes, unpadBytes)
}

44
fr32/fr32_test.go
View File

@ -1,4 +1,4 @@
package fr32
package fr32_test
import (
"bytes"
@ -9,9 +9,11 @@ import (
"testing"
ffi "github.com/filecoin-project/filecoin-ffi"
"github.com/filecoin-project/sector-storage/ffiwrapper"
"github.com/filecoin-project/specs-actors/actors/abi"
"github.com/stretchr/testify/require"
"github.com/filecoin-project/sector-storage/ffiwrapper"
"github.com/filecoin-project/sector-storage/fr32"
)
func padFFI(buf []byte) []byte {
@ -52,7 +54,7 @@ func TestPadChunkFFI(t *testing.T) {
var buf [128]byte
copy(buf[:], bytes.Repeat([]byte{b}, 127))
Pad(buf[:], buf[:])
fr32.Pad(buf[:], buf[:])
expect := padFFI(bytes.Repeat([]byte{b}, 127))
@ -74,7 +76,7 @@ func TestPadChunkRandEqFFI(t *testing.T) {
var buf [128]byte
Pad(input[:], buf[:])
fr32.Pad(input[:], buf[:])
expect := padFFI(input[:])
@ -88,10 +90,10 @@ func TestRoundtrip(t *testing.T) {
var buf [128]byte
input := bytes.Repeat([]byte{0x01}, 127)
Pad(input, buf[:])
fr32.Pad(input, buf[:])
var out [127]byte
Unpad(buf[:], out[:])
fr32.Unpad(buf[:], out[:])
require.Equal(t, input, out[:])
}
@ -112,10 +114,10 @@ func TestRoundtripChunkRand(t *testing.T) {
var buf [128]byte
copy(buf[:], input[:])
Pad(buf[:], buf[:])
fr32.Pad(buf[:], buf[:])
var out [127]byte
Unpad(buf[:], out[:])
fr32.Unpad(buf[:], out[:])
require.Equal(t, input[:], out[:])
}
@ -129,10 +131,10 @@ func TestRoundtrip16MRand(t *testing.T) {
buf := make([]byte, 16<<20)
Pad(input, buf)
fr32.Pad(input, buf)
out := make([]byte, up)
Unpad(buf, out)
fr32.Unpad(buf, out)
require.Equal(t, input, out)
@ -147,7 +149,7 @@ func BenchmarkPadChunk(b *testing.B) {
b.SetBytes(127)
for i := 0; i < b.N; i++ {
Pad(in, buf[:])
fr32.Pad(in, buf[:])
}
}
@ -159,8 +161,8 @@ func BenchmarkChunkRoundtrip(b *testing.B) {
b.SetBytes(127)
for i := 0; i < b.N; i++ {
Pad(buf[:], buf[:])
Unpad(buf[:], out[:])
fr32.Pad(buf[:], buf[:])
fr32.Unpad(buf[:], out[:])
}
}
@ -168,7 +170,7 @@ func BenchmarkUnpadChunk(b *testing.B) {
var buf [128]byte
copy(buf[:], bytes.Repeat([]byte{0xff}, 127))
Pad(buf[:], buf[:])
fr32.Pad(buf[:], buf[:])
var out [127]byte
b.SetBytes(127)
@ -177,7 +179,7 @@ func BenchmarkUnpadChunk(b *testing.B) {
bs := buf[:]
for i := 0; i < b.N; i++ {
Unpad(bs, out[:])
fr32.Unpad(bs, out[:])
}
}
@ -186,7 +188,7 @@ func BenchmarkUnpad16MChunk(b *testing.B) {
var buf [16 << 20]byte
Pad(bytes.Repeat([]byte{0xff}, int(up)), buf[:])
fr32.Pad(bytes.Repeat([]byte{0xff}, int(up)), buf[:])
var out [16 << 20]byte
b.SetBytes(16 << 20)
@ -194,7 +196,7 @@ func BenchmarkUnpad16MChunk(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
Unpad(buf[:], out[:])
fr32.Unpad(buf[:], out[:])
}
}
@ -210,7 +212,7 @@ func BenchmarkPad16MChunk(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
Pad(in, buf[:])
fr32.Pad(in, buf[:])
}
}
@ -226,7 +228,7 @@ func BenchmarkPad1GChunk(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
Pad(in, buf[:])
fr32.Pad(in, buf[:])
}
}
@ -235,7 +237,7 @@ func BenchmarkUnpad1GChunk(b *testing.B) {
var buf [1 << 30]byte
Pad(bytes.Repeat([]byte{0xff}, int(up)), buf[:])
fr32.Pad(bytes.Repeat([]byte{0xff}, int(up)), buf[:])
var out [1 << 30]byte
b.SetBytes(1 << 30)
@ -243,6 +245,6 @@ func BenchmarkUnpad1GChunk(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
Unpad(buf[:], out[:])
fr32.Unpad(buf[:], out[:])
}
}

12
fr32/readers_test.go
View File

@ -1,4 +1,4 @@
package fr32
package fr32_test
import (
"bytes"
@ -8,6 +8,8 @@ import (
"github.com/stretchr/testify/require"
"github.com/filecoin-project/specs-actors/actors/abi"
"github.com/filecoin-project/sector-storage/fr32"
)
func TestPadReader(t *testing.T) {
@ -15,7 +17,7 @@ func TestPadReader(t *testing.T) {
raw := bytes.Repeat([]byte{0x55}, int(ps))
r, err := NewPadReader(bytes.NewReader(raw), ps)
r, err := fr32.NewPadReader(bytes.NewReader(raw), ps)
if err != nil {
t.Fatal(err)
}
@ -26,7 +28,7 @@ func TestPadReader(t *testing.T) {
}
padOut := make([]byte, ps.Padded())
Pad(raw, padOut)
fr32.Pad(raw, padOut)
require.Equal(t, padOut, readerPadded)
}
@ -37,9 +39,9 @@ func TestUnpadReader(t *testing.T) {
raw := bytes.Repeat([]byte{0x77}, int(ps))
padOut := make([]byte, ps.Padded())
Pad(raw, padOut)
fr32.Pad(raw, padOut)
r, err := NewUnpadReader(bytes.NewReader(padOut), ps.Padded())
r, err := fr32.NewUnpadReader(bytes.NewReader(padOut), ps.Padded())
if err != nil {
t.Fatal(err)
}