Add more rich unit-test

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() {

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()
+}

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)