From e0da2a4e8cae525f6d4e5383abcd47b3737bcb0e Mon Sep 17 00:00:00 2001
From: Prasanna Kumar Kalever <prasanna.kalever@redhat.com>
Date: Wed, 19 Oct 2022 14:41:11 +0530
Subject: [PATCH] qa/workunits/rbd: Add test to list snapshots of consistency
 group

Signed-off-by: Prasanna Kumar Kalever <prasanna.kalever@redhat.com>
---
 qa/workunits/rbd/rbd_groups.sh | 47 ++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/qa/workunits/rbd/rbd_groups.sh b/qa/workunits/rbd/rbd_groups.sh
index 2c9fd318f3c..4c684adfb0e 100755
--- a/qa/workunits/rbd/rbd_groups.sh
+++ b/qa/workunits/rbd/rbd_groups.sh
@@ -111,6 +111,16 @@ create_snapshot()
     rbd group snap create $group_name@$snap_name
 }
 
+create_snapshots()
+{
+    local group_name=$1
+    local snap_name=$2
+    local snap_count=$3
+    for i in `seq 1 $snap_count`; do
+        rbd group snap create $group_name@$snap_name$i
+    done
+}
+
 remove_snapshot()
 {
     local group_name=$1
@@ -118,6 +128,16 @@ remove_snapshot()
     rbd group snap remove $group_name@$snap_name
 }
 
+remove_snapshots()
+{
+    local group_name=$1
+    local snap_name=$2
+    local snap_count=$3
+    for i in `seq 1 $snap_count`; do
+        rbd group snap remove $group_name@$snap_name$i
+    done
+}
+
 rename_snapshot()
 {
     local group_name=$1
@@ -146,6 +166,16 @@ check_snapshot_in_group()
     list_snapshots $group_name | grep $snap_name
 }
 
+check_snapshots_count_in_group()
+{
+    local group_name=$1
+    local snap_name=$2
+    local expected_count=$3
+    local actual_count
+    actual_count=$(list_snapshots $group_name | grep -c $snap_name)
+    (( actual_count == expected_count ))
+}
+
 check_snapshot_not_in_group()
 {
     local group_name=$1
@@ -206,4 +236,21 @@ remove_group $group
 remove_image $image
 echo "PASSED"
 
+echo "TEST: list snapshots of consistency group"
+image="test_image"
+group="test_consistency_group"
+snap="group_snap"
+create_image $image
+create_group $group
+add_image_to_group $image $group
+create_snapshots $group $snap 10
+check_snapshots_count_in_group $group $snap 10
+remove_snapshots $group $snap 10
+create_snapshots $group $snap 100
+check_snapshots_count_in_group $group $snap 100
+remove_snapshots $group $snap 100
+remove_group $group
+remove_image $image
+echo "PASSED"
+
 echo "OK"