From 83dc3b31cfd73309225d1b2ada263f6a394f231e Mon Sep 17 00:00:00 2001
From: Haojun Liao <hjliao@taosdata.com>
Date: Mon, 19 Jul 2021 15:13:57 +0800
Subject: [PATCH] [td-225]add test files.

---
 src/tsdb/src/tsdbRead.c      |  2 +-
 src/util/tests/arrayTest.cpp | 52 ++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+), 1 deletion(-)
 create mode 100644 src/util/tests/arrayTest.cpp

diff --git a/src/tsdb/src/tsdbRead.c b/src/tsdb/src/tsdbRead.c
index 924b5e24a8..6c16c72c2a 100644
--- a/src/tsdb/src/tsdbRead.c
+++ b/src/tsdb/src/tsdbRead.c
@@ -3095,7 +3095,7 @@ STimeWindow updateLastrowForEachGroup(STableGroupInfo *groupList) {
   // window does not being updated, so set the original
   if (window.skey == INT64_MAX && window.ekey == INT64_MIN) {
     window = TSWINDOW_INITIALIZER;
-    assert(totalNumOfTable == 0 && taosArrayGetSize(groupList->pGroupList) == 0);
+    assert(totalNumOfTable == 0 && taosArrayGetSize(groupList->pGroupList) == numOfGroups);
   }
 
   taosArrayRemoveBatch(groupList->pGroupList, TARRAY_GET_START(emptyGroup), taosArrayGetSize(emptyGroup));
diff --git a/src/util/tests/arrayTest.cpp b/src/util/tests/arrayTest.cpp
new file mode 100644
index 0000000000..08983b6c06
--- /dev/null
+++ b/src/util/tests/arrayTest.cpp
@@ -0,0 +1,52 @@
+#include <gtest/gtest.h>
+#include <stdlib.h>
+#include <time.h>
+#include <random>
+
+#include "tarray.h"
+
+namespace {
+
+static void remove_batch_test() {
+  SArray *pa = (SArray*) taosArrayInit(4, sizeof(int32_t));
+
+  for(int32_t i = 0; i < 20; ++i) {
+    int32_t a = i;
+    taosArrayPush(pa, &a);
+  }
+
+  SArray* delList = (SArray*)taosArrayInit(4, sizeof(int32_t));
+  taosArrayRemoveBatch(pa, (const int32_t*) TARRAY_GET_START(delList), taosArrayGetSize(delList));
+  EXPECT_EQ(taosArrayGetSize(pa), 20);
+
+  int32_t a = 5;
+  taosArrayPush(delList, &a);
+
+  taosArrayRemoveBatch(pa, (const int32_t*) TARRAY_GET_START(delList), taosArrayGetSize(delList));
+  EXPECT_EQ(taosArrayGetSize(pa), 19);
+  EXPECT_EQ(*(int*)taosArrayGet(pa, 5), 6);
+
+  taosArrayInsert(pa, 5, &a);
+  EXPECT_EQ(taosArrayGetSize(pa), 20);
+  EXPECT_EQ(*(int*)taosArrayGet(pa, 5), 5);
+
+  taosArrayClear(delList);
+
+  a = 6;
+  taosArrayPush(delList, &a);
+
+  a = 9;
+  taosArrayPush(delList, &a);
+
+  a = 14;
+  taosArrayPush(delList, &a);
+  taosArrayRemoveBatch(pa, (const int32_t*) TARRAY_GET_START(delList), taosArrayGetSize(delList));
+  EXPECT_EQ(taosArrayGetSize(pa), 17);
+}
+}  // namespace
+
+TEST(arrayTest, array_list_test) {
+
+  remove_batch_test();
+
+}