fix unit tests

loft-sh · Feb 13, 2025 · 48e8243 · 48e8243
1 parent 1e21ada
commit 48e8243
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/pkg/controllers/resources/nodes/syncer.go b/pkg/controllers/resources/nodes/syncer.go
@@ -307,6 +307,15 @@ func (s *nodeSyncer) Sync(ctx *synccontext.SyncContext, event *synccontext.SyncE
 	}
 
 	s.translateUpdateBackwards(event.Host, event.Virtual)
+
+	// Set the marker of managed-by vcluster so that
+	// we skip deleting the nodes which are not managed
+	// by vcluster in `SyncToHost` function
+	if len(event.Virtual.Labels) == 0 {
+		event.Virtual.Labels = map[string]string{}
+	}
+	event.Virtual.Labels[translate.MarkerLabel] = translate.VClusterName
+
 	return ctrl.Result{}, nil
 }
 

diff --git a/pkg/controllers/resources/nodes/translate.go b/pkg/controllers/resources/nodes/translate.go
@@ -111,11 +111,6 @@ func (s *nodeSyncer) translateUpdateBackwards(pNode *corev1.Node, vNode *corev1.
 		annotations["cluster-autoscaler.kubernetes.io/scale-down-disabled"] = "true"
 	}
 
-	// Set the marker of managed-by vcluster so that
-	// we skip deleting the nodes which are not managed
-	// by vcluster in `SyncToHost` function
-	labels[translate.MarkerLabel] = translate.VClusterName
-
 	// set annotations, spec & labels
 	vNode.Spec = *translatedSpec
 	vNode.Annotations = annotations