Test HandleIO loop with error reader

dustin · Nov 19, 2013 · 87b02f6 · 87b02f6
1 parent d661b11
commit 87b02f6
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/server/server_test.go b/server/server_test.go
@@ -54,6 +54,28 @@ func TestTransmitRes(t *testing.T) {
 	}
 }
 
+type closedrwc struct{}
+
+func (c closedrwc) Close() error {
+	return nil
+}
+
+func (c closedrwc) Write([]byte) (int, error) {
+	return 0, io.EOF
+}
+
+func (c closedrwc) Read([]byte) (int, error) {
+	return 0, io.EOF
+}
+
+func TestHandleIO(t *testing.T) {
+	c := closedrwc{}
+	err := HandleIO(c, nil)
+	if err != io.EOF {
+		t.Errorf("Expected EOF, got %v", err)
+	}
+}
+
 func BenchmarkTransmitRes(b *testing.B) {
 	bout := &bytes.Buffer{}