connection_test.go | 9 ++++++---

diff --git a/connection_test.go b/connection_test.go
index 9374a66d1defdd848e7673dec4136ba936df00f3..280afe9f2ceead4e0a4b351434d8c3e5645ff003 100644
--- a/connection_test.go
+++ b/connection_test.go
@@ -39,8 +39,11 @@ 			t.Fatalf("received message %q is not expected: %q", bb, b)
 		}
 	}
 	close(c.post)
-	_, ok := <-c.write
-	if ok {
-		t.Fatal("write channel didn't close")
+	// Drain the write channel until it closes.
+	for b := range c.write {
+		bs := string(b)
+		if bs != "\x00\x00\x00\x00" {
+			t.Fatal("got unexpected non-keepalive")
+		}
 	}
 }