GODRIVER-1875 reduce TestPool test timeout (#582)

mongodb · Feb 19, 2021 · 07b0039 · 07b0039
1 parent c3587ec
commit 07b0039
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/x/mongo/driver/topology/pool_test.go b/x/mongo/driver/topology/pool_test.go
@@ -216,7 +216,7 @@ func TestPool(t *testing.T) {
 			disconnectDone := make(chan struct{})
 			_, err = p.get(context.Background())
 			noerr(t, err)
-			getCtx, getCancel := context.WithCancel(context.Background())
+			getCtx, getCancel := context.WithTimeout(context.Background(), 30*time.Second)
 			defer getCancel()
 			go func() {
 				defer close(getDone)
@@ -226,20 +226,23 @@ func TestPool(t *testing.T) {
 						return
 					default:
 						loopCtx, loopCancel := context.WithTimeout(getCtx, 3*time.Second)
-						_, _ = p.get(loopCtx)
+						c, err := p.get(loopCtx)
 						loopCancel()
+						if err == nil {
+							_ = p.put(c)
+						}
 						time.Sleep(time.Microsecond)
 					}
 				}
 			}()
 			go func() {
+				defer close(disconnectDone)
 				_, err := p.get(getCtx)
 				noerr(t, err)
 				ctx, cancel := context.WithTimeout(context.Background(), 3*time.Microsecond)
 				defer cancel()
 				err = p.disconnect(ctx)
 				noerr(t, err)
-				close(disconnectDone)
 			}()
 			<-getDone
 			close(cleanup)