mirror of
https://github.com/luxfi/proto.git
synced 2026-10-30 15:21:09 +00:00
zeekayandHanzo Dev
920d56f2ec
The platform and exchange VMs are the last chain code living in luxfi/node and the reason the node cannot be retired. This brings over the part of them proto did not already have: the adopt registry, the ZAP wire generators for signer, warp and validator fees, the hybrid and runtime proof-of-possession signers, the native admission check, the zwarp client and server, the X-chain network layer, the xvmroot state root, the batch verifier and owner root, and the credential policy the P and X mempools share, which lands beside the mempool it gates as proto/txs/auth. The X-chain genesis builder moves into proto/x/genesis, where the codec it calls already lives, so it no longer reaches for the xvm package to build primary-asset bytes. Imports point at the packages that already have homes: cache, upgrade, version and util as their own modules, and gas, verify and index in luxfi/vm. The three generated files bring github.com/zap-proto/go with them, which requires go 1.26.5. Most of the remaining difference between the node's trees and proto is not a backlog. proto restructured this code in August and the node's copies predate it, so files that look absent are largely present here under other names. What genuinely cannot come yet is anything reaching for node/server/http, node/vms or node/wallet, and the tests written against tx constructors proto's tx layer no longer exports. Co-authored-by: Hanzo Dev <[email protected]>
361 lines
9.6 KiB
Go
361 lines
9.6 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package network
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/luxfi/metric"
|
|
"github.com/luxfi/mock/gomock"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
validators "github.com/luxfi/validators"
|
|
validatorstest "github.com/luxfi/validators/validatorstest"
|
|
"github.com/luxfi/ids"
|
|
"github.com/luxfi/math/set"
|
|
lux "github.com/luxfi/utxo"
|
|
"github.com/luxfi/proto/x/block/executor/executormock"
|
|
"github.com/luxfi/proto/x/fxs"
|
|
"github.com/luxfi/proto/x/txs"
|
|
xmempool "github.com/luxfi/proto/x/txs/mempool"
|
|
"github.com/luxfi/proto/txs/mempool"
|
|
"github.com/luxfi/utxo/nftfx"
|
|
"github.com/luxfi/utxo/propertyfx"
|
|
"github.com/luxfi/utxo/secp256k1fx"
|
|
"github.com/luxfi/warp"
|
|
)
|
|
|
|
// testSender implements warp.Sender for testing
|
|
type testSender struct {
|
|
SendGossipF func(context.Context, warp.SendConfig, []byte) error
|
|
}
|
|
|
|
var _ warp.Sender = (*testSender)(nil)
|
|
|
|
func (t *testSender) SendRequest(ctx context.Context, nodeIDs set.Set[ids.NodeID], requestID uint32, requestBytes []byte) error {
|
|
return nil
|
|
}
|
|
|
|
func (t *testSender) SendResponse(ctx context.Context, nodeID ids.NodeID, requestID uint32, responseBytes []byte) error {
|
|
return nil
|
|
}
|
|
|
|
func (t *testSender) SendError(ctx context.Context, nodeID ids.NodeID, requestID uint32, errorCode int32, errorMessage string) error {
|
|
return nil
|
|
}
|
|
|
|
func (t *testSender) SendGossip(ctx context.Context, config warp.SendConfig, gossipBytes []byte) error {
|
|
if t.SendGossipF != nil {
|
|
return t.SendGossipF(ctx, config, gossipBytes)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
var (
|
|
testConfig = Config{
|
|
MaxValidatorSetStaleness: time.Second,
|
|
TargetGossipSize: 1,
|
|
PushGossipNumValidators: 1,
|
|
PushGossipNumPeers: 0,
|
|
PushRegossipNumValidators: 1,
|
|
PushRegossipNumPeers: 0,
|
|
PushGossipDiscardedCacheSize: 1,
|
|
PushGossipMaxRegossipFrequency: time.Second,
|
|
PushGossipFrequency: time.Second,
|
|
PullGossipPollSize: 1,
|
|
PullGossipFrequency: time.Second,
|
|
PullGossipThrottlingPeriod: time.Second,
|
|
PullGossipThrottlingLimit: 1,
|
|
ExpectedBloomFilterElements: 10,
|
|
ExpectedBloomFilterFalsePositiveProbability: .1,
|
|
MaxBloomFilterFalsePositiveProbability: .5,
|
|
}
|
|
|
|
errTest = errors.New("test error")
|
|
)
|
|
|
|
func TestNetworkIssueTxFromRPC(t *testing.T) {
|
|
type test struct {
|
|
name string
|
|
mempool mempool.Mempool[*txs.Tx]
|
|
txVerifierFunc func(*gomock.Controller) TxVerifier
|
|
appSenderFunc func(*gomock.Controller) warp.Sender
|
|
tx *txs.Tx
|
|
expectedErr error
|
|
}
|
|
|
|
tests := []test{
|
|
{
|
|
name: "mempool has transaction",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
require.NoError(t, mempool.Add(&txs.Tx{Unsigned: &txs.BaseTx{}}))
|
|
return mempool
|
|
}(),
|
|
tx: &txs.Tx{Unsigned: &txs.BaseTx{}},
|
|
expectedErr: mempool.ErrDuplicateTx,
|
|
},
|
|
{
|
|
name: "transaction marked as dropped in mempool",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
mempool.MarkDropped(ids.Empty, errTest)
|
|
return mempool
|
|
}(),
|
|
tx: &txs.Tx{Unsigned: &txs.BaseTx{}},
|
|
expectedErr: errTest,
|
|
},
|
|
{
|
|
name: "tx too big",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
return mempool
|
|
}(),
|
|
txVerifierFunc: func(ctrl *gomock.Controller) TxVerifier {
|
|
txVerifier := executormock.NewManager(ctrl)
|
|
txVerifier.EXPECT().VerifyTx(gomock.Any()).Return(nil)
|
|
return txVerifier
|
|
},
|
|
tx: func() *txs.Tx {
|
|
tx := &txs.Tx{Unsigned: &txs.BaseTx{}}
|
|
bytes := make([]byte, mempool.MaxTxSize+1)
|
|
tx.SetBytes(bytes, bytes)
|
|
return tx
|
|
}(),
|
|
expectedErr: mempool.ErrTxTooLarge,
|
|
},
|
|
{
|
|
name: "tx conflicts",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
|
|
tx := &txs.Tx{
|
|
Unsigned: &txs.BaseTx{
|
|
BaseTx: lux.BaseTx{
|
|
Ins: []*lux.TransferableInput{
|
|
{
|
|
UTXOID: lux.UTXOID{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
require.NoError(t, mempool.Add(tx))
|
|
return mempool
|
|
}(),
|
|
txVerifierFunc: func(ctrl *gomock.Controller) TxVerifier {
|
|
txVerifier := executormock.NewManager(ctrl)
|
|
txVerifier.EXPECT().VerifyTx(gomock.Any()).Return(nil)
|
|
return txVerifier
|
|
},
|
|
tx: func() *txs.Tx {
|
|
tx := &txs.Tx{
|
|
Unsigned: &txs.BaseTx{
|
|
BaseTx: lux.BaseTx{
|
|
Ins: []*lux.TransferableInput{
|
|
{
|
|
UTXOID: lux.UTXOID{},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
TxID: ids.ID{1},
|
|
}
|
|
return tx
|
|
}(),
|
|
expectedErr: mempool.ErrConflictsWithOtherTx,
|
|
},
|
|
{
|
|
name: "mempool full",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
m, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
|
|
// 64 MiB mempool / 2 MiB max tx = 32 max-size txs
|
|
for i := 0; i < 32; i++ {
|
|
tx := &txs.Tx{Unsigned: &txs.BaseTx{}}
|
|
bytes := make([]byte, mempool.MaxTxSize)
|
|
tx.SetBytes(bytes, bytes)
|
|
tx.TxID = ids.GenerateTestID()
|
|
require.NoError(t, m.Add(tx))
|
|
}
|
|
|
|
return m
|
|
}(),
|
|
txVerifierFunc: func(ctrl *gomock.Controller) TxVerifier {
|
|
txVerifier := executormock.NewManager(ctrl)
|
|
txVerifier.EXPECT().VerifyTx(gomock.Any()).Return(nil)
|
|
return txVerifier
|
|
},
|
|
tx: func() *txs.Tx {
|
|
tx := &txs.Tx{Unsigned: &txs.BaseTx{BaseTx: lux.BaseTx{}}}
|
|
tx.SetBytes([]byte{1, 2, 3}, []byte{1, 2, 3})
|
|
return tx
|
|
}(),
|
|
expectedErr: mempool.ErrMempoolFull,
|
|
},
|
|
{
|
|
name: "happy path",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
return mempool
|
|
}(),
|
|
txVerifierFunc: func(ctrl *gomock.Controller) TxVerifier {
|
|
txVerifier := executormock.NewManager(ctrl)
|
|
txVerifier.EXPECT().VerifyTx(gomock.Any()).Return(nil)
|
|
return txVerifier
|
|
},
|
|
appSenderFunc: func(ctrl *gomock.Controller) warp.Sender {
|
|
appSender := &testSender{}
|
|
appSender.SendGossipF = func(context.Context, warp.SendConfig, []byte) error {
|
|
return nil
|
|
}
|
|
return appSender
|
|
},
|
|
tx: &txs.Tx{Unsigned: &txs.BaseTx{}},
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
ctrl := gomock.NewController(t)
|
|
|
|
parser, err := txs.NewParser(
|
|
[]fxs.Fx{
|
|
&secp256k1fx.Fx{},
|
|
&nftfx.Fx{},
|
|
&propertyfx.Fx{},
|
|
},
|
|
)
|
|
require.NoError(err)
|
|
|
|
txVerifierFunc := func(ctrl *gomock.Controller) TxVerifier {
|
|
return executormock.NewManager(ctrl)
|
|
}
|
|
if tt.txVerifierFunc != nil {
|
|
txVerifierFunc = tt.txVerifierFunc
|
|
}
|
|
|
|
appSenderFunc := func(ctrl *gomock.Controller) warp.Sender {
|
|
return &testSender{}
|
|
}
|
|
if tt.appSenderFunc != nil {
|
|
appSenderFunc = tt.appSenderFunc
|
|
}
|
|
|
|
n, err := New(
|
|
nil,
|
|
ids.EmptyNodeID,
|
|
ids.Empty,
|
|
&validatorstest.State{
|
|
GetCurrentHeightF: func(context.Context) (uint64, error) {
|
|
return 0, nil
|
|
},
|
|
GetValidatorSetF: func(context.Context, uint64, ids.ID) (map[ids.NodeID]*validators.GetValidatorOutput, error) {
|
|
return nil, nil
|
|
},
|
|
},
|
|
parser,
|
|
txVerifierFunc(ctrl),
|
|
tt.mempool,
|
|
appSenderFunc(ctrl),
|
|
metric.NewNoOp().Registry(),
|
|
testConfig,
|
|
)
|
|
require.NoError(err)
|
|
err = n.IssueTxFromRPC(tt.tx)
|
|
require.ErrorIs(err, tt.expectedErr)
|
|
|
|
require.NoError(n.txPushGossiper.Gossip(context.Background()))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNetworkIssueTxFromRPCWithoutVerification(t *testing.T) {
|
|
type test struct {
|
|
name string
|
|
mempool mempool.Mempool[*txs.Tx]
|
|
appSenderFunc func(*gomock.Controller) warp.Sender
|
|
expectedErr error
|
|
}
|
|
|
|
tests := []test{
|
|
{
|
|
name: "happy path",
|
|
mempool: func() mempool.Mempool[*txs.Tx] {
|
|
mempool, err := xmempool.New("", metric.NewRegistry())
|
|
require.NoError(t, err)
|
|
return mempool
|
|
}(),
|
|
appSenderFunc: func(ctrl *gomock.Controller) warp.Sender {
|
|
appSender := &testSender{}
|
|
appSender.SendGossipF = func(context.Context, warp.SendConfig, []byte) error {
|
|
return nil
|
|
}
|
|
return appSender
|
|
},
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
require := require.New(t)
|
|
ctrl := gomock.NewController(t)
|
|
|
|
parser, err := txs.NewParser(
|
|
[]fxs.Fx{
|
|
&secp256k1fx.Fx{},
|
|
&nftfx.Fx{},
|
|
&propertyfx.Fx{},
|
|
},
|
|
)
|
|
require.NoError(err)
|
|
|
|
appSenderFunc := func(ctrl *gomock.Controller) warp.Sender {
|
|
return &testSender{}
|
|
}
|
|
if tt.appSenderFunc != nil {
|
|
appSenderFunc = tt.appSenderFunc
|
|
}
|
|
|
|
n, err := New(
|
|
nil,
|
|
ids.EmptyNodeID,
|
|
ids.Empty,
|
|
&validatorstest.State{
|
|
GetCurrentHeightF: func(context.Context) (uint64, error) {
|
|
return 0, nil
|
|
},
|
|
GetValidatorSetF: func(context.Context, uint64, ids.ID) (map[ids.NodeID]*validators.GetValidatorOutput, error) {
|
|
return nil, nil
|
|
},
|
|
},
|
|
parser,
|
|
executormock.NewManager(ctrl), // Should never verify a tx
|
|
tt.mempool,
|
|
appSenderFunc(ctrl),
|
|
metric.NewNoOp().Registry(),
|
|
testConfig,
|
|
)
|
|
require.NoError(err)
|
|
err = n.IssueTxFromRPCWithoutVerification(&txs.Tx{Unsigned: &txs.BaseTx{}})
|
|
require.ErrorIs(err, tt.expectedErr)
|
|
|
|
require.NoError(n.txPushGossiper.Gossip(context.Background()))
|
|
})
|
|
}
|
|
}
|