fix test

Author: cjc7373

pkg/controller/lifecycle/lifecycle_test.go

@@ -171,7 +171,7 @@ var _ = Describe("lifecycle", func() {
 
 		It("pods", func() {
 			pod := pods[0]
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pod, pods)
 			Expect(err).Should(BeNil())
 
 			Expect(lifecycle).ShouldNot(BeNil())
@@ -188,7 +188,7 @@ var _ = Describe("lifecycle", func() {
 
 	Context("call action", func() {
 		It("not defined", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -198,7 +198,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("action request", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -228,7 +228,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("succeed", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -243,7 +243,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("succeed and stdout", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -261,7 +261,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("fail - error code", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -330,7 +330,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("fail - error msg", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -350,7 +350,7 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("parameters", func() {
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -394,7 +394,7 @@ var _ = Describe("lifecycle", func() {
 			key := "TEMPLATE_VAR1"
 			val := "template-vars1"
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, map[string]string{key: val}, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, map[string]string{key: val}, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -418,7 +418,7 @@ var _ = Describe("lifecycle", func() {
 			clusterReady := appsv1.ClusterReadyPreConditionType
 			lifecycleActions.PostProvision.PreCondition = &clusterReady
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -451,7 +451,7 @@ var _ = Describe("lifecycle", func() {
 			clusterReady := appsv1.ClusterReadyPreConditionType
 			lifecycleActions.PostProvision.PreCondition = &clusterReady
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -516,7 +516,7 @@ var _ = Describe("lifecycle", func() {
 				},
 			}
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -526,7 +526,34 @@ var _ = Describe("lifecycle", func() {
 		})
 
 		It("pod selector - all", func() {
-			// TODO: impl
+			lifecycleActions.PostProvision.Exec.TargetPodSelector = appsv1.AllReplicas
+			pods = []*corev1.Pod{
+				{
+					ObjectMeta: metav1.ObjectMeta{
+						Namespace: namespace,
+						Name:      "pod-0",
+					},
+				},
+				{
+					ObjectMeta: metav1.ObjectMeta{
+						Namespace: namespace,
+						Name:      "pod-1",
+					},
+				},
+			}
+
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
+			Expect(err).Should(BeNil())
+			Expect(lifecycle).ShouldNot(BeNil())
+
+			mockKBAgentClient(func(recorder *kbacli.MockClientMockRecorder) {
+				recorder.Action(gomock.Any(), gomock.Any()).DoAndReturn(func(ctx context.Context, req proto.ActionRequest) (proto.ActionResponse, error) {
+					return proto.ActionResponse{}, nil
+				}).Times(2)
+			})
+
+			err = lifecycle.PostProvision(ctx, k8sClient, nil)
+			Expect(err).Should(BeNil())
 		})
 
 		It("pod selector - role", func() {
@@ -577,7 +604,7 @@ var _ = Describe("lifecycle", func() {
 				},
 			}
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())
 
@@ -610,7 +637,7 @@ var _ = Describe("lifecycle", func() {
 				},
 			}
 
-			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, nil, pods)
+			lifecycle, err := New(namespace, clusterName, compName, lifecycleActions, nil, pods[0], pods)
 			Expect(err).Should(BeNil())
 			Expect(lifecycle).ShouldNot(BeNil())