fix testcase

Signed-off-by: Megrez Lu <[email protected]>
openkruise · Jul 11, 2024 · 008562a · 008562a
1 parent 11424ea
commit 008562a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 10 deletions.
diff --git a/...nfiguration/networking.istio.io/VirtualService/testdata/traffic_routing_with_matches.yaml b/...nfiguration/networking.istio.io/VirtualService/testdata/traffic_routing_with_matches.yaml
@@ -66,6 +66,10 @@ expected:
         - headers:
             user-agent:
               exact: pc
+        headers:
+          request:
+            set:
+              header-foo: bar
         route:
         - destination:
             host: svc-demo

diff --git a/pkg/trafficrouting/network/customNetworkProvider/custom_network_provider_test.go b/pkg/trafficrouting/network/customNetworkProvider/custom_network_provider_test.go
@@ -710,11 +710,12 @@ func TestLuaScript(t *testing.T) {
 								Annotations: testCase.Original.GetAnnotations(),
 								Spec:        testCase.Original.Object["spec"],
 							},
-							Matches:       step.TrafficRoutingStrategy.Matches,
-							CanaryWeight:  *weight,
-							StableWeight:  100 - *weight,
-							CanaryService: canaryService,
-							StableService: stableService,
+							Matches:               step.TrafficRoutingStrategy.Matches,
+							CanaryWeight:          *weight,
+							StableWeight:          100 - *weight,
+							CanaryService:         canaryService,
+							StableService:         stableService,
+							RequestHeaderModifier: step.TrafficRoutingStrategy.RequestHeaderModifier,
 						}
 						nSpec, err := executeLua(data, script)
 						if err != nil {
@@ -750,11 +751,12 @@ func TestLuaScript(t *testing.T) {
 							Annotations: testCase.Original.GetAnnotations(),
 							Spec:        testCase.Original.Object["spec"],
 						},
-						Matches:       matches,
-						CanaryWeight:  *weight,
-						StableWeight:  100 - *weight,
-						CanaryService: canaryService,
-						StableService: stableService,
+						Matches:               matches,
+						CanaryWeight:          *weight,
+						StableWeight:          100 - *weight,
+						CanaryService:         canaryService,
+						StableService:         stableService,
+						RequestHeaderModifier: trafficRouting.Spec.Strategy.RequestHeaderModifier,
 					}
 					nSpec, err := executeLua(data, script)
 					if err != nil {