add unit test

openyurtio · Mar 31, 2023 · 090631e · 090631e
1 parent d9dcae9
commit 090631e
Showing 1 changed file with 26 additions and 0 deletions.
diff --git a/pkg/controller/poolcoordinator/cert/util_test.go b/pkg/controller/poolcoordinator/cert/util_test.go
@@ -25,6 +25,8 @@ import (
 	kerrors "k8s.io/apimachinery/pkg/api/errors"
 	v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 	"k8s.io/client-go/kubernetes/fake"
+
+	"github.com/openyurtio/openyurt/pkg/util/ip"
 )
 
 const (
@@ -111,3 +113,27 @@ func TestGetAPIServerSVCURL(t *testing.T) {
 	assert.Equal(t, nil, err)
 	assert.Equal(t, "https://xxxx:644", url)
 }
+
+func TestWaitUntilSVCReady(t *testing.T) {
+	stop := make(chan struct{})
+	defer close(stop)
+
+	normalClient := fake.NewSimpleClientset(&corev1.Service{
+		ObjectMeta: v1.ObjectMeta{
+			Namespace: PoolcoordinatorNS,
+			Name:      PoolcoordinatorAPIServerSVC,
+		},
+		Spec: corev1.ServiceSpec{
+			ClusterIP: "xxxx",
+			Ports: []corev1.ServicePort{
+				{
+					Port: 644,
+				},
+			},
+		},
+	})
+	ips, _, err := waitUntilSVCReady(normalClient, PoolcoordinatorAPIServerSVC, stop)
+	assert.Equal(t, nil, err)
+	expectIPS := ip.ParseIPList([]string{"xxxx"})
+	assert.Equal(t, expectIPS, ips)
+}