Replace used-context cache contents instead of upserting

Author: skrcode

pom.xml

@@ -29,7 +29,7 @@
 
     <properties>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
-        <revision>0.3.21</revision>
+        <revision>0.3.22</revision>
         <maven.compiler.release>17</maven.compiler.release>
         <picocli.version>4.7.7</picocli.version>
         <javaparser.version>3.27.1</javaparser.version>

src/main/java/com/jaipilot/cli/JaiPilotVersionProvider.java

@@ -13,7 +13,7 @@ public static String resolveVersion() {
         Package commandPackage = JaiPilotCli.class.getPackage();
         String version = commandPackage == null ? null : commandPackage.getImplementationVersion();
         if (version == null || version.isBlank()) {
-            version = "0.3.21";
+            version = "0.3.22";
         }
         return version;
     }

src/main/java/com/jaipilot/cli/service/UsedContextClassPathCache.java

@@ -34,7 +34,7 @@ public List<String> read(Path absoluteClassPath) {
     }
 
     public void write(Path absoluteClassPath, List<String> usedContextClassPaths) {
-        Map<String, List<String>> entries = readAll();
+        Map<String, List<String>> entries = new LinkedHashMap<>();
         entries.put(cacheKey(absoluteClassPath), normalizeValues(usedContextClassPaths));
         writeAll(entries);
     }

src/test/java/com/jaipilot/cli/service/UsedContextClassPathCacheTest.java

@@ -21,4 +21,17 @@ void writeOverwritesExistingEntryWithoutMerging() {
 
         assertEquals(List.of("com/example/Pet.java"), cache.read(classPath));
     }
+
+    @Test
+    void writeReplacesCacheAcrossDifferentClassPaths() {
+        UsedContextClassPathCache cache = new UsedContextClassPathCache(tempDir.resolve("used-context-cache.json"));
+        Path firstClassPath = tempDir.resolve("src/main/java/com/example/CrashController.java");
+        Path secondClassPath = tempDir.resolve("src/main/java/com/example/PetController.java");
+
+        cache.write(firstClassPath, List.of("com/example/First.java"));
+        cache.write(secondClassPath, List.of("com/example/Second.java"));
+
+        assertEquals(List.of(), cache.read(firstClassPath));
+        assertEquals(List.of("com/example/Second.java"), cache.read(secondClassPath));
+    }
 }