102 lines
3.2 KiB
Go
102 lines
3.2 KiB
Go
|
package itests
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/go-fil-markets/storagemarket"
|
||
|
|
"github.com/filecoin-project/go-state-types/abi"
|
||
|
|
"github.com/filecoin-project/lotus/api"
|
||
|
|
"github.com/filecoin-project/lotus/build"
|
||
|
|
"github.com/filecoin-project/lotus/chain/types"
|
||
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestOfflineDealFlow(t *testing.T) {
|
||
|
|
blocktime := 10 * time.Millisecond
|
||
|
|
|
||
|
|
// For these tests where the block time is artificially short, just use
|
||
|
|
// a deal start epoch that is guaranteed to be far enough in the future
|
||
|
|
// so that the deal starts sealing in time
|
||
|
|
startEpoch := abi.ChainEpoch(2 << 12)
|
||
|
|
|
||
|
|
runTest := func(t *testing.T, fastRet bool) {
|
||
|
|
ctx := context.Background()
|
||
|
|
client, miner, ens := kit.EnsembleMinimal(t, kit.MockProofs())
|
||
|
|
ens.InterconnectAll().BeginMining(blocktime)
|
||
|
|
|
||
|
|
dh := kit.NewDealHarness(t, client, miner)
|
||
|
|
|
||
|
|
// Create a random file and import on the client.
|
||
|
|
res, inFile := client.CreateImportFile(ctx, 1, 0)
|
||
|
|
|
||
|
|
// Get the piece size and commP
|
||
|
|
rootCid := res.Root
|
||
|
|
pieceInfo, err := client.ClientDealPieceCID(ctx, rootCid)
|
||
|
|
require.NoError(t, err)
|
||
|
|
t.Log("FILE CID:", rootCid)
|
||
|
|
|
||
|
|
// Create a storage deal with the miner
|
||
|
|
maddr, err := miner.ActorAddress(ctx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
addr, err := client.WalletDefaultAddress(ctx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Manual storage deal (offline deal)
|
||
|
|
dataRef := &storagemarket.DataRef{
|
||
|
|
TransferType: storagemarket.TTManual,
|
||
|
|
Root: rootCid,
|
||
|
|
PieceCid: &pieceInfo.PieceCID,
|
||
|
|
PieceSize: pieceInfo.PieceSize.Unpadded(),
|
||
|
|
}
|
||
|
|
|
||
|
|
proposalCid, err := client.ClientStartDeal(ctx, &api.StartDealParams{
|
||
|
|
Data: dataRef,
|
||
|
|
Wallet: addr,
|
||
|
|
Miner: maddr,
|
||
|
|
EpochPrice: types.NewInt(1000000),
|
||
|
|
DealStartEpoch: startEpoch,
|
||
|
|
MinBlocksDuration: uint64(build.MinDealDuration),
|
||
|
|
FastRetrieval: fastRet,
|
||
|
|
})
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Wait for the deal to reach StorageDealCheckForAcceptance on the client
|
||
|
|
cd, err := client.ClientGetDealInfo(ctx, *proposalCid)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Eventually(t, func() bool {
|
||
|
|
cd, _ := client.ClientGetDealInfo(ctx, *proposalCid)
|
||
|
|
return cd.State == storagemarket.StorageDealCheckForAcceptance
|
||
|
|
}, 30*time.Second, 1*time.Second, "actual deal status is %s", storagemarket.DealStates[cd.State])
|
||
|
|
|
||
|
|
// Create a CAR file from the raw file
|
||
|
|
carFileDir := t.TempDir()
|
||
|
|
carFilePath := filepath.Join(carFileDir, "out.car")
|
||
|
|
err = client.ClientGenCar(ctx, api.FileRef{Path: inFile}, carFilePath)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Import the CAR file on the miner - this is the equivalent to
|
||
|
|
// transferring the file across the wire in a normal (non-offline) deal
|
||
|
|
err = miner.DealsImportData(ctx, *proposalCid, carFilePath)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
// Wait for the deal to be published
|
||
|
|
dh.WaitDealPublished(ctx, proposalCid)
|
||
|
|
|
||
|
|
t.Logf("deal published, retrieving")
|
||
|
|
|
||
|
|
// Retrieve the deal
|
||
|
|
outFile := dh.PerformRetrieval(ctx, proposalCid, rootCid, false)
|
||
|
|
|
||
|
|
kit.AssertFilesEqual(t, inFile, outFile)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
t.Run("stdretrieval", func(t *testing.T) { runTest(t, false) })
|
||
|
|
t.Run("fastretrieval", func(t *testing.T) { runTest(t, true) })
|
||
|
|
}
|