44 lines
1.5 KiB
Go
44 lines
1.5 KiB
Go
package fake
|
|
|
|
//go:generate mockgen -package=fake -source=txservice.go -destination=mock.go
|
|
|
|
import (
|
|
"context"
|
|
|
|
"go.uber.org/mock/gomock"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/common/hexutil"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
)
|
|
|
|
// NewTestServer returns a mocked test server
|
|
func NewTestServer(ctrl *gomock.Controller) (*rpc.Server, *MockPublicTransactionPoolAPI) {
|
|
srv := rpc.NewServer()
|
|
svc := NewMockPublicTransactionPoolAPI(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.Uint64 `json:"gas"`
|
|
GasPrice hexutil.Big `json:"gasPrice"`
|
|
Value hexutil.Big `json:"value"`
|
|
Data hexutil.Bytes `json:"data"`
|
|
}
|
|
|
|
// PublicTransactionPoolAPI 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 PublicTransactionPoolAPI interface {
|
|
GasPrice(ctx context.Context) (*hexutil.Big, error)
|
|
EstimateGas(ctx context.Context, args CallArgs) (hexutil.Uint64, error)
|
|
GetTransactionCount(ctx context.Context, address common.Address, blockNr rpc.BlockNumber) (*hexutil.Uint64, error)
|
|
SendRawTransaction(ctx context.Context, encodedTx hexutil.Bytes) (common.Hash, error)
|
|
}
|