From 577c2a69f23ec69fe912fe269b562a6ba097f4b3 Mon Sep 17 00:00:00 2001
From: Hongze Cheng <hzcheng@taosdata.com>
Date: Mon, 29 Apr 2024 13:43:31 +0800
Subject: [PATCH] add test

---
 source/libs/tdb/test/tdbTest.cpp | 39 ++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/source/libs/tdb/test/tdbTest.cpp b/source/libs/tdb/test/tdbTest.cpp
index c1aa15b4d2..e358ac0197 100644
--- a/source/libs/tdb/test/tdbTest.cpp
+++ b/source/libs/tdb/test/tdbTest.cpp
@@ -479,6 +479,45 @@ TEST(tdb_test, DISABLED_simple_upsert1) {
   tdbClose(pEnv);
 }
 
+TEST(tdb_test, simple_upsert2) {
+  int         ret;
+  TDB        *pEnv;
+  TTB        *pDb;
+  int         nData = 10000;
+  const char *key = "key";
+  int32_t     dataSize = 256 * 1024;
+  void       *data = taosMemoryMalloc(dataSize);
+  void       *pData = NULL;
+  SPoolMem   *pPool;
+  TXN        *txn;
+
+  taosRemoveDir("tdb");
+  memset(data, 'a', dataSize);
+
+  // open env
+  ret = tdbOpen("tdb", 4096, 64, &pEnv, 0, 0, 0);
+  GTEST_ASSERT_EQ(ret, 0);
+
+  // open database
+  ret = tdbTbOpen("db.db", -1, -1, NULL, pEnv, &pDb, 0);
+  GTEST_ASSERT_EQ(ret, 0);
+
+  pPool = openPool();
+
+  for (int iData = 0; iData < nData; iData++) {
+    tdbBegin(pEnv, &txn, poolMalloc, poolFree, pPool, TDB_TXN_WRITE | TDB_TXN_READ_UNCOMMITTED);
+
+    ret = tdbTbUpsert(pDb, key, strlen(key), data, dataSize, txn);
+    GTEST_ASSERT_EQ(ret, 0);
+
+    tdbCommit(pEnv, txn);
+    tdbPostCommit(pEnv, txn);
+  }
+
+  tdbTbClose(pDb);
+  tdbClose(pEnv);
+}
+
 TEST(tdb_test, multi_thread_query) {
   int           ret;
   TDB          *pEnv;