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 }