mock2/ connections
Juan Batiz-Benet committed
Dec 17, 2014 at 06:32 UTC
2f9d398d6602337768868d908c72e9449ca6d720
2 files changed
+8
-5
net/interface.go
+3
@@ -108,6 +108,9 @@ type Dialer interface {
108
// DialPeer attempts to establish a connection to a given peer
109
DialPeer(context.Context, peer.Peer) error
110
111
+ // ClosePeer closes the connection to a given peer
112
+ ClosePeer(peer.Peer) error
113
+
114
// Connectedness returns a state signaling connection capabilities
115
Connectedness(peer.Peer) Connectedness
116
}
net/mock2/mock_net.go
+5
-5
@@ -243,19 +243,19 @@ func (mn *mocknet) ConnectAll() error {
243
}
244
245
func (mn *mocknet) ConnectPeers(a, b peer.Peer) error {
246
- panic("nyi")
246
+ return mn.Net(a.ID()).DialPeer(mn.cg.Context(), b)
247
}
248
249
-func (mn *mocknet) ConnectNets(inet.Network, inet.Network) error {
250
- panic("nyi")
249
+func (mn *mocknet) ConnectNets(a, b inet.Network) error {
250
+ return a.DialPeer(mn.cg.Context(), b.LocalPeer())
251
}
252
253
func (mn *mocknet) DisconnectPeers(p1, p2 peer.Peer) error {
254
- panic("nyi")
254
+ return mn.Net(p1.ID()).ClosePeer(p2)
255
}
256
257
func (mn *mocknet) DisconnectNets(n1, n2 inet.Network) error {
258
- return mn.DisconnectPeers(n1.LocalPeer(), n2.LocalPeer())
258
+ return n1.ClosePeer(n2.LocalPeer())
259
}
260
261
func (mn *mocknet) LinksBetweenPeers(p1, p2 peer.Peer) []Link {