浏览代码

integration: cancel Watch when TestV3WatchWithPrevKV exits

Missing ctx cancel was causing goroutine leaks for the proxy tests.
Anthony Romano 9 年之前
父节点
当前提交
76bb33781f
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      integration/v3_watch_test.go

+ 5 - 1
integration/v3_watch_test.go

@@ -1128,9 +1128,13 @@ func TestV3WatchWithFilter(t *testing.T) {
 }
 
 func TestV3WatchWithPrevKV(t *testing.T) {
+	defer testutil.AfterTest(t)
 	clus := NewClusterV3(t, &ClusterConfig{Size: 1})
 	defer clus.Terminate(t)
 
+	wctx, wcancel := context.WithCancel(context.Background())
+	defer wcancel()
+
 	tests := []struct {
 		key  string
 		end  string
@@ -1150,7 +1154,7 @@ func TestV3WatchWithPrevKV(t *testing.T) {
 			t.Fatal(err)
 		}
 
-		ws, werr := toGRPC(clus.RandClient()).Watch.Watch(context.TODO())
+		ws, werr := toGRPC(clus.RandClient()).Watch.Watch(wctx)
 		if werr != nil {
 			t.Fatal(werr)
 		}