Extract test occurrences faster.

usethesource · Jul 2, 2024 · c51164d · c51164d
1 parent edec500
commit c51164d
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/rascal-lsp/src/main/rascal/lang/rascal/tests/Rename.rsc b/rascal-lsp/src/main/rascal/lang/rascal/tests/Rename.rsc
@@ -515,13 +515,8 @@ set[int] extractRenameOccurrences(loc moduleFileName, list[DocumentEdit] edits,
     }
 
     if ([changed(_, replaces)] := edits) {
-        idx = {};
-        for (replace(l, _) <- replaces, i := indexOf(oldNameOccurrences, l)) {
-            if (i == -1) throw "Cannot find <l> in <oldNameOccurrences>";
-            idx += i;
-        }
-
-        return idx;
+        repls = {l | replace(l, _) <- replaces};
+        return {i | i <- [0..size(oldNameOccurrences)], oldNameOccurrences[i] in repls};;
     } else {
         throw "Unexpected changes: <edits>";
     }