From eac22abbd53c2313cc8eff2044b26671a281568c Mon Sep 17 00:00:00 2001 From: Sonic Date: Mon, 12 Nov 2018 09:18:03 +0800 Subject: dex: porting test to enode (#9) --- p2p/peer.go | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'p2p/peer.go') diff --git a/p2p/peer.go b/p2p/peer.go index 2c357fdc9..3d6ce9dcd 100644 --- a/p2p/peer.go +++ b/p2p/peer.go @@ -125,6 +125,15 @@ func NewPeer(id enode.ID, name string, caps []Cap) *Peer { return peer } +// NewPeer returns peer with given enode.Node for testing purposes. +func NewPeerWithEnode(node *enode.Node, name string, caps []Cap) *Peer { + pipe, _ := net.Pipe() + conn := &conn{fd: pipe, transport: nil, node: node, caps: caps, name: name} + peer := newPeer(conn, nil) + close(peer.closed) // ensures Disconnect doesn't block + return peer +} + // ID returns the node's public key. func (p *Peer) ID() enode.ID { return p.rw.node.ID() -- cgit v1.2.3