lotus/itests/verifreg_test.go

145 lines
4.4 KiB
Go
Raw Normal View History

package itests
2021-05-24 11:41:13 +00:00
import (
"context"
"fmt"
2021-05-24 11:41:13 +00:00
"strings"
"testing"
"time"
2021-05-24 11:41:13 +00:00
"github.com/filecoin-project/go-state-types/abi"
"github.com/filecoin-project/go-state-types/big"
2021-06-01 21:11:54 +00:00
"github.com/filecoin-project/go-state-types/network"
"github.com/filecoin-project/lotus/chain/wallet"
2021-06-18 18:45:29 +00:00
"github.com/filecoin-project/lotus/itests/kit"
verifreg4 "github.com/filecoin-project/specs-actors/v4/actors/builtin/verifreg"
"github.com/stretchr/testify/require"
2021-06-01 21:11:54 +00:00
2021-05-31 20:12:06 +00:00
lapi "github.com/filecoin-project/lotus/api"
2021-05-24 11:41:13 +00:00
"github.com/filecoin-project/lotus/chain/actors"
"github.com/filecoin-project/lotus/chain/actors/builtin/verifreg"
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/lotus/node/impl"
2021-05-24 11:41:13 +00:00
)
func TestVerifiedClientTopUp(t *testing.T) {
blockTime := 100 * time.Millisecond
2021-06-01 19:28:48 +00:00
test := func(nv network.Version, shouldWork bool) func(*testing.T) {
return func(t *testing.T) {
rootKey, err := wallet.GenerateKey(types.KTSecp256k1)
require.NoError(t, err)
verifierKey, err := wallet.GenerateKey(types.KTSecp256k1)
require.NoError(t, err)
verifiedClientKey, err := wallet.GenerateKey(types.KTBLS)
require.NoError(t, err)
bal, err := types.ParseFIL("100fil")
require.NoError(t, err)
2021-06-18 18:45:29 +00:00
node, _, ens := kit.EnsembleMinimal(t, kit.MockProofs(),
kit.RootVerifier(rootKey, abi.NewTokenAmount(bal.Int64())),
kit.Account(verifierKey, abi.NewTokenAmount(bal.Int64())), // assign some balance to the verifier so they can send an AddClient message.
kit.ConstructorOpts(kit.InstantaneousNetworkVersion(nv)))
ens.InterconnectAll().BeginMining(blockTime)
api := node.FullNode.(*impl.FullNodeAPI)
2021-06-01 19:28:48 +00:00
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
// get VRH
2021-06-01 19:28:48 +00:00
vrh, err := api.StateVerifiedRegistryRootKey(ctx, types.TipSetKey{})
fmt.Println(vrh.String())
require.NoError(t, err)
2021-06-01 19:28:48 +00:00
// import the root key.
rootAddr, err := api.WalletImport(ctx, &rootKey.KeyInfo)
require.NoError(t, err)
// import the verifier's key.
verifierAddr, err := api.WalletImport(ctx, &verifierKey.KeyInfo)
require.NoError(t, err)
// import the verified client's key.
verifiedClientAddr, err := api.WalletImport(ctx, &verifiedClientKey.KeyInfo)
require.NoError(t, err)
params, err := actors.SerializeParams(&verifreg4.AddVerifierParams{Address: verifierAddr, Allowance: big.NewInt(100000000000)})
require.NoError(t, err)
2021-06-01 19:28:48 +00:00
msg := &types.Message{
From: rootAddr,
2021-06-01 19:28:48 +00:00
To: verifreg.Address,
Method: verifreg.Methods.AddVerifier,
Params: params,
Value: big.Zero(),
}
sm, err := api.MpoolPushMessage(ctx, msg, nil)
require.NoError(t, err, "AddVerifier failed")
2021-06-01 19:28:48 +00:00
res, err := api.StateWaitMsg(ctx, sm.Cid(), 1, lapi.LookbackNoLimit, true)
require.NoError(t, err)
require.EqualValues(t, 0, res.Receipt.ExitCode)
2021-06-01 19:28:48 +00:00
// assign datacap to a client
2021-06-01 19:28:48 +00:00
datacap := big.NewInt(10000)
params, err = actors.SerializeParams(&verifreg4.AddVerifiedClientParams{Address: verifiedClientAddr, Allowance: datacap})
require.NoError(t, err)
2021-06-01 19:28:48 +00:00
msg = &types.Message{
From: verifierAddr,
2021-06-01 19:28:48 +00:00
To: verifreg.Address,
Method: verifreg.Methods.AddVerifiedClient,
Params: params,
Value: big.Zero(),
}
sm, err = api.MpoolPushMessage(ctx, msg, nil)
require.NoError(t, err)
2021-06-01 19:28:48 +00:00
res, err = api.StateWaitMsg(ctx, sm.Cid(), 1, lapi.LookbackNoLimit, true)
require.NoError(t, err)
require.EqualValues(t, 0, res.Receipt.ExitCode)
// check datacap balance
dcap, err := api.StateVerifiedClientStatus(ctx, verifiedClientAddr, types.EmptyTSK)
require.NoError(t, err)
2021-06-01 19:28:48 +00:00
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: verifiedClientAddr, Allowance: datacap})
2021-06-01 19:28:48 +00:00
if err != nil {
t.Fatal(err)
}
msg = &types.Message{
From: verifierAddr,
2021-06-01 19:28:48 +00:00
To: verifreg.Address,
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)
}
2021-06-01 21:11:54 +00:00
if !shouldWork && (err == nil || !strings.Contains(err.Error(), "verified client already exists")) {
2021-06-01 19:28:48 +00:00
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))
2021-05-24 11:41:13 +00:00
}