github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-22 19:31:44 +00:00
Code Issues Projects Releases Wiki Activity

fix: resolve goroutine leak in statefulset integration tests

Browse Source 
Signed-off-by: kerthcet <kerthcet@gmail.com>
This commit is contained in:
kerthcet 2022-05-06 17:51:34 +08:00
parent 835cdd6d81
commit c74aeab5a5
2 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
test/integration/statefulset/statefulset_test.go
View File

@ -124,8 +124,8 @@ func TestSpecReplicasChange(t *testing.T) {
defer closeFn() defer closeFn()
ns := framework.CreateTestingNamespace("test-spec-replicas-change", s, t) ns := framework.CreateTestingNamespace("test-spec-replicas-change", s, t)
defer framework.DeleteTestingNamespace(ns, s, t) defer framework.DeleteTestingNamespace(ns, s, t)
stopCh := runControllerAndInformers(rm, informers) cancel := runControllerAndInformers(rm, informers)
defer close(stopCh) defer cancel()
createHeadlessService(t, c, newHeadlessService(ns.Name)) createHeadlessService(t, c, newHeadlessService(ns.Name))
sts := newSTS("sts", ns.Name, 2) sts := newSTS("sts", ns.Name, 2)
@ -167,8 +167,8 @@ func TestDeletingAndFailedPods(t *testing.T) {
defer closeFn() defer closeFn()
ns := framework.CreateTestingNamespace("test-deleting-and-failed-pods", s, t) ns := framework.CreateTestingNamespace("test-deleting-and-failed-pods", s, t)
defer framework.DeleteTestingNamespace(ns, s, t) defer framework.DeleteTestingNamespace(ns, s, t)
stopCh := runControllerAndInformers(rm, informers) cancel := runControllerAndInformers(rm, informers)
defer close(stopCh) defer cancel()
labelMap := labelMap() labelMap := labelMap()
sts := newSTS("sts", ns.Name, 2) sts := newSTS("sts", ns.Name, 2)
@ -268,8 +268,8 @@ func TestStatefulSetAvailable(t *testing.T) {
defer closeFn() defer closeFn()
ns := framework.CreateTestingNamespace("test-available-pods", s, t) ns := framework.CreateTestingNamespace("test-available-pods", s, t)
defer framework.DeleteTestingNamespace(ns, s, t) defer framework.DeleteTestingNamespace(ns, s, t)
stopCh := runControllerAndInformers(rm, informers) cancel := runControllerAndInformers(rm, informers)
defer close(stopCh) defer cancel()
labelMap := labelMap() labelMap := labelMap()
sts := newSTS("sts", ns.Name, 4) sts := newSTS("sts", ns.Name, 4)

12
test/integration/statefulset/util.go
View File

@ -24,7 +24,7 @@ import (
"time" "time"
appsv1 "k8s.io/api/apps/v1" appsv1 "k8s.io/api/apps/v1"
"k8s.io/api/core/v1" v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource" "k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels" "k8s.io/apimachinery/pkg/labels"
@ -182,11 +182,11 @@ func scSetup(t *testing.T) (*httptest.Server, framework.CloseFunc, *statefulset.
} }
// Run STS controller and informers // Run STS controller and informers
func runControllerAndInformers(sc *statefulset.StatefulSetController, informers informers.SharedInformerFactory) chan struct{} { func runControllerAndInformers(sc *statefulset.StatefulSetController, informers informers.SharedInformerFactory) context.CancelFunc {
stopCh := make(chan struct{}) ctx, cancel := context.WithCancel(context.Background())
informers.Start(stopCh) informers.Start(ctx.Done())
go sc.Run(context.TODO(), 5) go sc.Run(ctx, 5)
return stopCh return cancel
} }
func createHeadlessService(t *testing.T, clientSet clientset.Interface, headlessService *v1.Service) { func createHeadlessService(t *testing.T, clientSet clientset.Interface, headlessService *v1.Service) {