package dagstore import ( "context" "io" "testing" "github.com/stretchr/testify/require" "github.com/filecoin-project/dagstore" "github.com/filecoin-project/dagstore/mount" "github.com/filecoin-project/go-state-types/abi" "github.com/filecoin-project/go-fil-markets/retrievalmarket" "github.com/filecoin-project/go-fil-markets/retrievalmarket/impl/testnodes" tut "github.com/filecoin-project/go-fil-markets/shared_testutil" "github.com/filecoin-project/go-fil-markets/storagemarket" "github.com/filecoin-project/lotus/node/config" "github.com/filecoin-project/specs-actors/v2/actors/builtin/market" ) func TestShardRegistration(t *testing.T) { ps := tut.NewTestPieceStore() sa := testnodes.NewTestSectorAccessor() ctx := context.Background() cids := tut.GenerateCids(4) pieceCidUnsealed := cids[0] pieceCidSealed := cids[1] pieceCidUnsealed2 := cids[2] pieceCidUnsealed3 := cids[3] sealedSector := abi.SectorNumber(1) unsealedSector1 := abi.SectorNumber(2) unsealedSector2 := abi.SectorNumber(3) unsealedSector3 := abi.SectorNumber(4) // ps.ExpectPiece(pieceCidUnsealed, piecestore.PieceInfo{ // PieceCID: pieceCidUnsealed, // Deals: []piecestore.DealInfo{ // { // SectorID: unsealedSector1, // }, // }, // }) // // ps.ExpectPiece(pieceCidSealed, piecestore.PieceInfo{ // PieceCID: pieceCidSealed, // Deals: []piecestore.DealInfo{ // { // SectorID: sealedSector, // }, // }, // }) deals := []storagemarket.MinerDeal{{ // Should be registered State: storagemarket.StorageDealSealing, SectorNumber: unsealedSector1, ClientDealProposal: market.ClientDealProposal{ Proposal: market.DealProposal{ PieceCID: pieceCidUnsealed, }, }, }, { // Should be registered with lazy registration (because sector is sealed) State: storagemarket.StorageDealSealing, SectorNumber: sealedSector, ClientDealProposal: market.ClientDealProposal{ Proposal: market.DealProposal{ PieceCID: pieceCidSealed, }, }, }, { // Should be ignored because deal is no longer active State: storagemarket.StorageDealError, SectorNumber: unsealedSector2, ClientDealProposal: market.ClientDealProposal{ Proposal: market.DealProposal{ PieceCID: pieceCidUnsealed2, }, }, }, { // Should be ignored because deal is not yet sealing State: storagemarket.StorageDealFundsReserved, SectorNumber: unsealedSector3, ClientDealProposal: market.ClientDealProposal{ Proposal: market.DealProposal{ PieceCID: pieceCidUnsealed3, }, }, }} cfg := config.DefaultStorageMiner().DAGStore cfg.RootDir = t.TempDir() mapi := NewMinerAPI(ps, &wrappedSA{sa}, 10) dagst, w, err := NewDAGStore(cfg, mapi) require.NoError(t, err) require.NotNil(t, dagst) require.NotNil(t, w) err = dagst.Start(context.Background()) require.NoError(t, err) migrated, err := w.MigrateDeals(ctx, deals) require.True(t, migrated) require.NoError(t, err) info := dagst.AllShardsInfo() require.Len(t, info, 2) for _, i := range info { require.Equal(t, dagstore.ShardStateNew, i.ShardState) } // Run register shard migration again migrated, err = w.MigrateDeals(ctx, deals) require.False(t, migrated) require.NoError(t, err) // ps.VerifyExpectations(t) } type wrappedSA struct { retrievalmarket.SectorAccessor } func (w *wrappedSA) UnsealSectorAt(ctx context.Context, sectorID abi.SectorNumber, pieceOffset abi.UnpaddedPieceSize, length abi.UnpaddedPieceSize) (mount.Reader, error) { r, err := w.UnsealSector(ctx, sectorID, pieceOffset, length) if err != nil { return nil, err } return struct { io.ReadCloser io.Seeker io.ReaderAt }{ ReadCloser: r, Seeker: nil, ReaderAt: nil, }, err } var _ SectorAccessor = &wrappedSA{}