diff --git a/registry/kubernetes/listener_test.go b/registry/kubernetes/listener_test.go index f89154e32e4ef3b6423d6970f9bfc0dd0207d7dd..88e1589020fe8b62a2e2f03690cece081302cc06 100644 --- a/registry/kubernetes/listener_test.go +++ b/registry/kubernetes/listener_test.go @@ -19,6 +19,8 @@ package kubernetes import ( "encoding/json" + "net/http" + _ "net/http/pprof" "os" "strconv" "testing" @@ -238,6 +240,8 @@ func (s *KubernetesRegistryTestSuite) SetupSuite() { t.Fatal(err) } + go http.ListenAndServe(":6061", nil) + } func (s *KubernetesRegistryTestSuite) TestDataChange() { diff --git a/registry/kubernetes/registry_test.go b/registry/kubernetes/registry_test.go index 10eed205266da736f2b549fb97d2f6c3cb24937d..b9f817b8de8c220c166288c09da7116734a0e939 100644 --- a/registry/kubernetes/registry_test.go +++ b/registry/kubernetes/registry_test.go @@ -135,3 +135,12 @@ func (s *KubernetesRegistryTestSuite) TestNewRegistry() { t.Fatal("not in cluster, should be a err") } } + +func (s *KubernetesRegistryTestSuite) TestHandleClientRestart() { + + r := s.initRegistry() + r.WaitGroup().Add(1) + go r.HandleClientRestart() + time.Sleep(timeSecondDuration(1)) + r.CloseAndNilClient() +} diff --git a/remoting/kubernetes/watch_test.go b/remoting/kubernetes/watch_test.go index 7bfc35ecfdfa36fe44898a1e08e07b24de82b5eb..8889103be212381c07ffff3c3d4399f41aeee564 100644 --- a/remoting/kubernetes/watch_test.go +++ b/remoting/kubernetes/watch_test.go @@ -27,7 +27,7 @@ import ( func TestWatchSet(t *testing.T) { - ctx, cancel := context.WithTimeout(context.Background(), time.Second) + ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second) defer cancel() s := newWatcherSet(ctx)