Add lifecycle param to PodOptions struct (#1890)

* Add lifecycle param to PodOptions struct * Linter fix
kanisterio · Feb 8, 2023 · 6950681 · 6950681
1 parent f658f6f
commit 6950681
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/pkg/kube/pod.go b/pkg/kube/pod.go
@@ -71,6 +71,7 @@ type PodOptions struct {
 	RestartPolicy            v1.RestartPolicy
 	OwnerReferences          []metav1.OwnerReference
 	EnvironmentVariables     []v1.EnvVar
+	Lifecycle                *v1.Lifecycle
 }
 
 func GetPodObjectFromPodOptions(cli kubernetes.Interface, opts *PodOptions) (*v1.Pod, error) {
@@ -178,6 +179,10 @@ func GetPodObjectFromPodOptions(cli kubernetes.Interface, opts *PodOptions) (*v1
 		pod.Spec.Containers[0].SecurityContext = opts.ContainerSecurityContext
 	}
 
+	if opts.Lifecycle != nil {
+		pod.Spec.Containers[0].Lifecycle = opts.Lifecycle
+	}
+
 	for key, value := range opts.Labels {
 		pod.ObjectMeta.Labels[key] = value
 	}

diff --git a/pkg/kube/pod_test.go b/pkg/kube/pod_test.go
@@ -852,3 +852,25 @@ func (s *PodSuite) TestSetPodSecurityContextOverridesPodOverride(c *C) {
 	c.Assert(*pod.Spec.SecurityContext.RunAsUser, DeepEquals, uidAndGidExpected)
 	c.Assert(*pod.Spec.SecurityContext.RunAsGroup, DeepEquals, uidAndGidExpected)
 }
+
+func (s *PodSuite) TestSetLifecycleHook(c *C) {
+	lch := &v1.Lifecycle{
+		PostStart: &v1.LifecycleHandler{
+			Exec: &v1.ExecAction{
+				Command: []string{"/bin/bash", "-c", "echo 1"},
+			},
+		},
+	}
+
+	po := &PodOptions{
+		Namespace:    s.namespace,
+		GenerateName: "test-",
+		Image:        consts.LatestKanisterToolsImage,
+		Command:      []string{"sh", "-c", "tail -f /dev/null"},
+		Lifecycle:    lch,
+	}
+
+	pod, err := CreatePod(context.Background(), s.cli, po)
+	c.Assert(err, IsNil)
+	c.Assert(pod.Spec.Containers[0].Lifecycle, DeepEquals, lch)
+}