fix deadlock

apache · Oct 23, 2024 · 9684e2e · 9684e2e
1 parent 61ae751
commit 9684e2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/arrow/cdata/cdata_test.go b/arrow/cdata/cdata_test.go
@@ -1037,7 +1037,7 @@ func TestAsyncInterfacesSimple(t *testing.T) {
 	ctx := context.Background()
 	ch := CreateAsyncDeviceStreamHandler(ctx, 1, handler)
 
-	stream := make(chan RecordMessage, 1)
+	stream := make(chan RecordMessage, len(reclist))
 	go func() {
 		defer close(stream)
 		for _, r := range reclist {
@@ -1101,7 +1101,7 @@ func TestAsyncPropagateError(t *testing.T) {
 	ctx := context.Background()
 	ch := CreateAsyncDeviceStreamHandler(ctx, 1, handler)
 
-	stream := make(chan RecordMessage, 1)
+	stream := make(chan RecordMessage, 2)
 	go func() {
 		defer close(stream)
 		reclist[0].Retain()