From 260313e3f61f35d84d2fdb6cd238bd27581d990b Mon Sep 17 00:00:00 2001 From: Madhur Shrimal Date: Tue, 2 Jul 2024 17:43:15 -0700 Subject: [PATCH] mocks --- chainio/clients/elcontracts/reader.go | 2 ++ chainio/mocks/elContractsReader.go | 15 +++++++++++++++ 2 files changed, 17 insertions(+) diff --git a/chainio/clients/elcontracts/reader.go b/chainio/clients/elcontracts/reader.go index 8ef06270..d0c0c210 100644 --- a/chainio/clients/elcontracts/reader.go +++ b/chainio/clients/elcontracts/reader.go @@ -62,6 +62,8 @@ type ELReader interface { ) ([32]byte, error) GetDistributionRootsLength(opts *bind.CallOpts) (*big.Int, error) + + CurrRewardsCalculationEndTimestamp(opts *bind.CallOpts) (uint32, error) } type Config struct { diff --git a/chainio/mocks/elContractsReader.go b/chainio/mocks/elContractsReader.go index 03881c98..d8078172 100644 --- a/chainio/mocks/elContractsReader.go +++ b/chainio/mocks/elContractsReader.go @@ -74,6 +74,21 @@ func (mr *MockELReaderMockRecorder) CalculateOperatorAVSRegistrationDigestHash(a return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CalculateOperatorAVSRegistrationDigestHash", reflect.TypeOf((*MockELReader)(nil).CalculateOperatorAVSRegistrationDigestHash), arg0, arg1, arg2, arg3, arg4) } +// CurrRewardsCalculationEndTimestamp mocks base method. +func (m *MockELReader) CurrRewardsCalculationEndTimestamp(arg0 *bind.CallOpts) (uint32, error) { + m.ctrl.T.Helper() + ret := m.ctrl.Call(m, "CurrRewardsCalculationEndTimestamp", arg0) + ret0, _ := ret[0].(uint32) + ret1, _ := ret[1].(error) + return ret0, ret1 +} + +// CurrRewardsCalculationEndTimestamp indicates an expected call of CurrRewardsCalculationEndTimestamp. +func (mr *MockELReaderMockRecorder) CurrRewardsCalculationEndTimestamp(arg0 any) *gomock.Call { + mr.mock.ctrl.T.Helper() + return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CurrRewardsCalculationEndTimestamp", reflect.TypeOf((*MockELReader)(nil).CurrRewardsCalculationEndTimestamp), arg0) +} + // GetDistributionRootsLength mocks base method. func (m *MockELReader) GetDistributionRootsLength(arg0 *bind.CallOpts) (*big.Int, error) { m.ctrl.T.Helper()