status-go/rpc/verif_proxy_test.go

//go:build nimbus_light_client
// +build nimbus_light_client

package rpc

import (
	"context"
	"fmt"
	"os"
	"os/signal"
	"syscall"
	"testing"
	"time"

	"net/http"
	"net/http/httptest"

	"github.com/ethereum/go-ethereum/common"
	"github.com/ethereum/go-ethereum/common/hexutil"

	"github.com/stretchr/testify/require"
	"github.com/stretchr/testify/suite"

	gethrpc "github.com/ethereum/go-ethereum/rpc"
	"github.com/status-im/status-go/params"
)

type ProxySuite struct {
	suite.Suite
}

func TestProxySuite(t *testing.T) {
	suite.Run(t, new(ProxySuite))
}

func (s *ProxySuite) startRpcClient(infuraURL string) *Client {
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fmt.Fprintln(w, `{
			"id": 1,
			"jsonrpc": "2.0",
			"result": "0x234234e22b9ffc2387e18636e0534534a3d0c56b0243567432453264c16e78a2adc"
		}`)
	}))
	defer ts.Close()

	gethRPCClient, err := gethrpc.Dial(ts.URL)
	require.NoError(s.T(), err)

	db, close := setupTestNetworkDB(s.T())
	defer close()
	c, err := NewClient(gethRPCClient, 1, []params.Network{}, db)
	require.NoError(s.T(), err)

	return c
}

func (s *ProxySuite) TestRun() {
	s.T().Skip("skip test using infura")
	infuraURL := "https://mainnet.infura.io/v3/800c641949d64d768a5070a1b0511938"
	client := s.startRpcClient(infuraURL)

	// Run light client proxy
	ctx, cancel := context.WithCancel(context.Background())

	signals := make(chan os.Signal, 1)
	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM)
	fmt.Println("Before range signals")

	// Invoke eth_getBalance
	var result hexutil.Big
	var addr common.Address
	addr = common.HexToAddress("0x95222290DD7278Aa3Ddd389Cc1E1d165CC4BAfe5")
	chainID := uint64(1)

	time.Sleep(200 * time.Second)
	err := client.CallContext(ctx, &result, chainID, "eth_getBalance", addr, "latest")
	require.NoError(s.T(), err)

	client.UnregisterHandler("eth_getBalance")
	var resultRaw hexutil.Big
	err = client.CallContext(ctx, &resultRaw, chainID, "eth_getBalance", addr, "latest")
	s.Require().Equal(result, resultRaw)
	for range signals {
		fmt.Println("Signal caught, exiting")
		cancel()
	}
	fmt.Println("Exiting")

}