diff --git a/test/e2e/network/BUILD b/test/e2e/network/BUILD index 30f16ff5a3e..981e4425e67 100644 --- a/test/e2e/network/BUILD +++ b/test/e2e/network/BUILD @@ -54,7 +54,6 @@ go_library( "//staging/src/k8s.io/apimachinery/pkg/watch:go_default_library", "//staging/src/k8s.io/apiserver/pkg/authentication/serviceaccount:go_default_library", "//staging/src/k8s.io/client-go/kubernetes:go_default_library", - "//staging/src/k8s.io/client-go/rest:go_default_library", "//staging/src/k8s.io/client-go/tools/cache:go_default_library", "//staging/src/k8s.io/client-go/util/flowcontrol:go_default_library", "//staging/src/k8s.io/client-go/util/workqueue:go_default_library", diff --git a/test/e2e/network/service_latency.go b/test/e2e/network/service_latency.go index f4fa958aab3..3d73d32d932 100644 --- a/test/e2e/network/service_latency.go +++ b/test/e2e/network/service_latency.go @@ -27,7 +27,7 @@ import ( "k8s.io/apimachinery/pkg/runtime" "k8s.io/apimachinery/pkg/util/sets" "k8s.io/apimachinery/pkg/watch" - restclient "k8s.io/client-go/rest" + "k8s.io/client-go/kubernetes" "k8s.io/client-go/tools/cache" "k8s.io/client-go/util/flowcontrol" "k8s.io/kubernetes/test/e2e/framework" @@ -77,9 +77,12 @@ var _ = SIGDescribe("Service endpoints latency", func() { ) // Turn off rate limiting--it interferes with our measurements. - oldThrottle := f.ClientSet.CoreV1().RESTClient().GetRateLimiter() - f.ClientSet.CoreV1().RESTClient().(*restclient.RESTClient).Throttle = flowcontrol.NewFakeAlwaysRateLimiter() - defer func() { f.ClientSet.CoreV1().RESTClient().(*restclient.RESTClient).Throttle = oldThrottle }() + cfg, err := framework.LoadConfig() + if err != nil { + framework.Failf("Unable to load config: %v", err) + } + cfg.RateLimiter = flowcontrol.NewFakeAlwaysRateLimiter() + f.ClientSet = kubernetes.NewForConfigOrDie(cfg) failing := sets.NewString() d, err := runServiceLatencies(f, parallelTrials, totalTrials, acceptableFailureRatio)