359 lines
8.8 KiB
Go
359 lines
8.8 KiB
Go
|
package query
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/hex"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
gossipv1 "github.com/certusone/wormhole/node/pkg/proto/gossip/v1"
|
||
|
|
"github.com/wormhole-foundation/wormhole/sdk/vaa"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/accounts/abi"
|
||
|
|
)
|
||
|
|
|
||
|
|
func createQueryRequestForTesting(t *testing.T, chainId vaa.ChainID) *QueryRequest {
|
||
|
|
t.Helper()
|
||
|
|
// Create a query request.
|
||
|
|
wethAbi, err := abi.JSON(strings.NewReader("[{\"constant\":true,\"inputs\":[],\"name\":\"name\",\"outputs\":[{\"name\":\"\",\"type\":\"string\"}],\"payable\":false,\"stateMutability\":\"view\",\"type\":\"function\"},{\"constant\":true,\"inputs\":[],\"name\":\"totalSupply\",\"outputs\":[{\"name\":\"\",\"type\":\"uint256\"}],\"payable\":false,\"stateMutability\":\"view\",\"type\":\"function\"}]"))
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
data1, err := wethAbi.Pack("name")
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
data2, err := wethAbi.Pack("totalSupply")
|
||
|
|
if err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
to, _ := hex.DecodeString("0d500b1d8e8ef31e21c99d1db9a6444d3adf1270")
|
||
|
|
block := "0x28d9630"
|
||
|
|
callData := []*EthCallData{
|
||
|
|
{
|
||
|
|
To: to,
|
||
|
|
Data: data1,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
To: to,
|
||
|
|
Data: data2,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
callRequest := &EthCallQueryRequest{
|
||
|
|
BlockId: block,
|
||
|
|
CallData: callData,
|
||
|
|
}
|
||
|
|
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: chainId,
|
||
|
|
Query: callRequest,
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
|
||
|
|
return queryRequest
|
||
|
|
}
|
||
|
|
|
||
|
|
// A timestamp has nanos, but we only marshal down to micros, so trim our time to micros for testing purposes.
|
||
|
|
func timeForTest(t *testing.T, ts time.Time) time.Time {
|
||
|
|
t.Helper()
|
||
|
|
return time.UnixMicro(ts.UnixMicro())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQueryRequestMarshalUnmarshal(t *testing.T) {
|
||
|
|
queryRequest := createQueryRequestForTesting(t, vaa.ChainIDPolygon)
|
||
|
|
queryRequestBytes, err := queryRequest.Marshal()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
var queryRequest2 QueryRequest
|
||
|
|
err = queryRequest2.Unmarshal(queryRequestBytes)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
assert.True(t, queryRequest.Equal(&queryRequest2))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithNoPerChainQueriesShouldFail(t *testing.T) {
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{
|
||
|
|
{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
// Leave Query nil.
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithTooManyPerChainQueriesShouldFail(t *testing.T) {
|
||
|
|
perChainQueries := []*PerChainQueryRequest{}
|
||
|
|
for count := 0; count < 300; count++ {
|
||
|
|
callData := []*EthCallData{{
|
||
|
|
|
||
|
|
To: []byte(fmt.Sprintf("%-20s", fmt.Sprintf("To for %d", count))),
|
||
|
|
Data: []byte(fmt.Sprintf("CallData for %d", count)),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
perChainQueries = append(perChainQueries, &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: callData,
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: perChainQueries,
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestForInvalidChainIdShouldFail(t *testing.T) {
|
||
|
|
queryRequest := createQueryRequestForTesting(t, vaa.ChainIDUnset)
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithInvalidBlockIdShouldFail(t *testing.T) {
|
||
|
|
callData := []*EthCallData{{
|
||
|
|
To: []byte(fmt.Sprintf("%-20s", fmt.Sprintf("To for %d", 0))),
|
||
|
|
Data: []byte(fmt.Sprintf("CallData for %d", 0)),
|
||
|
|
}}
|
||
|
|
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "latest",
|
||
|
|
CallData: callData,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithNoCallDataEntriesShouldFail(t *testing.T) {
|
||
|
|
callData := []*EthCallData{}
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: callData,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithNilCallDataEntriesShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: nil,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfQueryRequestWithTooManyCallDataEntriesShouldFail(t *testing.T) {
|
||
|
|
callData := []*EthCallData{}
|
||
|
|
for count := 0; count < 300; count++ {
|
||
|
|
callData = append(callData, &EthCallData{
|
||
|
|
To: []byte(fmt.Sprintf("%-20s", fmt.Sprintf("To for %d", count))),
|
||
|
|
Data: []byte(fmt.Sprintf("CallData for %d", count)),
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: callData,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithNilToShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: nil,
|
||
|
|
Data: []byte("This can't be zero length"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithEmptyToShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: []byte{},
|
||
|
|
Data: []byte("This can't be zero length"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithWrongLengthToShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: []byte("TooShort"),
|
||
|
|
Data: []byte("This can't be zero length"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithNilDataShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: []byte(fmt.Sprintf("%-20s", fmt.Sprintf("To for %d", 0))),
|
||
|
|
Data: nil,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithEmptyDataShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: []byte(fmt.Sprintf("%-20s", fmt.Sprintf("To for %d", 0))),
|
||
|
|
Data: []byte{},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMarshalOfEthCallQueryWithWrongToLengthShouldFail(t *testing.T) {
|
||
|
|
perChainQuery := &PerChainQueryRequest{
|
||
|
|
ChainId: vaa.ChainIDPolygon,
|
||
|
|
Query: &EthCallQueryRequest{
|
||
|
|
BlockId: "0x28d9630",
|
||
|
|
CallData: []*EthCallData{
|
||
|
|
{
|
||
|
|
To: []byte("This is too short!"),
|
||
|
|
Data: []byte("This can't be zero length"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
queryRequest := &QueryRequest{
|
||
|
|
Nonce: 1,
|
||
|
|
PerChainQueries: []*PerChainQueryRequest{perChainQuery},
|
||
|
|
}
|
||
|
|
_, err := queryRequest.Marshal()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestPostSignedQueryRequestShouldFailIfNoOneIsListening(t *testing.T) {
|
||
|
|
queryRequest := createQueryRequestForTesting(t, vaa.ChainIDPolygon)
|
||
|
|
queryRequestBytes, err := queryRequest.Marshal()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
sig := [65]byte{}
|
||
|
|
signedQueryRequest := &gossipv1.SignedQueryRequest{
|
||
|
|
QueryRequest: queryRequestBytes,
|
||
|
|
Signature: sig[:],
|
||
|
|
}
|
||
|
|
|
||
|
|
var signedQueryReqSendC chan<- *gossipv1.SignedQueryRequest
|
||
|
|
assert.Error(t, PostSignedQueryRequest(signedQueryReqSendC, signedQueryRequest))
|
||
|
|
}
|