added for loop in testToJsonCustom (stdlibs)

wala · Mar 5, 2024 · efad143 · efad143
1 parent f777f66
commit efad143
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/core/src/test/java/com/ibm/wala/core/tests/cha/AnalysisScopeTest.java b/core/src/test/java/com/ibm/wala/core/tests/cha/AnalysisScopeTest.java
@@ -121,9 +121,11 @@ public void testToJsonCustom() throws IOException {
         new HashSet<>(List.of("Primordial", "Extension", "Application", "Synthetic"));
     assertEquals(loaders.keySet(), loaderKeys);
     assertEquals(stdlibs.length, loaders.get("Primordial").size());
-    assertThat(loaders.get("Primordial"), hasItem("JarFileModule:" + stdlibs[0]));
     assertEquals(stdlibs.length, loaders.get("Application").size());
-    assertThat(loaders.get("Application"), hasItem("JarFileModule:" + stdlibs[0]));
+    for (int i = 0; i < stdlibs.length; i++) {
+      assertThat(loaders.get("Primordial"), hasItem("JarFileModule:" + stdlibs[i]));
+      assertThat(loaders.get("Application"), hasItem("JarFileModule:" + stdlibs[i]));
+    }
     assertEquals(0, loaders.get("Extension").size());
     assertEquals(0, loaders.get("Synthetic").size());
   }