mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2025-01-08 03:51:20 +00:00
bb780a06d3
- Added tests for request_generator in devnet - Added tests for request_generator in rpctests
87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package requests
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/ledgerwatch/erigon/cmd/devnet/models"
|
|
"github.com/ledgerwatch/erigon/common"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func MockRequestGenerator(reqId int) *RequestGenerator {
|
|
return &RequestGenerator{
|
|
reqID: reqId,
|
|
client: nil,
|
|
}
|
|
}
|
|
|
|
func TestRequestGenerator_GetAdminNodeInfo(t *testing.T) {
|
|
testCases := []struct {
|
|
reqId int
|
|
expected string
|
|
}{
|
|
{1, `{"jsonrpc":"2.0","method":"admin_nodeInfo","id":1}`},
|
|
{2, `{"jsonrpc":"2.0","method":"admin_nodeInfo","id":2}`},
|
|
{3, `{"jsonrpc":"2.0","method":"admin_nodeInfo","id":3}`},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
reqGen := MockRequestGenerator(testCase.reqId)
|
|
got := reqGen.GetAdminNodeInfo()
|
|
require.EqualValues(t, testCase.expected, got)
|
|
}
|
|
}
|
|
|
|
func TestRequestGenerator_GetBalance(t *testing.T) {
|
|
testCases := []struct {
|
|
reqId int
|
|
address common.Address
|
|
blockNum string
|
|
expected string
|
|
}{
|
|
{
|
|
1,
|
|
common.HexToAddress("0x67b1d87101671b127f5f8714789c7192f7ad340e"),
|
|
models.BlockNumLatest, // TODO: change to models.Latest once former PR merge happens
|
|
`{"jsonrpc":"2.0","method":"eth_getBalance","params":["0x67b1d87101671b127f5f8714789c7192f7ad340e","latest"],"id":1}`,
|
|
},
|
|
{
|
|
2,
|
|
common.HexToAddress("0x71562b71999873db5b286df957af199ec94617f7"),
|
|
"earliest", // TODO: change to models.Earliest once former PR merge happens
|
|
`{"jsonrpc":"2.0","method":"eth_getBalance","params":["0x71562b71999873db5b286df957af199ec94617f7","earliest"],"id":2}`,
|
|
},
|
|
{
|
|
3,
|
|
common.HexToAddress("0x1b5fd2fed153fa7fac43300273c70c068bfa406a"),
|
|
"pending", // TODO: change to models.Pending once former PR merge happens
|
|
`{"jsonrpc":"2.0","method":"eth_getBalance","params":["0x1b5fd2fed153fa7fac43300273c70c068bfa406a","pending"],"id":3}`,
|
|
},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
reqGen := MockRequestGenerator(testCase.reqId)
|
|
got := reqGen.GetBalance(testCase.address, testCase.blockNum)
|
|
require.EqualValues(t, testCase.expected, got)
|
|
}
|
|
}
|
|
|
|
func TestRequestGenerator_TxpoolContent(t *testing.T) {
|
|
testCases := []struct {
|
|
reqId int
|
|
expected string
|
|
}{
|
|
{1, `{"jsonrpc":"2.0","method":"txpool_content","params":[],"id":1}`},
|
|
{2, `{"jsonrpc":"2.0","method":"txpool_content","params":[],"id":2}`},
|
|
{3, `{"jsonrpc":"2.0","method":"txpool_content","params":[],"id":3}`},
|
|
}
|
|
|
|
for _, testCase := range testCases {
|
|
reqGen := MockRequestGenerator(testCase.reqId)
|
|
got := reqGen.TxpoolContent()
|
|
require.EqualValues(t, testCase.expected, got)
|
|
}
|
|
}
|
|
|
|
// TODO: Add tests for GetTransactionCount
|