nginx
diff --git a/‎internal/k8s/controller.go‎
Lines changed: 2 additions & 2 deletions b/‎internal/k8s/controller.go‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎internal/k8s/controller_test.go‎
Lines changed: 9 additions & 11 deletions b/‎internal/k8s/controller_test.go‎
Lines changed: 9 additions & 11 deletions
@@ -3659,8 +3659,8 @@ func (lbc *LoadBalancerController) isPodMarkedForDeletion() bool {
 		nl.Debugf(lbc.Logger, "SIGTERM already received, controller is shutting down")
 		return true
 	}
-	podName := os.Getenv("POD_NAME")
-	podNamespace := os.Getenv("POD_NAMESPACE")
+	podName := lbc.metadata.pod.Name
+	podNamespace := lbc.metadata.pod.Namespace
 	pod, err := lbc.client.CoreV1().Pods(podNamespace).Get(context.Background(), podName, meta_v1.GetOptions{})
 	if err == nil && pod.DeletionTimestamp != nil {
 		nl.Debugf(lbc.Logger, "Pod %s/%s is marked for deletion", podNamespace, podName)
 
@@ -4,7 +4,6 @@ import (
 	"context"
 	"errors"
 	"fmt"
-	"os"
 	"reflect"
 	"sort"
 	"strings"
@@ -3682,15 +3681,6 @@ func TestIsPodMarkedForDeletion(t *testing.T) {
 		test := test
 		t.Run(test.name, func(t *testing.T) {
 			client := fake.NewSimpleClientset()
-
-			err := os.Setenv("POD_NAME", test.envPodName)
-			if err != nil {
-				t.Fatalf("Failed to set POD_NAME environment variable: %v", err)
-			}
-			err = os.Setenv("POD_NAMESPACE", test.envPodNamespace)
-			if err != nil {
-				t.Fatalf("Failed to set POD_NAMESPACE environment variable: %v", err)
-			}
 			if test.podExists {
 				pod := &api_v1.Pod{
 					ObjectMeta: meta_v1.ObjectMeta{
@@ -3711,7 +3701,15 @@ func TestIsPodMarkedForDeletion(t *testing.T) {
 			}
 
 			lbc := &LoadBalancerController{
-				client:       client,
+				client: client,
+				metadata: controllerMetadata{
+					pod: &api_v1.Pod{
+						ObjectMeta: meta_v1.ObjectMeta{
+							Name:      test.envPodName,
+							Namespace: test.envPodNamespace,
+						},
+					},
+				},
 				ShuttingDown: test.shutdownFlag,
 				Logger:       logger,
 			}