@@ -24,8 +24,6 @@ import (
2424
2525 sel "github.com/smartcontractkit/chain-selectors"
2626
27- "github.com/smartcontractkit/chainlink-common/pkg/types"
28-
2927 "github.com/smartcontractkit/chainlink-ccip/pkg/contractreader"
3028 "github.com/smartcontractkit/chainlink-ccip/pkg/reader"
3129 cciptypes "github.com/smartcontractkit/chainlink-ccip/pkg/types/ccipocr3"
@@ -83,7 +81,7 @@ func Test_USDCReader_MessageHashes(t *testing.T) {
8381 SourceMessageTransmitterAddr : ts .contractAddr .String (),
8482 },
8583 },
86- map [cciptypes.ChainSelector ]contractreader.ContractReaderFacade {
84+ map [cciptypes.ChainSelector ]contractreader.Extended {
8785 ethereumChain : ts .reader ,
8886 }, mokAddrCodec )
8987 require .NoError (t , err )
@@ -283,7 +281,7 @@ func Benchmark_MessageHashes(b *testing.B) {
283281 SourceMessageTransmitterAddr : ts .contractAddr .String (),
284282 },
285283 },
286- map [cciptypes.ChainSelector ]contractreader.ContractReaderFacade {
284+ map [cciptypes.ChainSelector ]contractreader.Extended {
287285 sourceChain : ts .reader ,
288286 }, mokAddrCodec )
289287 require .NoError (b , err )
@@ -473,13 +471,16 @@ func testSetup(ctx context.Context, t testing.TB, readerChain cciptypes.ChainSel
473471 require .NoError (t , db .Close ())
474472 })
475473
474+ // Convert to the extended contract reader interface.
475+ ecr := contractreader .NewExtendedContractReader (
476+ (contractreader .ContractReaderFacade )(cr ))
476477 return & testSetupData {
477478 contractAddr : address ,
478479 contract : contract ,
479480 sb : simulatedBackend ,
480481 auth : auth ,
481482 cl : cl ,
482- reader : cr ,
483+ reader : ecr ,
483484 orm : orm ,
484485 db : db ,
485486 lp : lp ,
@@ -492,7 +493,7 @@ type testSetupData struct {
492493 sb * simulated.Backend
493494 auth * bind.TransactOpts
494495 cl client.Client
495- reader types. ContractReader
496+ reader contractreader. Extended
496497 orm logpoller.ORM
497498 db * sqlx.DB
498499 lp logpoller.LogPoller
0 commit comments