diff --git a/test/e2e/traffic_manager_test.go b/test/e2e/traffic_manager_test.go index 9b3fe40c..bdae9bef 100644 --- a/test/e2e/traffic_manager_test.go +++ b/test/e2e/traffic_manager_test.go @@ -136,9 +136,13 @@ var _ = Describe("Test exporting service via Azure traffic manager", func() { By("Validating the trafficManagerBackend status") wantEndpoints := []fleetnetv1alpha1.TrafficManagerEndpointStatus{ { - Weight: ptr.To(int64(100)), - Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[0]), clusterLocation)), - Cluster: &fleetnetv1alpha1.ClusterStatus{Cluster: memberClusters[0].Name()}, + Weight: ptr.To(int64(100)), + Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[0]), clusterLocation)), + From: &fleetnetv1alpha1.FromCluster{ + ClusterStatus: fleetnetv1alpha1.ClusterStatus{ + Cluster: memberClusterNames[0], + }, + }, }, } status = validator.ValidateTrafficManagerBackendIfAcceptedAndIgnoringEndpointName(ctx, hubClient, name, false, wantEndpoints) @@ -156,14 +160,22 @@ var _ = Describe("Test exporting service via Azure traffic manager", func() { By("Validating the trafficManagerBackend status") wantEndpoints = []fleetnetv1alpha1.TrafficManagerEndpointStatus{ { - Weight: ptr.To(int64(50)), - Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[0]), clusterLocation)), - Cluster: &fleetnetv1alpha1.ClusterStatus{Cluster: memberClusters[0].Name()}, + Weight: ptr.To(int64(50)), + Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[0]), clusterLocation)), + From: &fleetnetv1alpha1.FromCluster{ + ClusterStatus: fleetnetv1alpha1.ClusterStatus{ + Cluster: memberClusterNames[0], + }, + }, }, { - Weight: ptr.To(int64(50)), - Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[1]), clusterLocation)), - Cluster: &fleetnetv1alpha1.ClusterStatus{Cluster: memberClusters[1].Name()}, + Weight: ptr.To(int64(50)), + Target: ptr.To(fmt.Sprintf(azureDNSFormat, wm.BuildServiceDNSLabelName(memberClusters[1]), clusterLocation)), + From: &fleetnetv1alpha1.FromCluster{ + ClusterStatus: fleetnetv1alpha1.ClusterStatus{ + Cluster: memberClusterNames[0], + }, + }, }, } status = validator.ValidateTrafficManagerBackendIfAcceptedAndIgnoringEndpointName(ctx, hubClient, name, true, wantEndpoints)