From 55b4a9339d8f9b42539cc6118f2c8c5fa8f7aa07 Mon Sep 17 00:00:00 2001
From: bryanchang0603 <changb9@mcamster.ca>
Date: Thu, 10 Jun 2021 10:47:30 +0800
Subject: [PATCH] [TD-4658] update enterprise test part

---
 tests/pytest/alter/alter_keep.py | 58 +++++++++++++++++++++-----------
 1 file changed, 38 insertions(+), 20 deletions(-)

diff --git a/tests/pytest/alter/alter_keep.py b/tests/pytest/alter/alter_keep.py
index f73b8ce460..9d5db9c2ad 100644
--- a/tests/pytest/alter/alter_keep.py
+++ b/tests/pytest/alter/alter_keep.py
@@ -67,40 +67,57 @@ class TDTestCase:
 
 
     def alterKeepEnterprise(self):
-        ## enterprise only accept three inputs
-        ## does not accept 1 paramaters nor 3 paramaters
+        tdLog.notice('running Keep Keep Test, Enterprise Version')
+        #testing keep parameter during create
         tdSql.query('show databases')
         tdSql.checkData(0,7,'3650,3650,3650')
+        tdSql.execute('drop database db')
 
-        tdSql.execute('alter database db keep 10')
+        tdSql.execute('create database db keep 100')
         tdSql.query('show databases')
-        tdSql.checkData(0,7,'10,10,10')
+        tdSql.checkData(0,7,'100,100,100')
+        tdSql.execute('drop database db')
 
-        ## the order for altering keep is keep(D), keep0, keep1.
-        ## if the order is changed, please modify the following test
-        ## to make sure the the test is accurate
+        tdSql.execute('create database db keep 20, 30')
+        tdSql.query('show databases')
+        tdSql.checkData(0,7,'20,30,30')
+        tdSql.execute('drop database db')
 
-        tdSql.execute('alter database db keep 10, 10 ,10')
+        tdSql.execute('create database db keep 30,40,50')
         tdSql.query('show databases')
-        tdSql.checkData(0,7,'10,10,10')
+        tdSql.checkData(0,7,'30,40,50')
+        tdSql.execute('drop database db')
+
+        tdSql.error('create database db keep ')
+        tdSql.error('create database db keep 20,30,40,50')
+        tdSql.error('create database db keep 0')
+        tdSql.error('create database db keep 100,50')
+        tdSql.error('create database db keep 100,40,50')
+        tdSql.error('create database db keep 20,100,50')
+        tdSql.error('create database db keep 100,100,50')
+
+        #testing keep parameter during alter
+        tdSql.execute('create database db')
 
-        tdSql.error('alter database db keep 100, 98 ,99')
+        tdSql.execute('alter database db keep 10')
         tdSql.query('show databases')
         tdSql.checkData(0,7,'10,10,10')
 
-        tdSql.execute('alter database db keep 200, 200 ,200')
+        tdSql.execute('alter database db keep 20,30')
         tdSql.query('show databases')
-        tdSql.checkData(0,7,'200,200,200')
+        tdSql.checkData(0,7,'20,30,30')
 
-        tdSql.execute('alter database db keep 198, 199 ,200')
+        tdSql.execute('alter database db keep 100,200,300')
         tdSql.query('show databases')
-        tdSql.checkData(0,7,'198,199,200')
+        tdSql.checkData(0,7,'100,200,300')
+
+        tdSql.error('alter database db keep ')
+        tdSql.error('alter database db keep 20,30,40,50')
+        tdSql.error('alter database db keep 0')
+        tdSql.error('alter database db keep 100,50')
+        tdSql.error('alter database db keep 100,40,50')
+        tdSql.error('alter database db keep 20,100,50')
 
-        # tdSql.execute('alter database db keep 3650,3650,3650')
-        # tdSql.error('alter database db keep 4000,3640')
-        # tdSql.error('alter database db keep 10,10')
-        # tdSql.query('show databases')
-        # tdSql.checkData(0,7,'3650,3650,3650')
 
     def run(self):
         tdSql.prepare()
@@ -112,7 +129,8 @@ class TDTestCase:
         else:
             tdLog.debug('running community test')
             self.alterKeepCommunity()
-        
+
+        tdSql.prepare()
 
         ##TODO: need to wait for TD-4445 to implement the following
         ##      tests
-- 
GitLab