From 18e2f5703ca439c471b1d30962dd9409da58b54e Mon Sep 17 00:00:00 2001
From: jiajingbin <jiajaybin@126.com>
Date: Tue, 31 Aug 2021 20:25:37 +0800
Subject: [PATCH] add testcases for issue-7666

---
 tests/pytest/insert/verifyMemToDiskCrash.py | 85 +++++++++++++++++++++
 tests/pytest/util/common.py                 |  6 ++
 2 files changed, 91 insertions(+)
 create mode 100644 tests/pytest/insert/verifyMemToDiskCrash.py

diff --git a/tests/pytest/insert/verifyMemToDiskCrash.py b/tests/pytest/insert/verifyMemToDiskCrash.py
new file mode 100644
index 0000000000..d234864317
--- /dev/null
+++ b/tests/pytest/insert/verifyMemToDiskCrash.py
@@ -0,0 +1,85 @@
+###################################################################
+#           Copyright (c) 2016 by TAOS Technologies, Inc.
+#                     All rights reserved.
+#
+#  This file is proprietary and confidential to TAOS Technologies.
+#  No part of this file may be reproduced, stored, transmitted,
+#  disclosed or used in any form or by any means other than as
+#  expressly provided by the written permission from Jianhui Tao
+#
+###################################################################
+
+# -*- coding: utf-8 -*-
+from util.log import tdLog
+from util.cases import tdCases
+from util.sql import tdSql
+import time 
+from util.common import tdCom
+
+
+class TDTestCase:
+    def init(self, conn, logSql):
+        tdLog.debug("start to execute %s" % __file__)
+        tdSql.init(conn.cursor(), logSql)
+
+    def checkMemDiskMerge(self):
+        tb_name = tdCom.getLongName(8, "letters")
+        tdSql.execute(
+            f'CREATE TABLE {tb_name} (ts timestamp, c1 int, c2 int)')
+        tdSql.execute(
+            f'insert into {tb_name} values ("2021-01-01 12:00:00.000", 1, 1)')
+        tdSql.execute(
+            f'insert into {tb_name} values ("2021-01-03 12:00:00.000", 3, 3)')
+        tdCom.restartTaosd()
+        tdSql.execute(
+            f'insert into {tb_name} values ("2021-01-02 12:00:00.000", Null, 2)')
+        tdSql.execute(
+            f'insert into {tb_name} values ("2021-01-04 12:00:00.000", Null, 4)')
+        query_sql = f'select * from {tb_name}'
+        res1 = tdSql.query(query_sql, True)
+        tdCom.restartTaosd()
+        res2 = tdSql.query(query_sql, True)
+        for i in range(4):
+            tdSql.checkEqual(res1[i], res2[i])
+    
+    def checkSuperSubBlockMerge(self):
+        tdCom.cleanTb()
+        tb_name = tdCom.getLongName(8, "letters")
+        tdSql.execute(
+            f'CREATE TABLE {tb_name} (ts timestamp, c1 int)')
+
+        start_ts = 1577808001000
+        for i in range(80):
+            tdSql.execute(
+                f'insert into {tb_name} values ({start_ts}, {i})')
+            start_ts += 1
+        tdCom.restartTaosd()
+
+        tdSql.query(f'select * from {tb_name}')
+        for i in range(10):
+            tdSql.execute(
+                f'insert into {tb_name} values ({start_ts}, Null)')
+            start_ts += 1
+        tdCom.restartTaosd()
+
+        tdSql.query(f'select * from {tb_name}')
+        for i in range(10):
+            new_ts = i + 10 + 10
+            tdSql.execute(
+                f'insert into {tb_name} values ({start_ts}, {new_ts})')
+            start_ts += 1
+        tdCom.restartTaosd()
+        tdSql.query(f'select * from {tb_name}')
+
+    def run(self):
+        tdSql.prepare()
+        # self.checkMemDiskMerge()
+        self.checkSuperSubBlockMerge()
+
+    def stop(self):
+        tdSql.close()
+        tdLog.success("%s successfully executed" % __file__)
+
+
+tdCases.addWindows(__file__, TDTestCase())
+tdCases.addLinux(__file__, TDTestCase())
diff --git a/tests/pytest/util/common.py b/tests/pytest/util/common.py
index 35abc4802f..6d387ac674 100644
--- a/tests/pytest/util/common.py
+++ b/tests/pytest/util/common.py
@@ -14,6 +14,7 @@
 import random
 import string
 from util.sql import tdSql
+from util.dnodes import tdDnodes
 
 class TDCom:
     def init(self, conn, logSql):
@@ -47,6 +48,11 @@ class TDCom:
             chars = ''.join(random.choice(string.ascii_letters.lower() + string.digits) for i in range(len))
         return chars
 
+    def restartTaosd(self, index=1, db_name="db"):
+        tdDnodes.stop(index)
+        tdDnodes.startWithoutSleep(index)
+        tdSql.execute(f"use {db_name}")
+
     def close(self):
         self.cursor.close()
 
-- 
GitLab