[HUDI-4877] Fix org.apache.hudi.index.bucket.TestHoodieSimpleBucketIn…

…dex#testTagLocation not work correct issue (apache#6717) Co-authored-by: xiaoxingstack <xiaoxingstack@didiglobal.com>
TengHuo · Oct 7, 2022 · 684a339 · 684a339
1 parent b18e2f0
commit 684a339
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...-spark-client/src/test/java/org/apache/hudi/index/bucket/TestHoodieSimpleBucketIndex.java b/...-spark-client/src/test/java/org/apache/hudi/index/bucket/TestHoodieSimpleBucketIndex.java
@@ -139,7 +139,9 @@ public void testTagLocation(boolean isInsert) throws Exception {
         .filter(r -> BucketIdentifier.bucketIdFromFileId(r.getCurrentLocation().getFileId())
             != getRecordBucketId(r)).findAny().isPresent());
     assertTrue(taggedRecordRDD.collectAsList().stream().filter(r -> r.getPartitionPath().equals("2015/01/31")
-        && !r.isCurrentLocationKnown()).count() == 1L);
+            && !r.isCurrentLocationKnown()).count() == 1L);
+    assertTrue(taggedRecordRDD.collectAsList().stream().filter(r -> r.getPartitionPath().equals("2016/01/31")
+            && r.isCurrentLocationKnown()).count() == 3L);
   }
 
   private HoodieWriteConfig makeConfig() {