test for messages in tipset

itests/get_messages_in_ts_test.go

@@ -0,0 +1,104 @@
+package itests
+
+import (
+	"context"
+	"fmt"
+	"github.com/filecoin-project/lotus/api"
+	"github.com/filecoin-project/lotus/chain/store"
+	"github.com/filecoin-project/lotus/chain/types"
+	"github.com/filecoin-project/lotus/itests/kit"
+	"github.com/stretchr/testify/require"
+	"testing"
+
+	"github.com/filecoin-project/go-state-types/big"
+	"time"
+)
+
+func TestChainGetMessagesInTs(t *testing.T) {
+	ctx := context.Background()
+
+	kit.QuietMiningLogs()
+
+	client, _, ens := kit.EnsembleMinimal(t, kit.MockProofs())
+	ens.InterconnectAll().BeginMining(10 * time.Millisecond)
+
+	// create a new address where to send funds.
+	addr, err := client.WalletNew(ctx, types.KTBLS)
+	require.NoError(t, err)
+
+	// get the existing balance from the default wallet to then split it.
+	bal, err := client.WalletBalance(ctx, client.DefaultKey.Address)
+	require.NoError(t, err)
+
+	const iterations = 100
+
+	// we'll send half our balance (saving the other half for gas),
+	// in `iterations` increments.
+	toSend := big.Div(bal, big.NewInt(2))
+	each := big.Div(toSend, big.NewInt(iterations))
+
+	waitAllCh := make(chan struct{})
+	go func() {
+		headChangeCh , err := client.ChainNotify(ctx)
+		require.NoError(t, err)
+		<- headChangeCh  //skip hccurrent
+
+		count := 0
+		for {
+			select {
+			case headChanges := <- headChangeCh:
+				for _, change := range headChanges {
+					if change.Type == store.HCApply {
+						msgs, err := client.ChainGetMessagesInTipset(ctx, change.Val.Key())
+						require.NoError(t, err)
+						count += len(msgs)
+						if count == iterations {
+							waitAllCh <- struct{}{}
+						}
+						fmt.Println(msgs)
+					}
+				}
+			}
+		}
+	}()
+
+	var sms []*types.SignedMessage
+	for i := 0; i < iterations; i++ {
+		msg := &types.Message{
+			From:  client.DefaultKey.Address,
+			To:    addr,
+			Value: each,
+		}
+
+		sm, err := client.MpoolPushMessage(ctx, msg, nil)
+		require.NoError(t, err)
+		require.EqualValues(t, i, sm.Message.Nonce)
+
+		sms = append(sms, sm)
+	}
+
+	select {
+		case <- waitAllCh:
+		case <- time.After(time.Minute):
+			t.Errorf("timeout to wait for pack messages")
+	}
+
+	for _, sm := range sms {
+		msgLookup, err := client.StateWaitMsg(ctx, sm.Cid(), 3, api.LookbackNoLimit, true)
+		require.NoError(t, err)
+
+		ts, err := client.ChainGetTipSet(ctx, msgLookup.TipSet)
+		require.NoError(t, err)
+
+		msgs, err := client.ChainGetMessagesInTipset(ctx, ts.Parents())
+		require.NoError(t, err)
+
+		var found bool
+		for _, msg := range msgs {
+			if msg.Cid == sm.Cid() {
+				found = true
+			}
+		}
+		require.EqualValues(t, true, found, "expect got message in tipset %v", msgLookup.TipSet)
+	}
+}