80 lines
1.8 KiB
Go
80 lines
1.8 KiB
Go
//stm: #unit
|
|
package messagepool
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ipfs/go-datastore"
|
|
|
|
builtin2 "github.com/filecoin-project/specs-actors/v2/actors/builtin"
|
|
|
|
"github.com/filecoin-project/lotus/chain/consensus/filcns"
|
|
"github.com/filecoin-project/lotus/chain/messagepool/gasguess"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/chain/wallet"
|
|
)
|
|
|
|
func TestRepubMessages(t *testing.T) {
|
|
//stm: @TOKEN_WALLET_NEW_001
|
|
oldRepublishBatchDelay := RepublishBatchDelay
|
|
RepublishBatchDelay = time.Microsecond
|
|
defer func() {
|
|
RepublishBatchDelay = oldRepublishBatchDelay
|
|
}()
|
|
|
|
tma := newTestMpoolAPI()
|
|
ds := datastore.NewMapDatastore()
|
|
|
|
mp, err := New(context.Background(), tma, ds, filcns.DefaultUpgradeSchedule(), "mptest", nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// the actors
|
|
w1, err := wallet.NewWallet(wallet.NewMemKeyStore())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
a1, err := w1.WalletNew(context.Background(), types.KTSecp256k1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
w2, err := wallet.NewWallet(wallet.NewMemKeyStore())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
a2, err := w2.WalletNew(context.Background(), types.KTSecp256k1)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
gasLimit := gasguess.Costs[gasguess.CostKey{Code: builtin2.StorageMarketActorCodeID, M: 2}]
|
|
|
|
tma.setBalance(a1, 1) // in FIL
|
|
|
|
for i := 0; i < 10; i++ {
|
|
m := makeTestMessage(w1, a1, a2, uint64(i), gasLimit, uint64(i+1))
|
|
//stm: @CHAIN_MEMPOOL_PUSH_001
|
|
_, err := mp.Push(context.TODO(), m)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
if tma.published != 10 {
|
|
t.Fatalf("expected to have published 10 messages, but got %d instead", tma.published)
|
|
}
|
|
|
|
mp.repubTrigger <- struct{}{}
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
if tma.published != 20 {
|
|
t.Fatalf("expected to have published 20 messages, but got %d instead", tma.published)
|
|
}
|
|
}
|