135 lines
3.1 KiB
Go
135 lines
3.1 KiB
Go
package sealing_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
commcid "github.com/filecoin-project/go-fil-commcid"
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
|
|
sealing "github.com/filecoin-project/storage-fsm"
|
|
)
|
|
|
|
type fakeChain struct {
|
|
h abi.ChainEpoch
|
|
}
|
|
|
|
func (f *fakeChain) ChainHead(ctx context.Context) (sealing.TipSetToken, abi.ChainEpoch, error) {
|
|
return []byte{1, 2, 3}, f.h, nil
|
|
}
|
|
|
|
func TestBasicPolicyEmptySector(t *testing.T) {
|
|
policy := sealing.NewBasicPreCommitPolicy(&fakeChain{
|
|
h: abi.ChainEpoch(55),
|
|
}, 10)
|
|
|
|
exp, err := policy.Expiration(context.Background())
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 65, int(exp))
|
|
}
|
|
|
|
func TestBasicPolicyMostConstrictiveSchedule(t *testing.T) {
|
|
policy := sealing.NewBasicPreCommitPolicy(&fakeChain{
|
|
h: abi.ChainEpoch(55),
|
|
}, 100)
|
|
|
|
pieces := []sealing.Piece{
|
|
{
|
|
Piece: abi.PieceInfo{
|
|
Size: abi.PaddedPieceSize(1024),
|
|
PieceCID: commcid.ReplicaCommitmentV1ToCID([]byte{1, 2, 3}),
|
|
},
|
|
DealInfo: &sealing.DealInfo{
|
|
DealID: abi.DealID(42),
|
|
DealSchedule: sealing.DealSchedule{
|
|
StartEpoch: abi.ChainEpoch(70),
|
|
EndEpoch: abi.ChainEpoch(75),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Piece: abi.PieceInfo{
|
|
Size: abi.PaddedPieceSize(1024),
|
|
PieceCID: commcid.ReplicaCommitmentV1ToCID([]byte{1, 2, 3}),
|
|
},
|
|
DealInfo: &sealing.DealInfo{
|
|
DealID: abi.DealID(43),
|
|
DealSchedule: sealing.DealSchedule{
|
|
StartEpoch: abi.ChainEpoch(80),
|
|
EndEpoch: abi.ChainEpoch(100),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
exp, err := policy.Expiration(context.Background(), pieces...)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 100, int(exp))
|
|
}
|
|
|
|
func TestBasicPolicyIgnoresExistingScheduleIfExpired(t *testing.T) {
|
|
policy := sealing.NewBasicPreCommitPolicy(&fakeChain{
|
|
h: abi.ChainEpoch(55),
|
|
}, 100)
|
|
|
|
pieces := []sealing.Piece{
|
|
{
|
|
Piece: abi.PieceInfo{
|
|
Size: abi.PaddedPieceSize(1024),
|
|
PieceCID: commcid.ReplicaCommitmentV1ToCID([]byte{1, 2, 3}),
|
|
},
|
|
DealInfo: &sealing.DealInfo{
|
|
DealID: abi.DealID(44),
|
|
DealSchedule: sealing.DealSchedule{
|
|
StartEpoch: abi.ChainEpoch(1),
|
|
EndEpoch: abi.ChainEpoch(10),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
exp, err := policy.Expiration(context.Background(), pieces...)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 155, int(exp))
|
|
}
|
|
|
|
func TestMissingDealIsIgnored(t *testing.T) {
|
|
policy := sealing.NewBasicPreCommitPolicy(&fakeChain{
|
|
h: abi.ChainEpoch(55),
|
|
}, 100)
|
|
|
|
pieces := []sealing.Piece{
|
|
{
|
|
Piece: abi.PieceInfo{
|
|
Size: abi.PaddedPieceSize(1024),
|
|
PieceCID: commcid.ReplicaCommitmentV1ToCID([]byte{1, 2, 3}),
|
|
},
|
|
DealInfo: &sealing.DealInfo{
|
|
DealID: abi.DealID(44),
|
|
DealSchedule: sealing.DealSchedule{
|
|
StartEpoch: abi.ChainEpoch(1),
|
|
EndEpoch: abi.ChainEpoch(10),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Piece: abi.PieceInfo{
|
|
Size: abi.PaddedPieceSize(1024),
|
|
PieceCID: commcid.ReplicaCommitmentV1ToCID([]byte{1, 2, 3}),
|
|
},
|
|
DealInfo: nil,
|
|
},
|
|
}
|
|
|
|
exp, err := policy.Expiration(context.Background(), pieces...)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 155, int(exp))
|
|
}
|