diff options
Diffstat (limited to 'core/test/fake-transport.go')
| -rw-r--r-- | core/test/fake-transport.go | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/core/test/fake-transport.go b/core/test/fake-transport.go index ed9871b..cecac54 100644 --- a/core/test/fake-transport.go +++ b/core/test/fake-transport.go @@ -65,6 +65,11 @@ func NewFakeTransportClient(pubKey crypto.PublicKey) TransportClient { } } +// Disconnect implements Transport.Disconnect method. +func (t *FakeTransport) Disconnect(endpoint types.NodeID) { + delete(t.peers, endpoint) +} + // Send implements Transport.Send method. func (t *FakeTransport) Send( endpoint types.NodeID, msg interface{}) (err error) { |