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

     1  // Code generated by mockery v2.28.1. DO NOT EDIT.
     2  
     3  package mocks
     4  
     5  import (
     6  	mock "github.com/stretchr/testify/mock"
     7  
     8  	peer "github.com/libp2p/go-libp2p/core/peer"
     9  
    10  	store "github.com/ethereum-optimism/optimism/op-node/p2p/store"
    11  )
    12  
    13  // Peerstore is an autogenerated mock type for the Peerstore type
    14  type Peerstore struct {
    15  	mock.Mock
    16  }
    17  
    18  // PeerInfo provides a mock function with given fields: _a0
    19  func (_m *Peerstore) PeerInfo(_a0 peer.ID) peer.AddrInfo {
    20  	ret := _m.Called(_a0)
    21  
    22  	var r0 peer.AddrInfo
    23  	if rf, ok := ret.Get(0).(func(peer.ID) peer.AddrInfo); ok {
    24  		r0 = rf(_a0)
    25  	} else {
    26  		r0 = ret.Get(0).(peer.AddrInfo)
    27  	}
    28  
    29  	return r0
    30  }
    31  
    32  // Peers provides a mock function with given fields:
    33  func (_m *Peerstore) Peers() peer.IDSlice {
    34  	ret := _m.Called()
    35  
    36  	var r0 peer.IDSlice
    37  	if rf, ok := ret.Get(0).(func() peer.IDSlice); ok {
    38  		r0 = rf()
    39  	} else {
    40  		if ret.Get(0) != nil {
    41  			r0 = ret.Get(0).(peer.IDSlice)
    42  		}
    43  	}
    44  
    45  	return r0
    46  }
    47  
    48  // SetScore provides a mock function with given fields: id, diff
    49  func (_m *Peerstore) SetScore(id peer.ID, diff store.ScoreDiff) (store.PeerScores, error) {
    50  	ret := _m.Called(id, diff)
    51  
    52  	var r0 store.PeerScores
    53  	var r1 error
    54  	if rf, ok := ret.Get(0).(func(peer.ID, store.ScoreDiff) (store.PeerScores, error)); ok {
    55  		return rf(id, diff)
    56  	}
    57  	if rf, ok := ret.Get(0).(func(peer.ID, store.ScoreDiff) store.PeerScores); ok {
    58  		r0 = rf(id, diff)
    59  	} else {
    60  		r0 = ret.Get(0).(store.PeerScores)
    61  	}
    62  
    63  	if rf, ok := ret.Get(1).(func(peer.ID, store.ScoreDiff) error); ok {
    64  		r1 = rf(id, diff)
    65  	} else {
    66  		r1 = ret.Error(1)
    67  	}
    68  
    69  	return r0, r1
    70  }
    71  
    72  type mockConstructorTestingTNewPeerstore interface {
    73  	mock.TestingT
    74  	Cleanup(func())
    75  }
    76  
    77  // NewPeerstore creates a new instance of Peerstore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
    78  func NewPeerstore(t mockConstructorTestingTNewPeerstore) *Peerstore {
    79  	mock := &Peerstore{}
    80  	mock.Mock.Test(t)
    81  
    82  	t.Cleanup(func() { mock.AssertExpectations(t) })
    83  
    84  	return mock
    85  }