From 92c1ce57a698d6e45388a403557b592191ad46f3 Mon Sep 17 00:00:00 2001 From: Jakob Borg Date: Sun, 27 Apr 2014 13:25:35 +0200 Subject: [PATCH] Fix protocol close test --- protocol/protocol_test.go | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/protocol/protocol_test.go b/protocol/protocol_test.go index 80c4f83a3..9fe9b7510 100644 --- a/protocol/protocol_test.go +++ b/protocol/protocol_test.go @@ -5,6 +5,7 @@ import ( "io" "testing" "testing/quick" + "time" ) func TestHeaderFunctions(t *testing.T) { @@ -172,7 +173,13 @@ func TestClose(t *testing.T) { c0 := NewConnection("c0", ar, bw, m0).(wireFormatConnection).next.(*rawConnection) NewConnection("c1", br, aw, m1) - c0.close(nil) + c0.close <- nil + + select { + case <-c0.closed: + case <-time.After(1 * time.Second): + t.Fatal("Did not close within a second") + } if !c0.isClosed() { t.Fatal("Connection should be closed")