updated tests

linkedin · May 15, 2024 · cc60a4c · cc60a4c
1 parent b00fef7
commit cc60a4c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/...roller/src/main/java/com/linkedin/venice/controller/StoreBackupVersionCleanupService.java b/...roller/src/main/java/com/linkedin/venice/controller/StoreBackupVersionCleanupService.java
@@ -260,11 +260,17 @@ protected boolean cleanupBackupVersion(Store store, String clusterName) {
       stats.recordBackupVersionMismatch();
       return false;
     }
-    int repushSourceVersion = store.getVersion(currentVersion).get().getRepushSourceVersion();
+    int repushSourceVersion = NON_EXISTING_VERSION;
+    for (Version version: versions) {
+      if (version.getRepushSourceVersion() > NON_EXISTING_VERSION) {
+        repushSourceVersion = version.getRepushSourceVersion();
+      }
+    }
+    int finalRepushSourceVersion = repushSourceVersion;
     List<Version> readyToBeRemovedVersions = versions.stream()
         .filter(
-            v -> repushSourceVersion > NON_EXISTING_VERSION
-                ? v.getNumber() == repushSourceVersion
+            v -> finalRepushSourceVersion > NON_EXISTING_VERSION
+                ? v.getNumber() == finalRepushSourceVersion
                 : v.getNumber() < currentVersion)
         .collect(Collectors.toList());