fix test

Author: jt2594838

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/storageengine/dataregion/memtable/AbstractMemTablePartialInsertTest.java

@@ -22,13 +22,15 @@
 import org.apache.iotdb.commons.exception.IllegalPathException;
 import org.apache.iotdb.commons.path.PartialPath;
 import org.apache.iotdb.commons.queryengine.plan.planner.plan.node.PlanNodeId;
+import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.exception.WriteProcessException;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.write.InsertRowNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.write.InsertTabletNode;
 
 import org.apache.tsfile.enums.TSDataType;
 import org.apache.tsfile.utils.BitMap;
 import org.apache.tsfile.write.schema.MeasurementSchema;
+import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
 
@@ -52,10 +54,21 @@
 public class AbstractMemTablePartialInsertTest {
 
   private PrimitiveMemTable memTable;
+  private static boolean prevEnableNullValueInWriteThroughputMetric;
 
   @Before
   public void setUp() {
     memTable = new PrimitiveMemTable("root.sg", "0");
+    prevEnableNullValueInWriteThroughputMetric =
+        IoTDBDescriptor.getInstance().getConfig().isIncludeNullValueInWriteThroughputMetric();
+    IoTDBDescriptor.getInstance().getConfig().setIncludeNullValueInWriteThroughputMetric(false);
+  }
+
+  @After
+  public void tearDown() {
+    IoTDBDescriptor.getInstance()
+        .getConfig()
+        .setIncludeNullValueInWriteThroughputMetric(prevEnableNullValueInWriteThroughputMetric);
   }
 
   // =========================================================================
@@ -70,7 +83,7 @@ public void testInsertAlignedRow_noFailure_noNullValue_pointsInsertedEqualsTotal
     // formula: getMeasurementColumnCnt(3) - failedNum(0) - nullPoints(0) = 3
     InsertRowNode node =
         buildAlignedInsertRowNode(
-            new String[] {"s0", "s1", "s2"}, new Object[] {1, 2L, 3.0f}, -1 /* no failure */);
+            new String[] {"s0", "s1", "s2"}, new Object[] {1, 2, 3}, -1 /* no failure */);
 
     int points = memTable.insertAlignedRow(node);
 
@@ -91,7 +104,7 @@ public void testInsertAlignedRow_oneFailedMeasurement_pointsInsertedNotNegative(
     // 2 measurements, first one fails
     InsertRowNode node =
         buildAlignedInsertRowNode(
-            new String[] {"s0", "s1"}, new Object[] {1, 2L}, 0 /* mark index 0 as failed */);
+            new String[] {"s0", "s1"}, new Object[] {1, 2}, 0 /* mark index 0 as failed */);
 
     int points = memTable.insertAlignedRow(node);
 
@@ -111,6 +124,7 @@ public void testInsertAlignedRow_allMeasurementsFailed_pointsInsertedIsZero()
     // mark both as failed
     node.markFailedMeasurement(0);
     node.markFailedMeasurement(1);
+    node.setFailedMeasurementNumber(2);
 
     int points = memTable.insertAlignedRow(node);
 
@@ -173,6 +187,7 @@ public void testInsertTablet_allMeasurementsFailed_pointsInsertedIsZero()
     InsertTabletNode node = buildInsertTabletNode(new String[] {"s0", "s1"}, rowCount, null, -1);
     node.markFailedMeasurement(0);
     node.markFailedMeasurement(1);
+    node.setFailedMeasurementNumber(2);
 
     int points = memTable.insertTablet(node, 0, rowCount);
 
@@ -236,6 +251,7 @@ private static InsertRowNode buildAlignedInsertRowNode(
             false);
     if (failedIndex >= 0) {
       node.markFailedMeasurement(failedIndex);
+      node.setFailedMeasurementNumber(1);
     }
     return node;
   }
@@ -274,6 +290,7 @@ private static InsertTabletNode buildInsertTabletNode(
             rowCount);
     if (failedIndex >= 0) {
       node.markFailedMeasurement(failedIndex);
+      node.setFailedMeasurementNumber(1);
     }
     return node;
   }