From 850edbf60d9530fe2ee3e7f1356e4ee72cc3722b Mon Sep 17 00:00:00 2001 From: Damjan Smickovski Date: Mon, 2 Oct 2023 13:04:21 +0200 Subject: [PATCH] Adding contract loader --- integration-tests/contracts/contract_loader.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/integration-tests/contracts/contract_loader.go b/integration-tests/contracts/contract_loader.go index fc0272b107e..e8ffa70d908 100644 --- a/integration-tests/contracts/contract_loader.go +++ b/integration-tests/contracts/contract_loader.go @@ -41,7 +41,7 @@ type ContractLoader interface { LoadWERC20Mock(addr common.Address) (WERC20Mock, error) } -// NewContractLoader returns an instance of a contract Loader based on the client type +// NewContractLoader returnsa an instance of a contract Loader based on the client type func NewContractLoader(bcClient blockchain.EVMClient, logger zerolog.Logger) (ContractLoader, error) { switch clientImpl := bcClient.Get().(type) { case *blockchain.EthereumClient: @@ -56,6 +56,8 @@ func NewContractLoader(bcClient blockchain.EVMClient, logger zerolog.Logger) (Co return &PolygonContractLoader{NewEthereumContractLoader(clientImpl, logger)}, nil case *blockchain.OptimismClient: return &OptimismContractLoader{NewEthereumContractLoader(clientImpl, logger)}, nil + case *blockchain.PolygonZkEvmClient: + return &PolygonZkEVMContractLoader{NewEthereumContractLoader(clientImpl, logger)}, nil } return nil, errors.New("unknown blockchain client implementation for contract Loader, register blockchain client in NewContractLoader") } @@ -86,6 +88,11 @@ type PolygonContractLoader struct { *EthereumContractLoader } +// PolygonZkEVMContractLoader wraps for Polygon +type PolygonZkEVMContractLoader struct { + *EthereumContractLoader +} + // OptimismContractLoader wraps for Optimism type OptimismContractLoader struct { *EthereumContractLoader