77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
|
package itests
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
commcid "github.com/filecoin-project/go-fil-commcid"
|
||
|
commp "github.com/filecoin-project/go-fil-commp-hashhash"
|
||
|
"github.com/filecoin-project/go-state-types/abi"
|
||
|
"github.com/filecoin-project/lotus/chain/actors/policy"
|
||
|
"github.com/filecoin-project/lotus/itests/kit"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
)
|
||
|
|
||
|
func TestDealPadding(t *testing.T) {
|
||
|
|
||
|
kit.QuietMiningLogs()
|
||
|
|
||
|
var blockTime = 250 * time.Millisecond
|
||
|
startEpoch := abi.ChainEpoch(2 << 12)
|
||
|
policy.SetPreCommitChallengeDelay(2)
|
||
|
|
||
|
client, miner, ens := kit.EnsembleMinimal(t, kit.ThroughRPC(), kit.WithAllSubsystems()) // no mock proofs.
|
||
|
ens.InterconnectAll().BeginMining(blockTime)
|
||
|
dh := kit.NewDealHarness(t, client, miner, miner)
|
||
|
|
||
|
ctx := context.Background()
|
||
|
client.WaitTillChain(ctx, kit.BlockMinedBy(miner.ActorAddr))
|
||
|
|
||
|
// Create a random file, would originally be a 256-byte sector
|
||
|
res, inFile := client.CreateImportFile(ctx, 1, 200)
|
||
|
|
||
|
// Get the piece size and commP
|
||
|
pieceInfo, err := client.ClientDealPieceCID(ctx, res.Root)
|
||
|
require.NoError(t, err)
|
||
|
t.Log("FILE CID:", res.Root)
|
||
|
|
||
|
runTest := func(t *testing.T, upscale abi.PaddedPieceSize) {
|
||
|
// test whether padding works as intended
|
||
|
newRawCp, err := commp.PadCommP(
|
||
|
pieceInfo.PieceCID.Hash()[len(pieceInfo.PieceCID.Hash())-32:],
|
||
|
uint64(pieceInfo.PieceSize),
|
||
|
uint64(upscale),
|
||
|
)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
pcid, err := commcid.DataCommitmentV1ToCID(newRawCp)
|
||
|
require.NoError(t, err)
|
||
|
|
||
|
dp := dh.DefaultStartDealParams()
|
||
|
dp.Data.Root = res.Root
|
||
|
dp.Data.PieceCid = &pcid
|
||
|
dp.Data.PieceSize = upscale.Unpadded()
|
||
|
dp.DealStartEpoch = startEpoch
|
||
|
proposalCid := dh.StartDeal(ctx, dp)
|
||
|
|
||
|
// TODO: this sleep is only necessary because deals don't immediately get logged in the dealstore, we should fix this
|
||
|
time.Sleep(time.Second)
|
||
|
|
||
|
di, err := client.ClientGetDealInfo(ctx, *proposalCid)
|
||
|
require.NoError(t, err)
|
||
|
require.True(t, di.PieceCID.Equals(pcid))
|
||
|
|
||
|
dh.WaitDealSealed(ctx, proposalCid, false, false, nil)
|
||
|
|
||
|
// Retrieve the deal
|
||
|
outFile := dh.PerformRetrieval(ctx, proposalCid, res.Root, false)
|
||
|
|
||
|
kit.AssertFilesEqual(t, inFile, outFile)
|
||
|
}
|
||
|
|
||
|
t.Run("padQuarterSector", func(t *testing.T) { runTest(t, 512) })
|
||
|
t.Run("padHalfSector", func(t *testing.T) { runTest(t, 1024) })
|
||
|
t.Run("padFullSector", func(t *testing.T) { runTest(t, 2048) })
|
||
|
}
|