apply recommendations

Signed-off-by: Benedikt Bongartz <[email protected]>
open-telemetry · Oct 14, 2024 · f9df64a · f9df64a
1 parent 5773a44
commit f9df64a
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 10 deletions.
diff --git a/pkg/sidecar/pod.go b/pkg/sidecar/pod.go
@@ -70,34 +70,34 @@ func add(cfg config.Config, logger logr.Logger, otelcol v1beta1.OpenTelemetryCol
 	return pod, nil
 }
 
+func isOtelColContainer(c corev1.Container) bool { return c.Name == naming.Container() }
+
 // remove the sidecar container from the given pod.
 func remove(pod corev1.Pod) corev1.Pod {
 	if !existsIn(pod) {
 		return pod
 	}
 
-	fn := func(c corev1.Container) bool { return c.Name == naming.Container() }
-	pod.Spec.Containers = slices.DeleteFunc(pod.Spec.Containers, fn)
+	pod.Spec.Containers = slices.DeleteFunc(pod.Spec.Containers, isOtelColContainer)
 
 	if featuregate.EnableNativeSidecarContainers.IsEnabled() {
 		// NOTE: we also remove init containers (native sidecars) since k8s 1.28.
 		// This should have no side effects.
-		pod.Spec.InitContainers = slices.DeleteFunc(pod.Spec.InitContainers, fn)
+		pod.Spec.InitContainers = slices.DeleteFunc(pod.Spec.InitContainers, isOtelColContainer)
 	}
 	return pod
 }
 
 // existsIn checks whether a sidecar container exists in the given pod.
 func existsIn(pod corev1.Pod) bool {
-	fn := func(c corev1.Container) bool { return c.Name == naming.Container() }
-	if slices.ContainsFunc(pod.Spec.Containers, fn) {
+	if slices.ContainsFunc(pod.Spec.Containers, isOtelColContainer) {
 		return true
 	}
 
 	if featuregate.EnableNativeSidecarContainers.IsEnabled() {
 		// NOTE: we also check init containers (native sidecars) since k8s 1.28.
 		// This should have no side effects.
-		if slices.ContainsFunc(pod.Spec.InitContainers, fn) {
+		if slices.ContainsFunc(pod.Spec.InitContainers, isOtelColContainer) {
 			return true
 		}
 	}

diff --git a/pkg/sidecar/pod_test.go b/pkg/sidecar/pod_test.go
@@ -32,7 +32,7 @@ import (
 
 var logger = logf.Log.WithName("unit-tests")
 
-func sidecarFeatureGate(t *testing.T) {
+func enableSidecarFeatureGate(t *testing.T) {
 	originalVal := featuregate.EnableNativeSidecarContainers.IsEnabled()
 	t.Logf("original is: %+v", originalVal)
 	require.NoError(t, colfeaturegate.GlobalRegistry().Set(featuregate.EnableNativeSidecarContainers.ID(), true))
@@ -42,7 +42,7 @@ func sidecarFeatureGate(t *testing.T) {
 }
 
 func TestAddNativeSidecar(t *testing.T) {
-	sidecarFeatureGate(t)
+	enableSidecarFeatureGate(t)
 	// prepare
 	pod := corev1.Pod{
 		Spec: corev1.PodSpec{
@@ -269,7 +269,7 @@ func TestRemoveNonExistingSidecar(t *testing.T) {
 }
 
 func TestExistsIn(t *testing.T) {
-	sidecarFeatureGate(t)
+	enableSidecarFeatureGate(t)
 
 	for _, tt := range []struct {
 		desc     string

diff --git a/tests/e2e-native-sidecar/00-assert.yaml b/tests/e2e-native-sidecar/00-assert.yaml
@@ -10,6 +10,7 @@ spec:
   - name: myapp
   initContainers:
   - name: otc-container
+    restartPolicy: Always
 status:
   containerStatuses:
   - name: myapp

diff --git a/tests/e2e-native-sidecar/00-install.yaml b/tests/e2e-native-sidecar/00-install.yaml
@@ -35,7 +35,7 @@ metadata:
 spec:
   containers:
   - name: myapp
-    image: jaegertracing/vertx-create-span:operator-e2e-tests
+    image: ghcr.io/open-telemetry/opentelemetry-operator/e2e-test-app-python:main
     ports:
       - containerPort: 8080
         protocol: TCP