lotus/lib/zerocomm/zerocomm_test.go
2020-02-26 14:54:34 -08:00

64 lines
1.3 KiB
Go

package zerocomm
import (
"bytes"
"fmt"
"testing"
"github.com/filecoin-project/go-sectorbuilder"
abi "github.com/filecoin-project/specs-actors/actors/abi"
"github.com/ipfs/go-cid"
)
func TestComms(t *testing.T) {
var expPieceComms [levels - skip]cid.Cid
{
l2, err := sectorbuilder.GeneratePieceCIDFromFile(abi.RegisteredProof_StackedDRG2KiBPoSt, bytes.NewReader(make([]byte, 127)), 127)
if err != nil {
return
}
expPieceComms[0] = l2
}
for i := 1; i < levels-2; i++ {
var err error
sectorbuilder.GenerateUnsealedCID()
expPieceComms[i], err = sectorbuilder.GenerateDataCommitment(128<<i, []sectorbuilder.PublicPieceInfo{
{
Size: 127 << (i - 1),
CommP: expPieceComms[i-1],
},
{
Size: 127 << (i - 1),
CommP: expPieceComms[i-1],
},
})
if err != nil {
panic(err)
}
}
for i, comm := range expPieceComms {
if string(comm[:]) != string(pieceComms[i][:]) {
t.Errorf("zero commitment %d didn't match", i)
}
}
for _, comm := range expPieceComms { // Could do codegen, but this is good enough
fmt.Printf("%#v,\n", comm)
}
}
func TestForSise(t *testing.T) {
exp, err := sectorbuilder.GeneratePieceCommitment(bytes.NewReader(make([]byte, 1016)), 1016)
if err != nil {
return
}
actual := ForSize(1016)
if string(exp[:]) != string(actual[:]) {
t.Errorf("zero commitment didn't match")
}
}