reduce number of loadgens

smartcontractkit · Mar 1, 2024 · 1e8563f · 1e8563f
1 parent 92cd617
commit 1e8563f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/integration-tests/ccip-tests/load/ccip_multicall_loadgen.go b/integration-tests/ccip-tests/load/ccip_multicall_loadgen.go
@@ -59,7 +59,7 @@ func NewMultiCallLoadGenerator(testCfg *testsetups.CCIPTestConfig, lanes []*acti
 			return nil, fmt.Errorf("all lanes should be from same network; expected %s, got %s", source, lanes[i].SourceChain.GetChainID())
 		}
 		if lanes[i].SrcNetworkLaneCfg.Multicall != multiCall {
-			return nil, fmt.Errorf("multicall address should be same for all lanes")
+			lanes[i].SrcNetworkLaneCfg.Multicall = multiCall
 		}
 	}
 	client := lanes[0].SourceChain

diff --git a/integration-tests/ccip-tests/load/helper.go b/integration-tests/ccip-tests/load/helper.go
@@ -217,13 +217,15 @@ func (l *LoadArgs) TriggerLoadBySource() {
 
 	var laneBySource = make(map[string][]*actions.CCIPLane)
 	for _, lane := range l.TestSetupArgs.Lanes {
-		laneBySource[lane.ForwardLane.SourceNetworkName] = append(laneBySource[lane.ForwardLane.SourceNetworkName], lane.ForwardLane)
+		chainId := lane.ForwardLane.SourceChain.GetChainID().String()
+		laneBySource[chainId] = append(laneBySource[chainId], lane.ForwardLane)
 		if lane.ReverseLane != nil {
-			laneBySource[lane.ReverseLane.SourceNetworkName] = append(laneBySource[lane.ReverseLane.SourceNetworkName], lane.ReverseLane)
+			chainId := lane.ReverseLane.SourceChain.GetChainID().String()
+			laneBySource[chainId] = append(laneBySource[chainId], lane.ReverseLane)
 		}
 	}
-	for source, lanes := range laneBySource {
-		source := source
+	for _, lanes := range laneBySource {
+		source := lanes[0].SourceNetworkName
 		lanes := lanes
 		l.LoadStarterWg.Add(1)
 		go func() {