Optimised fleet update function of e2e test

Signed-off-by: LiZhenCheng9527 <[email protected]>
LiZhenCheng9527 · Feb 20, 2024 · e86b321 · e86b321
1 parent 386cab0
commit e86b321
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/e2e/resources/fleet.go b/e2e/resources/fleet.go
@@ -60,7 +60,13 @@ func CreateFleet(client kurator.Interface, fleet *fleetv1a1.Fleet) error {
 
 // UpdateAttachedCluster update AttachedCluster
 func UpdateFleet(client kurator.Interface, fleet *fleetv1a1.Fleet) error {
-	_, err := client.FleetV1alpha1().Fleets(fleet.Namespace).Update(context.TODO(), fleet, metav1.UpdateOptions{})
+	fleetOnCluster, fleetGetErr := client.FleetV1alpha1().Fleets(fleet.Namespace).Get(context.TODO(), fleet.Name, metav1.GetOptions{})
+	if fleetGetErr != nil {
+		return fleetGetErr
+	}
+	DCfleet := fleetOnCluster.DeepCopy()
+	DCfleet.Spec = fleet.Spec
+	_, err := client.FleetV1alpha1().Fleets(fleet.Namespace).Update(context.TODO(), DCfleet, metav1.UpdateOptions{})
 	if err != nil {
 		return err
 	}