Fix test on MaxSize

hapifhir · Jun 18, 2024 · 3a7507c · 3a7507c
1 parent 4fa581d
commit 3a7507c
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/.../src/test/java/org/hl7/fhir/validation/cli/services/MaxSizeSessionCacheDecoratorTest.java b/.../src/test/java/org/hl7/fhir/validation/cli/services/MaxSizeSessionCacheDecoratorTest.java
@@ -40,15 +40,18 @@ public void trivialCase() {
 
     Assertions.assertEquals(3, maxSizeSessionCacheDecorator.getSessionIds().size());
 
-    List<ValidationEngine> newEngines = getMockedEngines(2);
+    LinkedHashMap<String, ValidationEngine> newEngines = addMockedEngines(maxSizeSessionCacheDecorator, 2);
 
-    for (ValidationEngine engine : newEngines) {
-      maxSizeSessionCacheDecorator.cacheSession(engine);
-    }
 
     Assertions.assertEquals(4, maxSizeSessionCacheDecorator.getSessionIds().size());
+    Assertions.assertTrue(maxSizeSessionCacheDecorator.getSessionIds().contains(getKeyByIndex(initialEngines, 1)));
+    Assertions.assertTrue(maxSizeSessionCacheDecorator.getSessionIds().contains(getKeyByIndex(initialEngines, 2)));
+    Assertions.assertTrue(maxSizeSessionCacheDecorator.getSessionIds().contains(getKeyByIndex(newEngines, 0)));
+    Assertions.assertTrue(maxSizeSessionCacheDecorator.getSessionIds().contains(getKeyByIndex(newEngines, 1)));
 
-    Assertions.assertTrue(maxSizeSessionCacheDecorator.getSessionIds().contains()
   }
 
+  private String getKeyByIndex(LinkedHashMap<String, ValidationEngine> engineMap, int index) {
+    return (String) engineMap.keySet().toArray()[index];
+  }
 }