Actually close connection based on unknown protocol version

2015-01-08 22:11:10 +01:00
parent b88e3c99c1
commit bc9fc1aece
2 changed files with 7 additions and 2 deletions
--- a/internal/protocol/protocol_test.go
+++ b/internal/protocol/protocol_test.go
@@ -189,7 +189,7 @@ func TestVersionErr(t *testing.T) {
 		msgID:   0,
 		msgType: 0,
 	}))
-	w.WriteUint32(0)
+	w.WriteUint32(0) // Avoids reader closing due to EOF

 	if !m1.isClosed() {
 		t.Error("Connection should close due to unknown version")
@@ -212,7 +212,7 @@ func TestTypeErr(t *testing.T) {
 		msgID:   0,
 		msgType: 42,
 	}))
-	w.WriteUint32(0)
+	w.WriteUint32(0) // Avoids reader closing due to EOF

 	if !m1.isClosed() {
 		t.Error("Connection should close due to unknown message type")