Address review comments

googleapis · Jun 3, 2021 · 027d9e5 · 027d9e5
1 parent 8bd3716
commit 027d9e5
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/bigtable/bttest/inmem_test.go b/bigtable/bttest/inmem_test.go
@@ -334,8 +334,13 @@ func TestTableRowsConcurrent(t *testing.T) {
 		}
 		finished <- true
 	}()
-	<-finished
-	<-finished
+	for i := 0; i < 2; i++ {
+		select {
+		case <-finished:
+		case <-time.After(2 * time.Second):
+			t.Fatalf("Timeout waiting for task %d\n", i)
+		}
+	}
 }
 
 func TestDropRowRange(t *testing.T) {