Gateway: add net_version support and test (#1934)

go/enclave/components/batch_executor.go

@@ -379,7 +379,7 @@ func (executor *batchExecutor) populateOutboundCrossChainData(ctx context.Contex
 
 		encodedTree, err := json.Marshal(xchainTree)
 		if err != nil {
-			panic(err) //todo: figure out what to do
+			panic(err) // todo: figure out what to do
 		}
 
 		batch.Header.CrossChainTree = encodedTree

go/enclave/crosschain/common.go

@@ -234,7 +234,7 @@ func (ms MessageStructs) HashPacked(index int) gethcommon.Hash {
 		},
 	}
 
-	//todo @siliev: err
+	// todo @siliev: err
 	packed, _ := args.Pack(messageStruct.Sender, messageStruct.Sequence, messageStruct.Nonce, messageStruct.Topic, messageStruct.Payload, messageStruct.ConsistencyLevel)
 	hash := crypto.Keccak256Hash(packed)
 	return hash

go/enclave/genesis/testnet_genesis.go

@@ -9,8 +9,10 @@ import (
 )
 
 const TestnetPrefundedPK = "8dfb8083da6275ae3e4f41e3e8a8c19d028d32c9247e24530933782f2a05035b" // The genesis main account private key.
-var GasBridgingKeys, _ = crypto.GenerateKey()                                                 // todo - make static
-var GasWithdrawalKeys, _ = crypto.GenerateKey()                                               // todo - make static
+var (
+	GasBridgingKeys, _   = crypto.GenerateKey() // todo - make static
+	GasWithdrawalKeys, _ = crypto.GenerateKey() // todo - make static
+)
 
 var TestnetGenesis = Genesis{
 	Accounts: []Account{

go/enclave/nodetype/common.go

@@ -40,6 +40,6 @@ func ExportCrossChainData(ctx context.Context, storage storage.Storage, fromSeqN
 		L1BlockHash:          block.Hash(),
 		L1BlockNum:           big.NewInt(0).Set(block.Header().Number),
 		CrossChainRootHashes: crossChainHashes,
-	} //todo: check fromSeqNo
+	} // todo: check fromSeqNo
 	return bundle, nil
 }

go/enclave/nodetype/sequencer.go

@@ -468,6 +468,7 @@ func (s *sequencer) signCrossChainBundle(bundle *common.ExtCrossChainBundle) err
 	}
 	return nil
 }
+
 func (s *sequencer) OnL1Block(ctx context.Context, block *types.Block, result *components.BlockIngestionType) error {
 	// nothing to do
 	return nil

integration/networktest/tests/gateway/gateway_test.go

@@ -1,12 +1,15 @@
 package gateway
 
 import (
+	"context"
+	"fmt"
 	"math/big"
 	"testing"
 
 	"github.com/ten-protocol/go-ten/integration/networktest"
 	"github.com/ten-protocol/go-ten/integration/networktest/actions"
 	"github.com/ten-protocol/go-ten/integration/networktest/env"
+	"github.com/ten-protocol/go-ten/integration/networktest/userwallet"
 	"github.com/ten-protocol/go-ten/integration/simulation/devnetwork"
 )
 
@@ -37,6 +40,34 @@ func TestGatewayHappyPath(t *testing.T) {
 
 			&actions.VerifyBalanceAfterTest{UserID: 1, ExpectedBalance: _transferAmount},
 			&actions.VerifyBalanceDiffAfterTest{UserID: 0, Snapshot: actions.SnapAfterAllocation, ExpectedDiff: big.NewInt(0).Neg(_transferAmount)},
+
+			// test net_version works through the gateway
+			actions.VerifyOnlyAction(func(ctx context.Context, network networktest.NetworkConnector) error {
+				user, err := actions.FetchTestUser(ctx, 0)
+				if err != nil {
+					return err
+				}
+				// verify user is a gateway user
+				gwUser, ok := user.(*userwallet.GatewayUser)
+				if !ok {
+					return fmt.Errorf("user is not a gateway user")
+				}
+				ethClient := gwUser.Client()
+				rpcClient := ethClient.Client()
+				// check net_version response
+				var result string
+				err = rpcClient.CallContext(ctx, &result, "net_version")
+				if err != nil {
+					return fmt.Errorf("failed to get net_version: %w", err)
+				}
+				fmt.Println("net_version response:", result)
+				expectedResult := "443"
+				if result != expectedResult {
+					return fmt.Errorf("expected net_version to be %s but got %s", expectedResult, result)
+				}
+
+				return nil
+			}),
 		),
 	)
 }

integration/networktest/userwallet/gateway.go

@@ -114,6 +114,10 @@ func (g *GatewayUser) Wallet() wallet.Wallet {
 	return g.wal
 }
 
+func (g *GatewayUser) Client() *ethclient.Client {
+	return g.client
+}
+
 func (g *GatewayUser) WSClient() (*ethclient.Client, error) {
 	if g.wsClient == nil {
 		var err error

tools/walletextension/rpcapi/net_api.go

@@ -0,0 +1,17 @@
+package rpcapi
+
+import (
+	"context"
+)
+
+type NetAPI struct {
+	we *Services
+}
+
+func NewNetAPI(we *Services) *NetAPI {
+	return &NetAPI{we}
+}
+
+func (api *NetAPI) Version(ctx context.Context) (*string, error) {
+	return UnauthenticatedTenRPCCall[string](ctx, api.we, &CacheCfg{CacheType: LongLiving}, "net_version")
+}

tools/walletextension/walletextension_container.go

@@ -83,6 +83,9 @@ func NewContainerFromConfig(config wecommon.Config, logger gethlog.Logger) *Cont
 		}, {
 			Namespace: "eth",
 			Service:   rpcapi.NewFilterAPI(walletExt),
+		}, {
+			Namespace: "net",
+			Service:   rpcapi.NewNetAPI(walletExt),
 		},
 	})