125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
package itests
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/filecoin-project/go-state-types/big"
|
|
"github.com/filecoin-project/go-state-types/exitcode"
|
|
|
|
lapi "github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/index"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/itests/kit"
|
|
"github.com/filecoin-project/lotus/node"
|
|
)
|
|
|
|
func init() {
|
|
// adjust those to make tests snappy
|
|
index.CoalesceMinDelay = time.Millisecond
|
|
index.CoalesceMaxDelay = 10 * time.Millisecond
|
|
index.CoalesceMergeInterval = time.Millisecond
|
|
}
|
|
|
|
func testMsgIndex(
|
|
t *testing.T,
|
|
name string,
|
|
run func(t *testing.T, makeMsgIndex func(cs *store.ChainStore) (index.MsgIndex, error)),
|
|
check func(t *testing.T, i int, msgIndex index.MsgIndex),
|
|
) {
|
|
|
|
// create the message indices in the test context
|
|
var mx sync.Mutex
|
|
var tmpDirs []string
|
|
var msgIndices []index.MsgIndex
|
|
|
|
t.Cleanup(func() {
|
|
for _, msgIndex := range msgIndices {
|
|
_ = msgIndex.Close()
|
|
}
|
|
|
|
for _, tmp := range tmpDirs {
|
|
_ = os.RemoveAll(tmp)
|
|
}
|
|
})
|
|
|
|
makeMsgIndex := func(cs *store.ChainStore) (index.MsgIndex, error) {
|
|
var err error
|
|
tmp := t.TempDir()
|
|
msgIndex, err := index.NewMsgIndex(context.Background(), tmp, cs)
|
|
if err == nil {
|
|
mx.Lock()
|
|
tmpDirs = append(tmpDirs, tmp)
|
|
msgIndices = append(msgIndices, msgIndex)
|
|
mx.Unlock()
|
|
}
|
|
return msgIndex, err
|
|
}
|
|
|
|
t.Run(name, func(t *testing.T) {
|
|
run(t, makeMsgIndex)
|
|
})
|
|
|
|
if len(msgIndices) == 0 {
|
|
t.Fatal("no message indices")
|
|
}
|
|
|
|
for i, msgIndex := range msgIndices {
|
|
check(t, i, msgIndex)
|
|
}
|
|
}
|
|
|
|
func checkNonEmptyMsgIndex(t *testing.T, _ int, msgIndex index.MsgIndex) {
|
|
mi, ok := msgIndex.(interface{ CountMessages() (int64, error) })
|
|
if !ok {
|
|
t.Fatal("index does not allow counting")
|
|
}
|
|
count, err := mi.CountMessages()
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, count, 0)
|
|
}
|
|
|
|
func TestMsgIndex(t *testing.T) {
|
|
testMsgIndex(t, "testSearchMsg", testSearchMsgWithIndex, checkNonEmptyMsgIndex)
|
|
}
|
|
|
|
func testSearchMsgWithIndex(t *testing.T, makeMsgIndex func(cs *store.ChainStore) (index.MsgIndex, error)) {
|
|
// copy of apiSuite.testSearchMsgWith; needs to be copied or else CI is angry, tests are built individually there
|
|
ctx := context.Background()
|
|
|
|
full, _, ens := kit.EnsembleMinimal(t, kit.MockProofs(), kit.ConstructorOpts(node.Override(new(index.MsgIndex), makeMsgIndex)))
|
|
|
|
senderAddr, err := full.WalletDefaultAddress(ctx)
|
|
require.NoError(t, err)
|
|
|
|
msg := &types.Message{
|
|
From: senderAddr,
|
|
To: senderAddr,
|
|
Value: big.Zero(),
|
|
}
|
|
|
|
ens.BeginMining(100 * time.Millisecond)
|
|
|
|
sm, err := full.MpoolPushMessage(ctx, msg, nil)
|
|
require.NoError(t, err)
|
|
|
|
//stm: @CHAIN_STATE_WAIT_MSG_001
|
|
res, err := full.StateWaitMsg(ctx, sm.Cid(), 1, lapi.LookbackNoLimit, true)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, exitcode.Ok, res.Receipt.ExitCode, "message not successful")
|
|
|
|
//stm: @CHAIN_STATE_SEARCH_MSG_001
|
|
searchRes, err := full.StateSearchMsg(ctx, types.EmptyTSK, sm.Cid(), lapi.LookbackNoLimit, true)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, searchRes)
|
|
|
|
require.Equalf(t, res.TipSet, searchRes.TipSet, "search ts: %s, different from wait ts: %s", searchRes.TipSet, res.TipSet)
|
|
}
|