feat(webhook): fix tensor fusion pod mutation handler tests

Author: 0x5457

.github/workflows/test-e2e.yml

@@ -1,8 +1,9 @@
 name: E2E Tests
 
 on:
-  push:
-  pull_request:
+  workflow_dispatch:
+  # push:
+  # pull_request:
 
 jobs:
   test-e2e:

config/crd/bases/tensor-fusion.ai_tensorfusionconnections.yaml

@@ -91,7 +91,7 @@ spec:
             properties:
               connectionURL:
                 type: string
-              node:
+              gpu:
                 type: string
               phase:
                 type: string

internal/webhook/v1/pod_webhook.go

@@ -67,6 +67,10 @@ func (m *TensorFusionPodMutator) Handle(ctx context.Context, req admission.Reque
 	log.Info("Mutating pod", "name", pod.Name, "namespace", pod.Namespace)
 
 	reqs := parseTFReq(pod)
+	if len(reqs) == 0 {
+		return admission.Allowed("no tensor fusion requirements found")
+	}
+
 	// 1. Inject initContainer and env variables
 	patches, err := m.patchTFClient(pod, reqs)
 	if err != nil {
@@ -108,9 +112,14 @@ func parseTFReq(pod *corev1.Pod) []TFReq {
 	for _, container := range pod.Spec.Containers {
 		containerName := container.Name
 
-		// Check if tensor fusion is enabled for this container
-		enableKey := fmt.Sprintf(constants.EnableContainerAnnotationFormat, containerName)
-		if enableStr, ok := pod.Annotations[enableKey]; !ok || enableStr != "true" {
+		// Check if TF requirements exist for this container
+		tflopsKey := fmt.Sprintf(constants.TFLOPSContainerAnnotationFormat, containerName)
+		vramKey := fmt.Sprintf(constants.VRAMContainerAnnotationFormat, containerName)
+
+		tflopsStr, hasTflops := pod.Annotations[tflopsKey]
+		vramStr, hasVram := pod.Annotations[vramKey]
+
+		if !hasTflops && !hasVram {
 			continue
 		}
 
@@ -119,17 +128,15 @@ func parseTFReq(pod *corev1.Pod) []TFReq {
 		}
 
 		// Parse TFLOPS requirement
-		tflopsKey := fmt.Sprintf(constants.TFLOPSContainerAnnotationFormat, containerName)
-		if tflopsStr, ok := pod.Annotations[tflopsKey]; ok {
+		if hasTflops {
 			tflops, err := resource.ParseQuantity(tflopsStr)
 			if err == nil {
 				req.Tflops = tflops
 			}
 		}
 
 		// Parse VRAM requirement
-		vramKey := fmt.Sprintf(constants.VRAMContainerAnnotationFormat, containerName)
-		if vramStr, ok := pod.Annotations[vramKey]; ok {
+		if hasVram {
 			vram, err := resource.ParseQuantity(vramStr)
 			if err == nil {
 				req.Vram = vram

internal/webhook/v1/pod_webhook_test.go

@@ -23,6 +23,7 @@ import (
 
 	tfv1 "github.com/NexusGPU/tensor-fusion-operator/api/v1"
 	"github.com/NexusGPU/tensor-fusion-operator/internal/config"
+	"github.com/NexusGPU/tensor-fusion-operator/internal/constants"
 	. "github.com/onsi/ginkgo/v2"
 	. "github.com/onsi/gomega"
 	admissionv1 "k8s.io/api/admission/v1"
@@ -67,8 +68,8 @@ var _ = Describe("TensorFusionPodMutator", func() {
 					Name:      "test-pod",
 					Namespace: "default",
 					Annotations: map[string]string{
-						"tf.nexusgpu.com/tflops": "100",
-						"tf.nexusgpu.com/vram":   "16Gi",
+						constants.TensorFusionDomain + "/tflops-main": "100",
+						constants.TensorFusionDomain + "/vram-main":   "16Gi",
 					},
 				},
 				Spec: corev1.PodSpec{
@@ -158,8 +159,8 @@ var _ = Describe("TensorFusionPodMutator", func() {
 			pod := &corev1.Pod{
 				ObjectMeta: metav1.ObjectMeta{
 					Annotations: map[string]string{
-						"tf.nexusgpu.com/tflops": "100",
-						"tf.nexusgpu.com/vram":   "16Gi",
+						constants.TensorFusionDomain + "/tflops-test-container": "100",
+						constants.TensorFusionDomain + "/vram-test-container":   "16Gi",
 					},
 				},
 				Spec: corev1.PodSpec{