PR(FIX): Refactor AddPolicy

shahzadlone · Oct 20, 2024 · 4bbc532 · 4bbc532
1 parent cae50fd
commit 4bbc532
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/integration/acp.go b/tests/integration/acp.go
@@ -112,8 +112,10 @@ func addPolicyACP(
 		require.Fail(s.t, "Expected error should not have an expected policyID with it.", s.testCase.Description)
 	}
 
-	for i, node := range getNodes(action.NodeID, s.nodes) {
-		identity := getIdentity(s, i, action.Identity)
+	nodeIDs, nodes := getNodesWithIDs(action.NodeID, s.nodes)
+	for index, node := range nodes {
+		nodeID := nodeIDs[index]
+		identity := getIdentity(s, nodeID, action.Identity)
 		ctx := db.SetContextIdentity(s.ctx, identity)
 		policyResult, err := node.AddPolicy(ctx, action.Policy)