package mock import ( "net" "github.com/tendermint/tendermint/internal/p2p" "github.com/tendermint/tendermint/internal/p2p/conn" "github.com/tendermint/tendermint/libs/service" "github.com/tendermint/tendermint/types" ) type Peer struct { *service.BaseService ip net.IP id types.NodeID addr *p2p.NetAddress kv map[string]interface{} Outbound, Persistent bool } // NewPeer creates and starts a new mock peer. If the ip // is nil, random routable address is used. func NewPeer(ip net.IP) *Peer { var netAddr *p2p.NetAddress if ip == nil { _, netAddr = p2p.CreateRoutableAddr() } else { netAddr = types.NewNetAddressIPPort(ip, 26656) } nodeKey := types.GenNodeKey() netAddr.ID = nodeKey.ID mp := &Peer{ ip: ip, id: nodeKey.ID, addr: netAddr, kv: make(map[string]interface{}), } mp.BaseService = service.NewBaseService(nil, "MockPeer", mp) if err := mp.Start(); err != nil { panic(err) } return mp } func (mp *Peer) FlushStop() { mp.Stop() } //nolint:errcheck //ignore error func (mp *Peer) TrySend(chID byte, msgBytes []byte) bool { return true } func (mp *Peer) Send(chID byte, msgBytes []byte) bool { return true } func (mp *Peer) NodeInfo() types.NodeInfo { return types.NodeInfo{ NodeID: mp.addr.ID, ListenAddr: mp.addr.DialString(), } } func (mp *Peer) Status() conn.ConnectionStatus { return conn.ConnectionStatus{} } func (mp *Peer) ID() types.NodeID { return mp.id } func (mp *Peer) IsOutbound() bool { return mp.Outbound } func (mp *Peer) IsPersistent() bool { return mp.Persistent } func (mp *Peer) Get(key string) interface{} { if value, ok := mp.kv[key]; ok { return value } return nil } func (mp *Peer) Set(key string, value interface{}) { mp.kv[key] = value } func (mp *Peer) RemoteIP() net.IP { return mp.ip } func (mp *Peer) SocketAddr() *p2p.NetAddress { return mp.addr } func (mp *Peer) RemoteAddr() net.Addr { return &net.TCPAddr{IP: mp.ip, Port: 8800} } func (mp *Peer) CloseConn() error { return nil }