diff --git a/protocol/protocol_test.go b/protocol/protocol_test.go index 9fe9b7510..80c4f83a3 100644 --- a/protocol/protocol_test.go +++ b/protocol/protocol_test.go @@ -5,7 +5,6 @@ import ( "io" "testing" "testing/quick" - "time" ) func TestHeaderFunctions(t *testing.T) { @@ -173,13 +172,7 @@ func TestClose(t *testing.T) { c0 := NewConnection("c0", ar, bw, m0).(wireFormatConnection).next.(*rawConnection) NewConnection("c1", br, aw, m1) - c0.close <- nil - - select { - case <-c0.closed: - case <-time.After(1 * time.Second): - t.Fatal("Did not close within a second") - } + c0.close(nil) if !c0.isClosed() { t.Fatal("Connection should be closed")