From d318c8f9831db04533bd2e4821e49ffe60d29aa5 Mon Sep 17 00:00:00 2001
From: Alex Duan <417921451@qq.com>
Date: Sun, 31 Dec 2023 17:04:18 +0800
Subject: [PATCH] fix: add data correct check point

---
 tests/army/community/cluster/snapshot.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/army/community/cluster/snapshot.py b/tests/army/community/cluster/snapshot.py
index 40783ad4ce..66f9d9f598 100644
--- a/tests/army/community/cluster/snapshot.py
+++ b/tests/army/community/cluster/snapshot.py
@@ -49,14 +49,16 @@ class TDTestCase(TBase):
         # split vgroups
         self.splitVGroups()
         self.trimDb()
+        self.checkAggCorrect()
 
         # balance vgroups
         self.balanceVGroupLeader()
 
         # replica to 1
         self.alterReplica(1)
+        self.checkAggCorrect()
 
-        vgids = self.getVGroup()
+        vgids = self.getVGroup(self.db)
         selid = random.choice(vgids)
         self.balanceVGroupLeaderOn(selid)