zdtm: update check_pages_counts() to support memfd_secret dump stats

Signed-off-by: Dhanuka Warusadura <[email protected]>
checkpoint-restore · Aug 28, 2023 · 1213549 · 1213549
1 parent 66da41d
commit 1213549
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/zdtm.py b/test/zdtm.py
@@ -1250,8 +1250,12 @@ def check_pages_counts(self):
         with open(self.__stats_file("dump"), 'rb') as stfile:
             stats = crpc.images.load(stfile)
             stent = stats['entries'][0]['dump']
-            stats_written = int(stent['shpages_written']) + int(
-                stent['pages_written'])
+            if stent['secmempages_written']:
+                stats_written = int(stent['secmempages_written']) + int(
+                    stent['pages_written']) + 1
+            else:
+                stats_written = int(stent['shpages_written']) + int(
+                    stent['pages_written'])
 
         if self.__stream:
             self.spawn_criu_image_streamer("extract")