fix: add test case for pod controller (#306)

Author: Code2Life

internal/constants/constants.go

@@ -49,7 +49,8 @@ const (
 
 	LastSyncTimeAnnotationKey = Domain + "/last-sync"
 	WorkloadKey               = Domain + "/workload"
-	GpuPoolKey                = Domain + "/gpupool"
+
+	GpuPoolKey = Domain + "/gpupool"
 
 	// Annotation key constants
 	GpuCountAnnotation             = Domain + "/gpu-count"
@@ -70,6 +71,8 @@ const (
 	GPUDeviceIDsAnnotation            = Domain + "/gpu-ids"
 	SetPendingOwnedWorkloadAnnotation = Domain + "/pending-owned-workload"
 	PricingAnnotation                 = Domain + "/hourly-pricing"
+	// In remote vGPU mode, selected workload is set by user with /workload annotation or generated by system
+	SelectedWorkloadAnnotation = Domain + "/selected-workload"
 
 	WorkloadModeAnnotation = Domain + "/workload-mode"
 	WorkloadModeDynamic    = "dynamic"

internal/controller/pod_controller.go

@@ -172,7 +172,7 @@ func (r *PodReconciler) setPendingOwnedWorkload(ctx context.Context, pod *corev1
 }
 
 func buildTensorFusionConnectionObj(pod *corev1.Pod) *tfv1.TensorFusionConnection {
-	workloadName, ok := pod.Labels[constants.WorkloadKey]
+	workloadName, ok := pod.Annotations[constants.SelectedWorkloadAnnotation]
 	if !ok {
 		return nil
 	}