Skip to content

Commit

Permalink
removed evm prefix for transmitchecker
Browse files Browse the repository at this point in the history
  • Loading branch information
simsonraj committed Jun 20, 2023
1 parent 0a0a311 commit 491a8d7
Show file tree
Hide file tree
Showing 18 changed files with 44 additions and 44 deletions.
10 changes: 5 additions & 5 deletions core/chains/evm/txmgr/broadcaster_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -159,7 +159,7 @@ func TestEthBroadcaster_ProcessUnstartedEthTxs_Success(t *testing.T) {
encodedPayload := []byte{1, 2, 3}
value := big.Int(assets.NewEthValue(142))
gasLimit := uint32(242)
checker := txmgr.EvmTransmitCheckerSpec{
checker := txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeSimulate,
}

Expand Down Expand Up @@ -402,7 +402,7 @@ func TestEthBroadcaster_ProcessUnstartedEthTxs_Success(t *testing.T) {
Value: big.Int(assets.NewEthValue(442)),
FeeLimit: gasLimit,
Strategy: txmgrcommon.NewSendEveryStrategy(),
Checker: txmgr.EvmTransmitCheckerSpec{
Checker: txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeSimulate,
},
}
Expand Down Expand Up @@ -503,7 +503,7 @@ func TestEthBroadcaster_TransmitChecking(t *testing.T) {
eb, err := NewTestEthBroadcaster(t, txStore, ethClient, ethKeyStore, evmcfg, checkerFactory, false)
require.NoError(t, err)

checker := txmgr.EvmTransmitCheckerSpec{
checker := txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeSimulate,
}
t.Run("when transmit checking times out, sends tx as normal", func(t *testing.T) {
Expand Down Expand Up @@ -1870,7 +1870,7 @@ func TestEthBroadcaster_SyncNonce(t *testing.T) {

}

func checkerToJson(t *testing.T, checker txmgr.EvmTransmitCheckerSpec) *datatypes.JSON {
func checkerToJson(t *testing.T, checker txmgr.TransmitCheckerSpec) *datatypes.JSON {
b, err := json.Marshal(checker)
require.NoError(t, err)
j := datatypes.JSON(b)
Expand All @@ -1881,7 +1881,7 @@ type testCheckerFactory struct {
err error
}

func (t *testCheckerFactory) BuildChecker(spec txmgr.EvmTransmitCheckerSpec) (txmgr.EvmTransmitChecker, error) {
func (t *testCheckerFactory) BuildChecker(spec txmgr.TransmitCheckerSpec) (txmgr.TransmitChecker, error) {
return &testChecker{t.err}, nil
}

Expand Down
14 changes: 7 additions & 7 deletions core/chains/evm/txmgr/transmitchecker.go
Original file line number Diff line number Diff line change
Expand Up @@ -25,18 +25,18 @@ import (
)

type (
EvmTransmitChecker = txmgr.TransmitChecker[*big.Int, common.Address, common.Hash, common.Hash, evmtypes.Nonce, gas.EvmFee]
EvmTransmitCheckerSpec = txmgrtypes.TransmitCheckerSpec[common.Address]
TransmitChecker = txmgr.TransmitChecker[*big.Int, common.Address, common.Hash, common.Hash, evmtypes.Nonce, gas.EvmFee]
TransmitCheckerSpec = txmgrtypes.TransmitCheckerSpec[common.Address]
)

var (
// NoChecker is a TransmitChecker that always determines a transaction should be submitted.
NoChecker EvmTransmitChecker = noChecker{}
NoChecker TransmitChecker = noChecker{}

_ TransmitCheckerFactory = &CheckerFactory{}
_ EvmTransmitChecker = &SimulateChecker{}
_ EvmTransmitChecker = &VRFV1Checker{}
_ EvmTransmitChecker = &VRFV2Checker{}
_ TransmitChecker = &SimulateChecker{}
_ TransmitChecker = &VRFV1Checker{}
_ TransmitChecker = &VRFV2Checker{}
)

// CheckerFactory is a real implementation of TransmitCheckerFactory.
Expand All @@ -45,7 +45,7 @@ type CheckerFactory struct {
}

// BuildChecker satisfies the TransmitCheckerFactory interface.
func (c *CheckerFactory) BuildChecker(spec EvmTransmitCheckerSpec) (EvmTransmitChecker, error) {
func (c *CheckerFactory) BuildChecker(spec TransmitCheckerSpec) (TransmitChecker, error) {
switch spec.CheckerType {
case TransmitCheckerTypeSimulate:
return &SimulateChecker{c.Client}, nil
Expand Down
12 changes: 6 additions & 6 deletions core/chains/evm/txmgr/transmitchecker_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,13 +37,13 @@ func TestFactory(t *testing.T) {
factory := &txmgr.CheckerFactory{Client: client}

t.Run("no checker", func(t *testing.T) {
c, err := factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{})
c, err := factory.BuildChecker(txmgr.TransmitCheckerSpec{})
require.NoError(t, err)
require.Equal(t, txmgr.NoChecker, c)
})

t.Run("vrf v1 checker", func(t *testing.T) {
c, err := factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{
c, err := factory.BuildChecker(txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV1,
VRFCoordinatorAddress: testutils.NewAddressPtr(),
})
Expand All @@ -52,7 +52,7 @@ func TestFactory(t *testing.T) {
})

t.Run("vrf v2 checker", func(t *testing.T) {
c, err := factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{
c, err := factory.BuildChecker(txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV2,
VRFCoordinatorAddress: testutils.NewAddressPtr(),
VRFRequestBlockNumber: big.NewInt(1),
Expand All @@ -61,7 +61,7 @@ func TestFactory(t *testing.T) {
require.IsType(t, &txmgr.VRFV2Checker{}, c)

// request block number not provided should error out.
c, err = factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{
c, err = factory.BuildChecker(txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV2,
VRFCoordinatorAddress: testutils.NewAddressPtr(),
})
Expand All @@ -70,15 +70,15 @@ func TestFactory(t *testing.T) {
})

t.Run("simulate checker", func(t *testing.T) {
c, err := factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{
c, err := factory.BuildChecker(txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeSimulate,
})
require.NoError(t, err)
require.Equal(t, &txmgr.SimulateChecker{Client: client}, c)
})

t.Run("invalid checker type", func(t *testing.T) {
_, err := factory.BuildChecker(txmgr.EvmTransmitCheckerSpec{
_, err := factory.BuildChecker(txmgr.TransmitCheckerSpec{
CheckerType: "invalid",
})
require.EqualError(t, err, "unrecognized checker type: invalid")
Expand Down
8 changes: 4 additions & 4 deletions core/chains/evm/txmgr/txmgr_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -220,7 +220,7 @@ func TestTxm_CreateTransaction(t *testing.T) {
t.Run("simulate transmit checker", func(t *testing.T) {
pgtest.MustExec(t, db, `DELETE FROM eth_txes`)

checker := txmgr.EvmTransmitCheckerSpec{
checker := txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeSimulate,
}
evmConfig.maxQueued = uint64(1)
Expand All @@ -237,7 +237,7 @@ func TestTxm_CreateTransaction(t *testing.T) {
var dbEtx txmgr.DbEthTx
require.NoError(t, db.Get(&dbEtx, `SELECT * FROM eth_txes ORDER BY id ASC LIMIT 1`))

var c txmgr.EvmTransmitCheckerSpec
var c txmgr.TransmitCheckerSpec
require.NotNil(t, etx.TransmitChecker)
require.NoError(t, json.Unmarshal(*etx.TransmitChecker, &c))
require.Equal(t, checker, c)
Expand All @@ -258,7 +258,7 @@ func TestTxm_CreateTransaction(t *testing.T) {
SubID: &testDefaultSubID,
}
evmConfig.maxQueued = uint64(1)
checker := txmgr.EvmTransmitCheckerSpec{
checker := txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV2,
VRFCoordinatorAddress: testutils.NewAddressPtr(),
}
Expand All @@ -280,7 +280,7 @@ func TestTxm_CreateTransaction(t *testing.T) {
require.NoError(t, err)
require.Equal(t, meta, m)

var c txmgr.EvmTransmitCheckerSpec
var c txmgr.TransmitCheckerSpec
require.NotNil(t, etx.TransmitChecker)
require.NoError(t, json.Unmarshal(*etx.TransmitChecker, &c))
require.Equal(t, checker, c)
Expand Down
2 changes: 1 addition & 1 deletion core/internal/cltest/factories.go
Original file line number Diff line number Diff line change
Expand Up @@ -337,7 +337,7 @@ func EvmTxRequestWithStrategy(strategy txmgrtypes.TxStrategy) func(*txmgr.TxRequ
}
}

func EvmTxRequestWithChecker(checker txmgr.EvmTransmitCheckerSpec) func(*txmgr.TxRequest) {
func EvmTxRequestWithChecker(checker txmgr.TransmitCheckerSpec) func(*txmgr.TxRequest) {
return func(tx *txmgr.TxRequest) {
tx.Checker = checker
}
Expand Down
2 changes: 1 addition & 1 deletion core/services/fluxmonitorv2/delegate.go
Original file line number Diff line number Diff line change
Expand Up @@ -69,7 +69,7 @@ func (d *Delegate) ServicesForSpec(jb job.Job) (services []job.ServiceCtx, err e
}
cfg := chain.Config()
strategy := txmgrcommon.NewQueueingTxStrategy(jb.ExternalJobID, cfg.FluxMonitor().DefaultTransactionQueueDepth(), cfg.Database().DefaultQueryTimeout())
var checker txmgr.EvmTransmitCheckerSpec
var checker txmgr.TransmitCheckerSpec
if chain.Config().FluxMonitor().SimulateTransactions() {
checker.CheckerType = txmgr.TransmitCheckerTypeSimulate
}
Expand Down
2 changes: 1 addition & 1 deletion core/services/fluxmonitorv2/flux_monitor_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ var (
type answerSet struct{ latestAnswer, polledAnswer int64 }

func newORM(t *testing.T, db *sqlx.DB, cfg pg.QConfig, txm txmgr.TxManager) fluxmonitorv2.ORM {
return fluxmonitorv2.NewORM(db, logger.TestLogger(t), cfg, txm, txmgrcommon.NewSendEveryStrategy(), txmgr.EvmTransmitCheckerSpec{})
return fluxmonitorv2.NewORM(db, logger.TestLogger(t), cfg, txm, txmgrcommon.NewSendEveryStrategy(), txmgr.TransmitCheckerSpec{})
}

var (
Expand Down
4 changes: 2 additions & 2 deletions core/services/fluxmonitorv2/orm.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,12 +34,12 @@ type orm struct {
q pg.Q
txm transmitter
strategy types.TxStrategy
checker txmgr.EvmTransmitCheckerSpec
checker txmgr.TransmitCheckerSpec
logger logger.Logger
}

// NewORM initializes a new ORM
func NewORM(db *sqlx.DB, lggr logger.Logger, cfg pg.QConfig, txm transmitter, strategy types.TxStrategy, checker txmgr.EvmTransmitCheckerSpec) ORM {
func NewORM(db *sqlx.DB, lggr logger.Logger, cfg pg.QConfig, txm transmitter, strategy types.TxStrategy, checker txmgr.TransmitCheckerSpec) ORM {
namedLogger := lggr.Named("FluxMonitorORM")
q := pg.NewQ(db, namedLogger, cfg)
return &orm{
Expand Down
2 changes: 1 addition & 1 deletion core/services/fluxmonitorv2/orm_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -175,7 +175,7 @@ func TestORM_CreateEthTransaction(t *testing.T) {

var (
txm = txmmocks.NewMockEvmTxManager(t)
orm = fluxmonitorv2.NewORM(db, logger.TestLogger(t), cfg, txm, strategy, txmgr.EvmTransmitCheckerSpec{})
orm = fluxmonitorv2.NewORM(db, logger.TestLogger(t), cfg, txm, strategy, txmgr.TransmitCheckerSpec{})

_, from = cltest.MustInsertRandomKey(t, ethKeyStore, 0)
to = testutils.NewAddress()
Expand Down
2 changes: 1 addition & 1 deletion core/services/ocr/delegate.go
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ func (d *Delegate) ServicesForSpec(jb job.Job) (services []job.ServiceCtx, err e
cfg := chain.Config()
strategy := txmgrcommon.NewQueueingTxStrategy(jb.ExternalJobID, cfg.OCR().DefaultTransactionQueueDepth(), cfg.Database().DefaultQueryTimeout())

var checker txmgr.EvmTransmitCheckerSpec
var checker txmgr.TransmitCheckerSpec
if chain.Config().OCR().SimulateTransactions() {
checker.CheckerType = txmgr.TransmitCheckerTypeSimulate
}
Expand Down
4 changes: 2 additions & 2 deletions core/services/ocrcommon/transmitter.go
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ type transmitter struct {
gasLimit uint32
effectiveTransmitterAddress common.Address
strategy types.TxStrategy
checker txmgr.EvmTransmitCheckerSpec
checker txmgr.TransmitCheckerSpec
chainID *big.Int
keystore roundRobinKeystore
}
Expand All @@ -43,7 +43,7 @@ func NewTransmitter(
gasLimit uint32,
effectiveTransmitterAddress common.Address,
strategy types.TxStrategy,
checker txmgr.EvmTransmitCheckerSpec,
checker txmgr.TransmitCheckerSpec,
chainID *big.Int,
keystore roundRobinKeystore,
) (Transmitter, error) {
Expand Down
4 changes: 2 additions & 2 deletions core/services/ocrcommon/transmitter_pipeline.go
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ type pipelineTransmitter struct {
gasLimit uint32
effectiveTransmitterAddress common.Address
strategy types.TxStrategy
checker txmgr.EvmTransmitCheckerSpec
checker txmgr.TransmitCheckerSpec
pr pipeline.Runner
spec job.Job
chainID string
Expand All @@ -46,7 +46,7 @@ func NewPipelineTransmitter(
gasLimit uint32,
effectiveTransmitterAddress common.Address,
strategy types.TxStrategy,
checker txmgr.EvmTransmitCheckerSpec,
checker txmgr.TransmitCheckerSpec,
pr pipeline.Runner,
spec job.Job,
chainID string,
Expand Down
2 changes: 1 addition & 1 deletion core/services/ocrcommon/transmitter_pipeline_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ func Test_PipelineTransmitter_CreateEthTransaction(t *testing.T) {
toAddress := testutils.NewAddress()
payload := []byte{1, 2, 3}
strategy := newMockTxStrategy(t)
checker := txmgr.EvmTransmitCheckerSpec{CheckerType: txmgr.TransmitCheckerTypeSimulate}
checker := txmgr.TransmitCheckerSpec{CheckerType: txmgr.TransmitCheckerTypeSimulate}
runner := pipelinemocks.NewRunner(t)

transmitter := ocrcommon.NewPipelineTransmitter(
Expand Down
8 changes: 4 additions & 4 deletions core/services/ocrcommon/transmitter_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ func Test_DefaultTransmitter_CreateEthTransaction(t *testing.T) {
gasLimit,
effectiveTransmitterAddress,
strategy,
txmgr.EvmTransmitCheckerSpec{},
txmgr.TransmitCheckerSpec{},
chainID,
ethKeyStore,
)
Expand Down Expand Up @@ -87,7 +87,7 @@ func Test_DefaultTransmitter_Forwarding_Enabled_CreateEthTransaction(t *testing.
gasLimit,
effectiveTransmitterAddress,
strategy,
txmgr.EvmTransmitCheckerSpec{},
txmgr.TransmitCheckerSpec{},
chainID,
ethKeyStore,
)
Expand Down Expand Up @@ -138,7 +138,7 @@ func Test_DefaultTransmitter_Forwarding_Enabled_CreateEthTransaction_Round_Robin
gasLimit,
effectiveTransmitterAddress,
strategy,
txmgr.EvmTransmitCheckerSpec{},
txmgr.TransmitCheckerSpec{},
chainID,
ethKeyStore,
)
Expand Down Expand Up @@ -168,7 +168,7 @@ func Test_DefaultTransmitter_Forwarding_Enabled_CreateEthTransaction_No_Keystore
gasLimit,
effectiveTransmitterAddress,
strategy,
txmgr.EvmTransmitCheckerSpec{},
txmgr.TransmitCheckerSpec{},
chainID,
nil,
)
Expand Down
4 changes: 2 additions & 2 deletions core/services/pipeline/task.eth_tx.go
Original file line number Diff line number Diff line change
Expand Up @@ -199,8 +199,8 @@ func decodeMeta(metaMap MapParam) (*txmgr.TxMeta, error) {
return &txMeta, nil
}

func decodeTransmitChecker(checkerMap MapParam) (txmgr.EvmTransmitCheckerSpec, error) {
var transmitChecker txmgr.EvmTransmitCheckerSpec
func decodeTransmitChecker(checkerMap MapParam) (txmgr.TransmitCheckerSpec, error) {
var transmitChecker txmgr.TransmitCheckerSpec
checkerDecoder, err := mapstructure.NewDecoder(&mapstructure.DecoderConfig{
Result: &transmitChecker,
ErrorUnused: true,
Expand Down
2 changes: 1 addition & 1 deletion core/services/pipeline/task.eth_tx_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ func TestETHTxTask(t *testing.T) {
FeeLimit: gasLimit,
Meta: txMeta,
Strategy: txmgrcommon.NewSendEveryStrategy(),
Checker: txmgr.EvmTransmitCheckerSpec{
Checker: txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV2,
VRFCoordinatorAddress: &addr,
},
Expand Down
4 changes: 2 additions & 2 deletions core/services/relay/evm/evm.go
Original file line number Diff line number Diff line change
Expand Up @@ -337,7 +337,7 @@ func newContractTransmitter(lggr logger.Logger, rargs relaytypes.RelayArgs, tran
scoped := configWatcher.chain.Config()
strategy := txmgrcommon.NewQueueingTxStrategy(rargs.ExternalJobID, scoped.OCR2().DefaultTransactionQueueDepth(), scoped.Database().DefaultQueryTimeout())

var checker txm.EvmTransmitCheckerSpec
var checker txm.TransmitCheckerSpec
if configWatcher.chain.Config().OCR2().SimulateTransactions() {
checker.CheckerType = txm.TransmitCheckerTypeSimulate
}
Expand Down Expand Up @@ -388,7 +388,7 @@ func newPipelineContractTransmitter(lggr logger.Logger, rargs relaytypes.RelayAr
scoped := configWatcher.chain.Config()
strategy := txmgrcommon.NewQueueingTxStrategy(rargs.ExternalJobID, scoped.OCR2().DefaultTransactionQueueDepth(), scoped.Database().DefaultQueryTimeout())

var checker txm.EvmTransmitCheckerSpec
var checker txm.TransmitCheckerSpec
if configWatcher.chain.Config().OCR2().SimulateTransactions() {
checker.CheckerType = txm.TransmitCheckerTypeSimulate
}
Expand Down
2 changes: 1 addition & 1 deletion core/services/vrf/listener_v2.go
Original file line number Diff line number Diff line change
Expand Up @@ -980,7 +980,7 @@ func (lsn *listenerV2) processRequestsPerSub(
RequestTxHash: &p.req.req.Raw.TxHash,
},
Strategy: txmgrcommon.NewSendEveryStrategy(),
Checker: txmgr.EvmTransmitCheckerSpec{
Checker: txmgr.TransmitCheckerSpec{
CheckerType: txmgr.TransmitCheckerTypeVRFV2,
VRFCoordinatorAddress: &coordinatorAddress,
VRFRequestBlockNumber: new(big.Int).SetUint64(p.req.req.Raw.BlockNumber),
Expand Down

0 comments on commit 491a8d7

Please sign in to comment.