forked from allora-network/b7s
/
peerstore.go
54 lines (43 loc) · 1.18 KB
/
peerstore.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package mocks
import (
"testing"
"github.com/libp2p/go-libp2p/core/peer"
"github.com/multiformats/go-multiaddr"
"github.com/Bruce960205/b7s/models/blockless"
)
type PeerStore struct {
GetFunc func(peer.ID) (blockless.Peer, error)
StoreFunc func(peer.ID, multiaddr.Multiaddr, peer.AddrInfo) error
PeersFunc func() ([]blockless.Peer, error)
RemoveFunc func(peer.ID) error
}
func BaselinePeerStore(t *testing.T) *PeerStore {
t.Helper()
peerstore := PeerStore{
GetFunc: func(peer.ID) (blockless.Peer, error) {
return blockless.Peer{}, nil
},
StoreFunc: func(peer.ID, multiaddr.Multiaddr, peer.AddrInfo) error {
return nil
},
PeersFunc: func() ([]blockless.Peer, error) {
return []blockless.Peer{}, nil
},
RemoveFunc: func(peer.ID) error {
return GenericError
},
}
return &peerstore
}
func (p *PeerStore) Get(id peer.ID) (blockless.Peer, error) {
return p.GetFunc(id)
}
func (p *PeerStore) Store(id peer.ID, addr multiaddr.Multiaddr, info peer.AddrInfo) error {
return p.StoreFunc(id, addr, info)
}
func (p *PeerStore) Peers() ([]blockless.Peer, error) {
return p.PeersFunc()
}
func (p *PeerStore) Remove(id peer.ID) error {
return p.RemoveFunc(id)
}