From: zhengyin <zhengyin@cmss.chinamobile.com>
Date: Thu, 31 Oct 2019 10:36:09 +0000 (+0800)
Subject: test/pybind: add snap_get_name and snap_get_id test case
X-Git-Tag: v15.1.0~930^2~2
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=8dd5867f2c28e0d9a8d9d8eada26d2d34db53127;p=ceph.git

test/pybind: add snap_get_name and snap_get_id test case

Signed-off-by: Zheng Yin <zhengyin@cmss.chinamobile.com>
---

diff --git a/src/test/pybind/test_rbd.py b/src/test/pybind/test_rbd.py
index 1f2a5b8ddd86..bb5a1d8e025b 100644
--- a/src/test/pybind/test_rbd.py
+++ b/src/test/pybind/test_rbd.py
@@ -915,6 +915,38 @@ class TestImage(object):
         eq(read, data)
         self.image.remove_snap('snap1')
 
+    def test_snap_get_name(self):
+        eq([], list(self.image.list_snaps()))
+        self.image.create_snap('snap1')
+        self.image.create_snap('snap2')
+        self.image.create_snap('snap3')
+
+        for snap in self.image.list_snaps():
+            expected_snap_name = self.image.snap_get_name(snap['id'])
+            eq(expected_snap_name, snap['name'])
+        self.image.remove_snap('snap1')
+        self.image.remove_snap('snap2')
+        self.image.remove_snap('snap3')
+        eq([], list(self.image.list_snaps()))
+
+        assert_raises(ImageNotFound, self.image.snap_get_name, 1)
+
+    def test_snap_get_id(self):
+        eq([], list(self.image.list_snaps()))
+        self.image.create_snap('snap1')
+        self.image.create_snap('snap2')
+        self.image.create_snap('snap3')
+
+        for snap in self.image.list_snaps():
+            expected_snap_id = self.image.snap_get_id(snap['name'])
+            eq(expected_snap_id, snap['id'])
+        self.image.remove_snap('snap1')
+        self.image.remove_snap('snap2')
+        self.image.remove_snap('snap3')
+        eq([], list(self.image.list_snaps()))
+
+        assert_raises(ImageNotFound, self.image.snap_get_id, 'snap1')
+
     def test_set_snap_sparse(self):
         self.image.create_snap('snap1')
         read = self.image.read(0, 256)