github.com/ethereum-optimism/optimism@v1.7.2/op-node/p2p/gating/mocks/ExpiryStore.go (about)

     1  // Code generated by mockery v2.28.1. DO NOT EDIT.
     2  
     3  package mocks
     4  
     5  import (
     6  	net "net"
     7  
     8  	mock "github.com/stretchr/testify/mock"
     9  
    10  	peer "github.com/libp2p/go-libp2p/core/peer"
    11  
    12  	time "time"
    13  )
    14  
    15  // ExpiryStore is an autogenerated mock type for the ExpiryStore type
    16  type ExpiryStore struct {
    17  	mock.Mock
    18  }
    19  
    20  type ExpiryStore_Expecter struct {
    21  	mock *mock.Mock
    22  }
    23  
    24  func (_m *ExpiryStore) EXPECT() *ExpiryStore_Expecter {
    25  	return &ExpiryStore_Expecter{mock: &_m.Mock}
    26  }
    27  
    28  // GetIPBanExpiration provides a mock function with given fields: ip
    29  func (_m *ExpiryStore) GetIPBanExpiration(ip net.IP) (time.Time, error) {
    30  	ret := _m.Called(ip)
    31  
    32  	var r0 time.Time
    33  	var r1 error
    34  	if rf, ok := ret.Get(0).(func(net.IP) (time.Time, error)); ok {
    35  		return rf(ip)
    36  	}
    37  	if rf, ok := ret.Get(0).(func(net.IP) time.Time); ok {
    38  		r0 = rf(ip)
    39  	} else {
    40  		r0 = ret.Get(0).(time.Time)
    41  	}
    42  
    43  	if rf, ok := ret.Get(1).(func(net.IP) error); ok {
    44  		r1 = rf(ip)
    45  	} else {
    46  		r1 = ret.Error(1)
    47  	}
    48  
    49  	return r0, r1
    50  }
    51  
    52  // ExpiryStore_GetIPBanExpiration_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetIPBanExpiration'
    53  type ExpiryStore_GetIPBanExpiration_Call struct {
    54  	*mock.Call
    55  }
    56  
    57  // GetIPBanExpiration is a helper method to define mock.On call
    58  //   - ip net.IP
    59  func (_e *ExpiryStore_Expecter) GetIPBanExpiration(ip interface{}) *ExpiryStore_GetIPBanExpiration_Call {
    60  	return &ExpiryStore_GetIPBanExpiration_Call{Call: _e.mock.On("GetIPBanExpiration", ip)}
    61  }
    62  
    63  func (_c *ExpiryStore_GetIPBanExpiration_Call) Run(run func(ip net.IP)) *ExpiryStore_GetIPBanExpiration_Call {
    64  	_c.Call.Run(func(args mock.Arguments) {
    65  		run(args[0].(net.IP))
    66  	})
    67  	return _c
    68  }
    69  
    70  func (_c *ExpiryStore_GetIPBanExpiration_Call) Return(_a0 time.Time, _a1 error) *ExpiryStore_GetIPBanExpiration_Call {
    71  	_c.Call.Return(_a0, _a1)
    72  	return _c
    73  }
    74  
    75  func (_c *ExpiryStore_GetIPBanExpiration_Call) RunAndReturn(run func(net.IP) (time.Time, error)) *ExpiryStore_GetIPBanExpiration_Call {
    76  	_c.Call.Return(run)
    77  	return _c
    78  }
    79  
    80  // GetPeerBanExpiration provides a mock function with given fields: id
    81  func (_m *ExpiryStore) GetPeerBanExpiration(id peer.ID) (time.Time, error) {
    82  	ret := _m.Called(id)
    83  
    84  	var r0 time.Time
    85  	var r1 error
    86  	if rf, ok := ret.Get(0).(func(peer.ID) (time.Time, error)); ok {
    87  		return rf(id)
    88  	}
    89  	if rf, ok := ret.Get(0).(func(peer.ID) time.Time); ok {
    90  		r0 = rf(id)
    91  	} else {
    92  		r0 = ret.Get(0).(time.Time)
    93  	}
    94  
    95  	if rf, ok := ret.Get(1).(func(peer.ID) error); ok {
    96  		r1 = rf(id)
    97  	} else {
    98  		r1 = ret.Error(1)
    99  	}
   100  
   101  	return r0, r1
   102  }
   103  
   104  // ExpiryStore_GetPeerBanExpiration_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetPeerBanExpiration'
   105  type ExpiryStore_GetPeerBanExpiration_Call struct {
   106  	*mock.Call
   107  }
   108  
   109  // GetPeerBanExpiration is a helper method to define mock.On call
   110  //   - id peer.ID
   111  func (_e *ExpiryStore_Expecter) GetPeerBanExpiration(id interface{}) *ExpiryStore_GetPeerBanExpiration_Call {
   112  	return &ExpiryStore_GetPeerBanExpiration_Call{Call: _e.mock.On("GetPeerBanExpiration", id)}
   113  }
   114  
   115  func (_c *ExpiryStore_GetPeerBanExpiration_Call) Run(run func(id peer.ID)) *ExpiryStore_GetPeerBanExpiration_Call {
   116  	_c.Call.Run(func(args mock.Arguments) {
   117  		run(args[0].(peer.ID))
   118  	})
   119  	return _c
   120  }
   121  
   122  func (_c *ExpiryStore_GetPeerBanExpiration_Call) Return(_a0 time.Time, _a1 error) *ExpiryStore_GetPeerBanExpiration_Call {
   123  	_c.Call.Return(_a0, _a1)
   124  	return _c
   125  }
   126  
   127  func (_c *ExpiryStore_GetPeerBanExpiration_Call) RunAndReturn(run func(peer.ID) (time.Time, error)) *ExpiryStore_GetPeerBanExpiration_Call {
   128  	_c.Call.Return(run)
   129  	return _c
   130  }
   131  
   132  // SetIPBanExpiration provides a mock function with given fields: ip, expiry
   133  func (_m *ExpiryStore) SetIPBanExpiration(ip net.IP, expiry time.Time) error {
   134  	ret := _m.Called(ip, expiry)
   135  
   136  	var r0 error
   137  	if rf, ok := ret.Get(0).(func(net.IP, time.Time) error); ok {
   138  		r0 = rf(ip, expiry)
   139  	} else {
   140  		r0 = ret.Error(0)
   141  	}
   142  
   143  	return r0
   144  }
   145  
   146  // ExpiryStore_SetIPBanExpiration_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetIPBanExpiration'
   147  type ExpiryStore_SetIPBanExpiration_Call struct {
   148  	*mock.Call
   149  }
   150  
   151  // SetIPBanExpiration is a helper method to define mock.On call
   152  //   - ip net.IP
   153  //   - expiry time.Time
   154  func (_e *ExpiryStore_Expecter) SetIPBanExpiration(ip interface{}, expiry interface{}) *ExpiryStore_SetIPBanExpiration_Call {
   155  	return &ExpiryStore_SetIPBanExpiration_Call{Call: _e.mock.On("SetIPBanExpiration", ip, expiry)}
   156  }
   157  
   158  func (_c *ExpiryStore_SetIPBanExpiration_Call) Run(run func(ip net.IP, expiry time.Time)) *ExpiryStore_SetIPBanExpiration_Call {
   159  	_c.Call.Run(func(args mock.Arguments) {
   160  		run(args[0].(net.IP), args[1].(time.Time))
   161  	})
   162  	return _c
   163  }
   164  
   165  func (_c *ExpiryStore_SetIPBanExpiration_Call) Return(_a0 error) *ExpiryStore_SetIPBanExpiration_Call {
   166  	_c.Call.Return(_a0)
   167  	return _c
   168  }
   169  
   170  func (_c *ExpiryStore_SetIPBanExpiration_Call) RunAndReturn(run func(net.IP, time.Time) error) *ExpiryStore_SetIPBanExpiration_Call {
   171  	_c.Call.Return(run)
   172  	return _c
   173  }
   174  
   175  // SetPeerBanExpiration provides a mock function with given fields: id, expiry
   176  func (_m *ExpiryStore) SetPeerBanExpiration(id peer.ID, expiry time.Time) error {
   177  	ret := _m.Called(id, expiry)
   178  
   179  	var r0 error
   180  	if rf, ok := ret.Get(0).(func(peer.ID, time.Time) error); ok {
   181  		r0 = rf(id, expiry)
   182  	} else {
   183  		r0 = ret.Error(0)
   184  	}
   185  
   186  	return r0
   187  }
   188  
   189  // ExpiryStore_SetPeerBanExpiration_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetPeerBanExpiration'
   190  type ExpiryStore_SetPeerBanExpiration_Call struct {
   191  	*mock.Call
   192  }
   193  
   194  // SetPeerBanExpiration is a helper method to define mock.On call
   195  //   - id peer.ID
   196  //   - expiry time.Time
   197  func (_e *ExpiryStore_Expecter) SetPeerBanExpiration(id interface{}, expiry interface{}) *ExpiryStore_SetPeerBanExpiration_Call {
   198  	return &ExpiryStore_SetPeerBanExpiration_Call{Call: _e.mock.On("SetPeerBanExpiration", id, expiry)}
   199  }
   200  
   201  func (_c *ExpiryStore_SetPeerBanExpiration_Call) Run(run func(id peer.ID, expiry time.Time)) *ExpiryStore_SetPeerBanExpiration_Call {
   202  	_c.Call.Run(func(args mock.Arguments) {
   203  		run(args[0].(peer.ID), args[1].(time.Time))
   204  	})
   205  	return _c
   206  }
   207  
   208  func (_c *ExpiryStore_SetPeerBanExpiration_Call) Return(_a0 error) *ExpiryStore_SetPeerBanExpiration_Call {
   209  	_c.Call.Return(_a0)
   210  	return _c
   211  }
   212  
   213  func (_c *ExpiryStore_SetPeerBanExpiration_Call) RunAndReturn(run func(peer.ID, time.Time) error) *ExpiryStore_SetPeerBanExpiration_Call {
   214  	_c.Call.Return(run)
   215  	return _c
   216  }
   217  
   218  type mockConstructorTestingTNewExpiryStore interface {
   219  	mock.TestingT
   220  	Cleanup(func())
   221  }
   222  
   223  // NewExpiryStore creates a new instance of ExpiryStore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   224  func NewExpiryStore(t mockConstructorTestingTNewExpiryStore) *ExpiryStore {
   225  	mock := &ExpiryStore{}
   226  	mock.Mock.Test(t)
   227  
   228  	t.Cleanup(func() { mock.AssertExpectations(t) })
   229  
   230  	return mock
   231  }