github.com/KYVENetwork/cometbft/v38@v38.0.3/p2p/mocks/peer.go (about)

     1  // Code generated by mockery. DO NOT EDIT.
     2  
     3  package mocks
     4  
     5  import (
     6  	log "github.com/KYVENetwork/cometbft/v38/libs/log"
     7  	conn "github.com/KYVENetwork/cometbft/v38/p2p/conn"
     8  
     9  	mock "github.com/stretchr/testify/mock"
    10  
    11  	net "net"
    12  
    13  	p2p "github.com/KYVENetwork/cometbft/v38/p2p"
    14  )
    15  
    16  // Peer is an autogenerated mock type for the Peer type
    17  type Peer struct {
    18  	mock.Mock
    19  }
    20  
    21  // CloseConn provides a mock function with given fields:
    22  func (_m *Peer) CloseConn() error {
    23  	ret := _m.Called()
    24  
    25  	var r0 error
    26  	if rf, ok := ret.Get(0).(func() error); ok {
    27  		r0 = rf()
    28  	} else {
    29  		r0 = ret.Error(0)
    30  	}
    31  
    32  	return r0
    33  }
    34  
    35  // FlushStop provides a mock function with given fields:
    36  func (_m *Peer) FlushStop() {
    37  	_m.Called()
    38  }
    39  
    40  // Get provides a mock function with given fields: _a0
    41  func (_m *Peer) Get(_a0 string) interface{} {
    42  	ret := _m.Called(_a0)
    43  
    44  	var r0 interface{}
    45  	if rf, ok := ret.Get(0).(func(string) interface{}); ok {
    46  		r0 = rf(_a0)
    47  	} else {
    48  		if ret.Get(0) != nil {
    49  			r0 = ret.Get(0).(interface{})
    50  		}
    51  	}
    52  
    53  	return r0
    54  }
    55  
    56  // GetRemovalFailed provides a mock function with given fields:
    57  func (_m *Peer) GetRemovalFailed() bool {
    58  	ret := _m.Called()
    59  
    60  	var r0 bool
    61  	if rf, ok := ret.Get(0).(func() bool); ok {
    62  		r0 = rf()
    63  	} else {
    64  		r0 = ret.Get(0).(bool)
    65  	}
    66  
    67  	return r0
    68  }
    69  
    70  // ID provides a mock function with given fields:
    71  func (_m *Peer) ID() p2p.ID {
    72  	ret := _m.Called()
    73  
    74  	var r0 p2p.ID
    75  	if rf, ok := ret.Get(0).(func() p2p.ID); ok {
    76  		r0 = rf()
    77  	} else {
    78  		r0 = ret.Get(0).(p2p.ID)
    79  	}
    80  
    81  	return r0
    82  }
    83  
    84  // IsOutbound provides a mock function with given fields:
    85  func (_m *Peer) IsOutbound() bool {
    86  	ret := _m.Called()
    87  
    88  	var r0 bool
    89  	if rf, ok := ret.Get(0).(func() bool); ok {
    90  		r0 = rf()
    91  	} else {
    92  		r0 = ret.Get(0).(bool)
    93  	}
    94  
    95  	return r0
    96  }
    97  
    98  // IsPersistent provides a mock function with given fields:
    99  func (_m *Peer) IsPersistent() bool {
   100  	ret := _m.Called()
   101  
   102  	var r0 bool
   103  	if rf, ok := ret.Get(0).(func() bool); ok {
   104  		r0 = rf()
   105  	} else {
   106  		r0 = ret.Get(0).(bool)
   107  	}
   108  
   109  	return r0
   110  }
   111  
   112  // IsRunning provides a mock function with given fields:
   113  func (_m *Peer) IsRunning() bool {
   114  	ret := _m.Called()
   115  
   116  	var r0 bool
   117  	if rf, ok := ret.Get(0).(func() bool); ok {
   118  		r0 = rf()
   119  	} else {
   120  		r0 = ret.Get(0).(bool)
   121  	}
   122  
   123  	return r0
   124  }
   125  
   126  // NodeInfo provides a mock function with given fields:
   127  func (_m *Peer) NodeInfo() p2p.NodeInfo {
   128  	ret := _m.Called()
   129  
   130  	var r0 p2p.NodeInfo
   131  	if rf, ok := ret.Get(0).(func() p2p.NodeInfo); ok {
   132  		r0 = rf()
   133  	} else {
   134  		if ret.Get(0) != nil {
   135  			r0 = ret.Get(0).(p2p.NodeInfo)
   136  		}
   137  	}
   138  
   139  	return r0
   140  }
   141  
   142  // OnReset provides a mock function with given fields:
   143  func (_m *Peer) OnReset() error {
   144  	ret := _m.Called()
   145  
   146  	var r0 error
   147  	if rf, ok := ret.Get(0).(func() error); ok {
   148  		r0 = rf()
   149  	} else {
   150  		r0 = ret.Error(0)
   151  	}
   152  
   153  	return r0
   154  }
   155  
   156  // OnStart provides a mock function with given fields:
   157  func (_m *Peer) OnStart() error {
   158  	ret := _m.Called()
   159  
   160  	var r0 error
   161  	if rf, ok := ret.Get(0).(func() error); ok {
   162  		r0 = rf()
   163  	} else {
   164  		r0 = ret.Error(0)
   165  	}
   166  
   167  	return r0
   168  }
   169  
   170  // OnStop provides a mock function with given fields:
   171  func (_m *Peer) OnStop() {
   172  	_m.Called()
   173  }
   174  
   175  // Quit provides a mock function with given fields:
   176  func (_m *Peer) Quit() <-chan struct{} {
   177  	ret := _m.Called()
   178  
   179  	var r0 <-chan struct{}
   180  	if rf, ok := ret.Get(0).(func() <-chan struct{}); ok {
   181  		r0 = rf()
   182  	} else {
   183  		if ret.Get(0) != nil {
   184  			r0 = ret.Get(0).(<-chan struct{})
   185  		}
   186  	}
   187  
   188  	return r0
   189  }
   190  
   191  // RemoteAddr provides a mock function with given fields:
   192  func (_m *Peer) RemoteAddr() net.Addr {
   193  	ret := _m.Called()
   194  
   195  	var r0 net.Addr
   196  	if rf, ok := ret.Get(0).(func() net.Addr); ok {
   197  		r0 = rf()
   198  	} else {
   199  		if ret.Get(0) != nil {
   200  			r0 = ret.Get(0).(net.Addr)
   201  		}
   202  	}
   203  
   204  	return r0
   205  }
   206  
   207  // RemoteIP provides a mock function with given fields:
   208  func (_m *Peer) RemoteIP() net.IP {
   209  	ret := _m.Called()
   210  
   211  	var r0 net.IP
   212  	if rf, ok := ret.Get(0).(func() net.IP); ok {
   213  		r0 = rf()
   214  	} else {
   215  		if ret.Get(0) != nil {
   216  			r0 = ret.Get(0).(net.IP)
   217  		}
   218  	}
   219  
   220  	return r0
   221  }
   222  
   223  // Reset provides a mock function with given fields:
   224  func (_m *Peer) Reset() error {
   225  	ret := _m.Called()
   226  
   227  	var r0 error
   228  	if rf, ok := ret.Get(0).(func() error); ok {
   229  		r0 = rf()
   230  	} else {
   231  		r0 = ret.Error(0)
   232  	}
   233  
   234  	return r0
   235  }
   236  
   237  // Send provides a mock function with given fields: _a0
   238  func (_m *Peer) Send(_a0 p2p.Envelope) bool {
   239  	ret := _m.Called(_a0)
   240  
   241  	var r0 bool
   242  	if rf, ok := ret.Get(0).(func(p2p.Envelope) bool); ok {
   243  		r0 = rf(_a0)
   244  	} else {
   245  		r0 = ret.Get(0).(bool)
   246  	}
   247  
   248  	return r0
   249  }
   250  
   251  // Set provides a mock function with given fields: _a0, _a1
   252  func (_m *Peer) Set(_a0 string, _a1 interface{}) {
   253  	_m.Called(_a0, _a1)
   254  }
   255  
   256  // SetLogger provides a mock function with given fields: _a0
   257  func (_m *Peer) SetLogger(_a0 log.Logger) {
   258  	_m.Called(_a0)
   259  }
   260  
   261  // SetRemovalFailed provides a mock function with given fields:
   262  func (_m *Peer) SetRemovalFailed() {
   263  	_m.Called()
   264  }
   265  
   266  // SocketAddr provides a mock function with given fields:
   267  func (_m *Peer) SocketAddr() *p2p.NetAddress {
   268  	ret := _m.Called()
   269  
   270  	var r0 *p2p.NetAddress
   271  	if rf, ok := ret.Get(0).(func() *p2p.NetAddress); ok {
   272  		r0 = rf()
   273  	} else {
   274  		if ret.Get(0) != nil {
   275  			r0 = ret.Get(0).(*p2p.NetAddress)
   276  		}
   277  	}
   278  
   279  	return r0
   280  }
   281  
   282  // Start provides a mock function with given fields:
   283  func (_m *Peer) Start() error {
   284  	ret := _m.Called()
   285  
   286  	var r0 error
   287  	if rf, ok := ret.Get(0).(func() error); ok {
   288  		r0 = rf()
   289  	} else {
   290  		r0 = ret.Error(0)
   291  	}
   292  
   293  	return r0
   294  }
   295  
   296  // Status provides a mock function with given fields:
   297  func (_m *Peer) Status() conn.ConnectionStatus {
   298  	ret := _m.Called()
   299  
   300  	var r0 conn.ConnectionStatus
   301  	if rf, ok := ret.Get(0).(func() conn.ConnectionStatus); ok {
   302  		r0 = rf()
   303  	} else {
   304  		r0 = ret.Get(0).(conn.ConnectionStatus)
   305  	}
   306  
   307  	return r0
   308  }
   309  
   310  // Stop provides a mock function with given fields:
   311  func (_m *Peer) Stop() error {
   312  	ret := _m.Called()
   313  
   314  	var r0 error
   315  	if rf, ok := ret.Get(0).(func() error); ok {
   316  		r0 = rf()
   317  	} else {
   318  		r0 = ret.Error(0)
   319  	}
   320  
   321  	return r0
   322  }
   323  
   324  // String provides a mock function with given fields:
   325  func (_m *Peer) String() string {
   326  	ret := _m.Called()
   327  
   328  	var r0 string
   329  	if rf, ok := ret.Get(0).(func() string); ok {
   330  		r0 = rf()
   331  	} else {
   332  		r0 = ret.Get(0).(string)
   333  	}
   334  
   335  	return r0
   336  }
   337  
   338  // TrySend provides a mock function with given fields: _a0
   339  func (_m *Peer) TrySend(_a0 p2p.Envelope) bool {
   340  	ret := _m.Called(_a0)
   341  
   342  	var r0 bool
   343  	if rf, ok := ret.Get(0).(func(p2p.Envelope) bool); ok {
   344  		r0 = rf(_a0)
   345  	} else {
   346  		r0 = ret.Get(0).(bool)
   347  	}
   348  
   349  	return r0
   350  }
   351  
   352  type mockConstructorTestingTNewPeer interface {
   353  	mock.TestingT
   354  	Cleanup(func())
   355  }
   356  
   357  // NewPeer creates a new instance of Peer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
   358  func NewPeer(t mockConstructorTestingTNewPeer) *Peer {
   359  	mock := &Peer{}
   360  	mock.Mock.Test(t)
   361  
   362  	t.Cleanup(func() { mock.AssertExpectations(t) })
   363  
   364  	return mock
   365  }