41 lines
1.4 KiB
Go
41 lines
1.4 KiB
Go
|
package fake
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"math/big"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
||
|
|
"github.com/ethereum/go-ethereum/rpc"
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewTestServer(ctrl *gomock.Controller) (*rpc.Server, *MockFakePublicTransactionPoolAPI) {
|
||
|
|
srv := rpc.NewServer()
|
||
|
|
svc := NewMockFakePublicTransactionPoolAPI(ctrl)
|
||
|
|
if err := srv.RegisterName("eth", svc); err != nil {
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return srv, svc
|
||
|
|
}
|
||
|
|
|
||
|
|
// CallArgs copied from module go-ethereum/internal/ethapi
|
||
|
|
type CallArgs struct {
|
||
|
|
From common.Address `json:"from"`
|
||
|
|
To *common.Address `json:"to"`
|
||
|
|
Gas hexutil.Big `json:"gas"`
|
||
|
|
GasPrice hexutil.Big `json:"gasPrice"`
|
||
|
|
Value hexutil.Big `json:"value"`
|
||
|
|
Data hexutil.Bytes `json:"data"`
|
||
|
|
}
|
||
|
|
|
||
|
|
// FakePublicTransactionPoolAPI used to generate mock by mockgen util.
|
||
|
|
// This was done because PublicTransactionPoolAPI is located in internal/ethapi module
|
||
|
|
// and there is no easy way to generate mocks from internal modules.
|
||
|
|
type FakePublicTransactionPoolAPI interface {
|
||
|
|
GasPrice(ctx context.Context) (*big.Int, error)
|
||
|
|
EstimateGas(ctx context.Context, args CallArgs) (*hexutil.Big, error)
|
||
|
|
GetTransactionCount(ctx context.Context, address common.Address, blockNr rpc.BlockNumber) (*hexutil.Uint64, error)
|
||
|
|
SendRawTransaction(ctx context.Context, encodedTx hexutil.Bytes) (common.Hash, error)
|
||
|
|
}
|