Skip to content

Commit

Permalink
other(tests): remove unused utils
Browse files Browse the repository at this point in the history
  • Loading branch information
@ptrus
ptrus committed Feb 23, 2022
1 parent e2769ab commit 38a83ca
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 270 deletions.
5 changes: 1 addition & 4 deletions tests/rpc/net_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,14 +20,11 @@ func TestMain(m *testing.M) {
// Run tests.
code := m.Run()

if err := Shutdown(); err != nil {
log.Printf("%v", err)
}
os.Exit(code)
}

func TestNet_Version(t *testing.T) {
rpcRes := Call(t, "net_version", []string{})
rpcRes := call(t, "net_version", []string{})
var res string
err := json.Unmarshal(rpcRes.Result, &res)

Expand Down
266 changes: 0 additions & 266 deletions tests/rpc/utils.go
View file
Original file line number Diff line number Diff line change
@@ -1,19 +1,14 @@
package rpc

import (
"bytes"
"context"
"encoding/hex"
"encoding/json"
"flag"
"fmt"
"math/big"
"net/http"
"testing"
"time"

cmnEth "github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/common/hexutil"
"github.com/ethereum/go-ethereum/ethclient"
"github.com/oasisprotocol/oasis-core/go/common"
cmnGrpc "github.com/oasisprotocol/oasis-core/go/common/grpc"
Expand All @@ -27,11 +22,9 @@ import (
oasisTesting "github.com/oasisprotocol/oasis-sdk/client-sdk/go/testing"
"github.com/oasisprotocol/oasis-sdk/client-sdk/go/types"
"github.com/stretchr/testify/require"
"github.com/uptrace/bun"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"

"github.com/oasisprotocol/emerald-web3-gateway/db/migrations"
"github.com/oasisprotocol/emerald-web3-gateway/filters"
"github.com/oasisprotocol/emerald-web3-gateway/indexer"
"github.com/oasisprotocol/emerald-web3-gateway/log"
Expand Down Expand Up @@ -274,262 +267,3 @@ func InitialDeposit(rc client.RuntimeClient, amount quantity.Quantity, to types.

return nil
}

// Shutdown stops web3 gateway.
func Shutdown() error {
if err := migrations.DropTables(context.Background(), db.DB.(*bun.DB)); err != nil {
return fmt.Errorf("db cleanup failed: %w", err)
}

return w3.Close()
}

func GetAddress() ([]byte, error) {
rpcRes, err := CallWithError("eth_accounts", []string{})
if err != nil {
return nil, err
}

var res []hexutil.Bytes
err = json.Unmarshal(rpcRes.Result, &res)
if err != nil {
return nil, err
}

return res[0], nil
}

func CreateRequest(method string, params interface{}) Request {
return Request{
Version: "2.0",
Method: method,
Params: params,
ID: 1,
}
}

func Call(t *testing.T, method string, params interface{}) *Response {
req, err := json.Marshal(CreateRequest(method, params))
require.NoError(t, err)

var rpcRes *Response
time.Sleep(1 * time.Second)

url, err := w3.GetHTTPEndpoint()
require.NoError(t, err)
httpReq, err := http.NewRequestWithContext(context.Background(), "POST", url, bytes.NewBuffer(req))
if err != nil {
require.NoError(t, err)
}

httpReq.Header.Set("Content-Type", "application/json")
client := &http.Client{}
res, err := client.Do(httpReq)
if err != nil {
require.NoError(t, fmt.Errorf("could not perform request: %w", err))
}

decoder := json.NewDecoder(res.Body)
rpcRes = new(Response)
err = decoder.Decode(&rpcRes)
require.NoError(t, err)

err = res.Body.Close()
require.NoError(t, err)
require.Nil(t, rpcRes.Error)

return rpcRes
}

func CallWithError(method string, params interface{}) (*Response, error) {
req, err := json.Marshal(CreateRequest(method, params))
if err != nil {
return nil, err
}

var rpcRes *Response
time.Sleep(1 * time.Second)

url, err := w3.GetHTTPEndpoint()
if err != nil {
return nil, err
}
httpReq, err := http.NewRequestWithContext(context.Background(), "POST", url, bytes.NewBuffer(req))
if err != nil {
return nil, err
}

httpReq.Header.Set("Content-Type", "application/json")
client := &http.Client{}
res, err := client.Do(httpReq)
if err != nil {
return nil, fmt.Errorf("could not perform request: %w", err)
}

decoder := json.NewDecoder(res.Body)
rpcRes = new(Response)
err = decoder.Decode(&rpcRes)
if err != nil {
return nil, err
}

err = res.Body.Close()
if err != nil {
return nil, err
}

if rpcRes.Error != nil {
return nil, fmt.Errorf(rpcRes.Error.Message)
}

return rpcRes, nil
}

// HexToBigInt turns a 0x prefixed hex string to a big.Int.
func HexToBigInt(t *testing.T, in string) *big.Int {
s := in[2:]
b, err := hex.DecodeString(s)
require.NoError(t, err)
return big.NewInt(0).SetBytes(b)
}

// sendTestTransaction sends a dummy transaction.
func SendTestTransaction(t *testing.T, addr []byte) hexutil.Bytes {
param := make([]map[string]string, 1)
param[0] = make(map[string]string)
param[0]["from"] = "0x" + fmt.Sprintf("%x", addr)
param[0]["to"] = "0x1122334455667788990011223344556677889900"
param[0]["value"] = "0x1"

rpcRes := Call(t, "personal_unlockAccount", []interface{}{param[0]["from"], ""})
require.Nil(t, rpcRes.Error)

rpcRes = Call(t, "eth_sendTransaction", param)
require.Nil(t, rpcRes.Error)

var hash hexutil.Bytes
err := json.Unmarshal(rpcRes.Result, &hash)
require.NoError(t, err)
return hash
}

// deployTestContract deploys a contract that emits an event in the constructor.
func DeployTestContract(t *testing.T, addr []byte) (hexutil.Bytes, map[string]interface{}) {
param := make([]map[string]string, 1)
param[0] = make(map[string]string)
param[0]["from"] = "0x" + fmt.Sprintf("%x", addr)
param[0]["data"] = "0x6080604052348015600f57600080fd5b5060117f775a94827b8fd9b519d36cd827093c664f93347070a554f65e4a6f56cd73889860405160405180910390a2603580604b6000396000f3fe6080604052600080fdfea165627a7a723058206cab665f0f557620554bb45adf266708d2bd349b8a4314bdff205ee8440e3c240029"
param[0]["gas"] = "0x200000"

rpcRes := Call(t, "personal_unlockAccount", []interface{}{param[0]["from"], ""})
require.Nil(t, rpcRes.Error)

rpcRes = Call(t, "eth_sendTransaction", param)
require.Nil(t, rpcRes.Error)

var hash hexutil.Bytes
err := json.Unmarshal(rpcRes.Result, &hash)
require.NoError(t, err)

receipt := WaitForReceipt(t, hash)
require.NotNil(t, receipt, "transaction failed")
require.Equal(t, "0x1", receipt["status"].(string))

return hash, receipt
}

func DeployTestContractWithFunction(t *testing.T, addr []byte) hexutil.Bytes {
// pragma solidity ^0.5.1;

// contract Test {
// event Hello(uint256 indexed world);
// event TestEvent(uint256 indexed a, uint256 indexed b);

// uint256 myStorage;

// constructor() public {
// emit Hello(17);
// }

// function test(uint256 a, uint256 b) public {
// myStorage = a;
// emit TestEvent(a, b);
// }
// }

bytecode := "0x608060405234801561001057600080fd5b5060117f775a94827b8fd9b519d36cd827093c664f93347070a554f65e4a6f56cd73889860405160405180910390a260d08061004d6000396000f3fe6080604052348015600f57600080fd5b506004361060285760003560e01c8063eb8ac92114602d575b600080fd5b606060048036036040811015604157600080fd5b8101908080359060200190929190803590602001909291905050506062565b005b8160008190555080827ff3ca124a697ba07e8c5e80bebcfcc48991fc16a63170e8a9206e30508960d00360405160405180910390a3505056fea265627a7a723158201d94d2187aaf3a6790527b615fcc40970febf0385fa6d72a2344848ebd0df3e964736f6c63430005110032"

param := make([]map[string]string, 1)
param[0] = make(map[string]string)
param[0]["from"] = "0x" + fmt.Sprintf("%x", addr)
param[0]["data"] = bytecode
param[0]["gas"] = "0x200000"

rpcRes := Call(t, "personal_unlockAccount", []interface{}{param[0]["from"], ""})
require.Nil(t, rpcRes.Error)

rpcRes = Call(t, "eth_sendTransaction", param)

var hash hexutil.Bytes
err := json.Unmarshal(rpcRes.Result, &hash)
require.NoError(t, err)

receipt := WaitForReceipt(t, hash)
require.NotNil(t, receipt, "transaction failed")
require.Equal(t, "0x1", receipt["status"].(string))

return hash
}

func GetTransactionReceipt(t *testing.T, hash hexutil.Bytes) map[string]interface{} {
param := []string{hash.String()}
rpcRes := Call(t, "eth_getTransactionReceipt", param)

receipt := make(map[string]interface{})
err := json.Unmarshal(rpcRes.Result, &receipt)
require.NoError(t, err)

return receipt
}

func WaitForReceipt(t *testing.T, hash hexutil.Bytes) map[string]interface{} {
for i := 0; i < 12; i++ {
receipt := GetTransactionReceipt(t, hash)
if receipt != nil {
return receipt
}

time.Sleep(time.Second)
}

return nil
}

func GetNonce(t *testing.T, block string) hexutil.Uint64 {
from, err := GetAddress()
require.NoError(t, err)

param := []interface{}{hexutil.Bytes(from), block}
rpcRes := Call(t, "eth_getTransactionCount", param)

var nonce hexutil.Uint64
err = json.Unmarshal(rpcRes.Result, &nonce)
require.NoError(t, err)
return nonce
}

func UnlockAllAccounts(t *testing.T) {
var accts []cmnEth.Address
rpcRes := Call(t, "eth_accounts", []map[string]string{})
err := json.Unmarshal(rpcRes.Result, &accts)
require.NoError(t, err)

for _, acct := range accts {
t.Logf("account: %v", acct)
rpcRes = Call(t, "personal_unlockAccount", []interface{}{acct, ""})
var unlocked bool
err = json.Unmarshal(rpcRes.Result, &unlocked)
require.NoError(t, err)
require.True(t, unlocked)
}
}

0 comments on commit 38a83ca

Please sign in to comment.