Update bufpipe_test.go

Add test for close deadlock
acomagu · May 10, 2021 · 55b70ec · 55b70ec
1 parent 2cfa345
commit 55b70ec
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/bufpipe_test.go b/bufpipe_test.go
@@ -128,3 +128,20 @@ func TestPipeReader_CloseWithError(t *testing.T) {
 	is.Equal(err, expect)
 	is.Equal(n, 0)
 }
+
+func TestPipeReader_WriterCloseNoDeadlock(t *testing.T) {
+	r, w := bufpipe.New(nil)
+
+	done := make(chan struct{})
+	go func(t *testing.T) {
+		buf := make([]byte, 800)
+		r.Read(buf)
+		done <- struct{}{}
+	}(t)
+
+	time.Sleep(300 * time.Millisecond)
+	w.Close()
+
+	<-done
+}
+