Added function for startup probe (#397)

Author: chatton

pkg/kube/container/container_test.go

@@ -37,6 +37,13 @@ func TestContainer(t *testing.T) {
 			probes.WithFailureThreshold(15),
 			probes.WithPeriodSeconds(10),
 		)),
+		WithStartupProbe(
+			probes.Apply(
+				probes.WithExecCommand([]string{"startup-exec"}),
+				probes.WithFailureThreshold(20),
+				probes.WithPeriodSeconds(30),
+			),
+		),
 		WithResourceRequirements(resourcerequirements.Defaults()),
 		WithCommand([]string{"container-cmd"}),
 		WithEnvs(
@@ -71,6 +78,11 @@ func TestContainer(t *testing.T) {
 	assert.Equal(t, int32(10), liveNessProbe.PeriodSeconds)
 	assert.Equal(t, "liveness-exec", liveNessProbe.Exec.Command[0])
 
+	startupProbe := c.StartupProbe
+	assert.Equal(t, int32(20), startupProbe.FailureThreshold)
+	assert.Equal(t, int32(30), startupProbe.PeriodSeconds)
+	assert.Equal(t, "startup-exec", startupProbe.Exec.Command[0])
+
 	assert.Equal(t, c.Resources, resourcerequirements.Defaults())
 
 	assert.Len(t, c.Command, 1)

pkg/kube/container/containers.go

@@ -75,12 +75,22 @@ func WithReadinessProbe(probeFunc func(*corev1.Probe)) Modification {
 }
 
 // WithLivenessProbe modifies the container's Liveness Probe
-func WithLivenessProbe(readinessProbeFunc func(*corev1.Probe)) Modification {
+func WithLivenessProbe(livenessProbeFunc func(*corev1.Probe)) Modification {
 	return func(container *corev1.Container) {
 		if container.LivenessProbe == nil {
 			container.LivenessProbe = &corev1.Probe{}
 		}
-		readinessProbeFunc(container.LivenessProbe)
+		livenessProbeFunc(container.LivenessProbe)
+	}
+}
+
+// WithStartupProbe modifies the container's Startup Probe
+func WithStartupProbe(startupProbeFunc func(*corev1.Probe)) Modification {
+	return func(container *corev1.Container) {
+		if container.StartupProbe == nil {
+			container.StartupProbe = &corev1.Probe{}
+		}
+		startupProbeFunc(container.StartupProbe)
 	}
 }