Update TsFileInsertionEventParserTest.java

Author: Caideyipi

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/pipe/event/TsFileInsertionEventParserTest.java

@@ -340,11 +340,22 @@ public void testTableParserSkipsUnnecessaryBitMaps() throws Exception {
               false)) {
         final Iterator<TabletInsertionEvent> iterator = parser.toTabletInsertionEvents().iterator();
         Assert.assertTrue(iterator.hasNext());
-        final Tablet parsedTablet =
-            ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
-        assertBitMapExistence(parsedTablet, false, false, true);
-        Assert.assertTrue(parsedTablet.isNull(1, 2));
-        Assert.assertFalse(iterator.hasNext());
+        Tablet parsedTablet = ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
+        if (parsedTablet.getSchemas().size() == 3) {
+          assertBitMapExistence(parsedTablet, false, false, true);
+          Assert.assertTrue(parsedTablet.isNull(1, 2));
+          Assert.assertFalse(iterator.hasNext());
+        } else {
+          Assert.assertNull(parsedTablet.getBitMaps());
+          while (iterator.hasNext()) {
+            parsedTablet = ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
+            if (parsedTablet.getSchemas().get(1).getMeasurementName().equals("dense")) {
+              Assert.assertNull(parsedTablet.getBitMaps());
+            } else {
+              Assert.assertTrue(parsedTablet.isNull(1, 1));
+            }
+          }
+        }
       }
     } finally {
       CommonDescriptor.getInstance()
@@ -1095,11 +1106,23 @@ private void testTreeParserSkipsUnnecessaryBitMaps(final boolean isQuery) throws
                   false)) {
         final Iterator<TabletInsertionEvent> iterator = parser.toTabletInsertionEvents().iterator();
         Assert.assertTrue(iterator.hasNext());
-        final Tablet parsedTablet =
-            ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
-        assertBitMapExistence(parsedTablet, false, true);
-        Assert.assertTrue(parsedTablet.isNull(1, 1));
-        Assert.assertFalse(iterator.hasNext());
+        Tablet parsedTablet = ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
+        if (parsedTablet.getSchemas().size() > 1) {
+          assertBitMapExistence(parsedTablet, false, true);
+          Assert.assertTrue(parsedTablet.isNull(1, 1));
+          Assert.assertFalse(iterator.hasNext());
+        } else {
+          Assert.assertNull(parsedTablet.getBitMaps());
+          Assert.assertTrue(iterator.hasNext());
+          while (iterator.hasNext()) {
+            parsedTablet = ((PipeRawTabletInsertionEvent) iterator.next()).convertToTablet();
+            if (parsedTablet.getSchemas().get(0).getMeasurementName().equals("dense")) {
+              Assert.assertNull(parsedTablet.getBitMaps());
+            } else {
+              Assert.assertTrue(parsedTablet.isNull(0, 1));
+            }
+          }
+        }
       }
     } finally {
       CommonDescriptor.getInstance()