aboutsummaryrefslogtreecommitdiffstats
path: root/p2p/peer_test.go
diff options
context:
space:
mode:
authorFelix Lange <fjl@twurst.com>2015-05-14 09:04:04 +0800
committerFelix Lange <fjl@twurst.com>2015-05-14 20:53:29 +0800
commit691cb90284b9b63dc9c80bf0716ba35036ca78fe (patch)
treedd169615d057af4f35b619c44fb66219760749ba /p2p/peer_test.go
parent7efeb4bd9647b6ab5376e171c1ff7cb516da5698 (diff)
downloaddexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar.gz
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar.bz2
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar.lz
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar.xz
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.tar.zst
dexon-691cb90284b9b63dc9c80bf0716ba35036ca78fe.zip
p2p: log remote reason when disconnect is requested
The returned reason is currently not used except for the log message. This change makes the log messages a bit more useful. The handshake code also returns the remote reason.
Diffstat (limited to 'p2p/peer_test.go')
-rw-r--r--p2p/peer_test.go13
1 files changed, 7 insertions, 6 deletions
diff --git a/p2p/peer_test.go b/p2p/peer_test.go
index fb76818a0..7d17d447c 100644
--- a/p2p/peer_test.go
+++ b/p2p/peer_test.go
@@ -172,12 +172,13 @@ func TestPeerDisconnect(t *testing.T) {
if err := SendItems(rw, discMsg, DiscQuitting); err != nil {
t.Fatal(err)
}
- if err := ExpectMsg(rw, discMsg, []interface{}{DiscRequested}); err != nil {
- t.Error(err)
- }
- closer()
- if reason := <-disc; reason != DiscRequested {
- t.Errorf("run returned wrong reason: got %v, want %v", reason, DiscRequested)
+ select {
+ case reason := <-disc:
+ if reason != DiscQuitting {
+ t.Errorf("run returned wrong reason: got %v, want %v", reason, DiscRequested)
+ }
+ case <-time.After(500 * time.Millisecond):
+ t.Error("peer did not return")
}
}