140 lines
3.6 KiB
Go
140 lines
3.6 KiB
Go
package test
|
|
|
|
import (
|
|
"context"
|
|
"github.com/filecoin-project/go-state-types/network"
|
|
"strings"
|
|
|
|
lapi "github.com/filecoin-project/lotus/api"
|
|
|
|
"github.com/filecoin-project/lotus/chain/actors"
|
|
"github.com/filecoin-project/lotus/chain/actors/builtin/verifreg"
|
|
"github.com/filecoin-project/lotus/node/impl"
|
|
verifreg4 "github.com/filecoin-project/specs-actors/v4/actors/builtin/verifreg"
|
|
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/filecoin-project/go-state-types/big"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
)
|
|
|
|
func AddVerifiedClient(t *testing.T, b APIBuilder) {
|
|
test := func(nv network.Version, shouldWork bool) func(*testing.T) {
|
|
return func(t *testing.T) {
|
|
|
|
nodes, miners := b(t, []FullNodeOpts{FullNodeWithActorsUpgradeAt(nv, -1)}, OneMiner)
|
|
api := nodes[0].FullNode.(*impl.FullNodeAPI)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
//Get VRH
|
|
vrh, err := api.StateVerifiedRegistryRootKey(ctx, types.TipSetKey{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
//Add verifier
|
|
verifier, err := api.WalletDefaultAddress(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
params, err := actors.SerializeParams(&verifreg4.AddVerifierParams{Address: verifier, Allowance: big.NewInt(100000000000)})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
msg := &types.Message{
|
|
To: verifreg.Address,
|
|
From: vrh,
|
|
Method: verifreg.Methods.AddVerifier,
|
|
Params: params,
|
|
Value: big.Zero(),
|
|
}
|
|
|
|
bm := NewBlockMiner(ctx, t, miners[0], 100*time.Millisecond)
|
|
bm.MineBlocks()
|
|
defer bm.Stop()
|
|
|
|
sm, err := api.MpoolPushMessage(ctx, msg, nil)
|
|
if err != nil {
|
|
t.Fatal("AddVerifier failed: ", err)
|
|
}
|
|
res, err := api.StateWaitMsg(ctx, sm.Cid(), 1, lapi.LookbackNoLimit, true)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Receipt.ExitCode != 0 {
|
|
t.Fatal("did not successfully send message")
|
|
}
|
|
|
|
//Assign datacap to a client
|
|
datacap := big.NewInt(10000)
|
|
clientAddress, err := api.WalletNew(ctx, types.KTBLS)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
params, err = actors.SerializeParams(&verifreg4.AddVerifiedClientParams{Address: clientAddress, Allowance: datacap})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
msg = &types.Message{
|
|
To: verifreg.Address,
|
|
From: verifier,
|
|
Method: verifreg.Methods.AddVerifiedClient,
|
|
Params: params,
|
|
Value: big.Zero(),
|
|
}
|
|
|
|
sm, err = api.MpoolPushMessage(ctx, msg, nil)
|
|
if err != nil {
|
|
t.Fatal("AddVerifiedClient faield: ", err)
|
|
}
|
|
res, err = api.StateWaitMsg(ctx, sm.Cid(), 1, lapi.LookbackNoLimit, true)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Receipt.ExitCode != 0 {
|
|
t.Fatal("did not successfully send message")
|
|
}
|
|
|
|
//check datacap balance
|
|
dcap, err := api.StateVerifiedClientStatus(ctx, clientAddress, types.EmptyTSK)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !dcap.Equals(datacap) {
|
|
t.Fatal("")
|
|
}
|
|
|
|
//try to assign datacap to the same client should fail for actor v4 and below
|
|
params, err = actors.SerializeParams(&verifreg4.AddVerifiedClientParams{Address: clientAddress, Allowance: datacap})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
msg = &types.Message{
|
|
To: verifreg.Address,
|
|
From: verifier,
|
|
Method: verifreg.Methods.AddVerifiedClient,
|
|
Params: params,
|
|
Value: big.Zero(),
|
|
}
|
|
|
|
_, err = api.MpoolPushMessage(ctx, msg, nil)
|
|
if shouldWork && err != nil {
|
|
t.Fatal("expected nil err", err)
|
|
}
|
|
|
|
if !shouldWork && err == nil || !strings.Contains(err.Error(), "verified client already exists") {
|
|
t.Fatal("Add datacap to an existing verified client should fail")
|
|
}
|
|
}
|
|
}
|
|
|
|
t.Run("nv12", test(network.Version12, false))
|
|
t.Run("nv13", test(network.Version13, true))
|
|
}
|