qa/workunits/rbd: handle exception thrown from close during lock test

Signed-off-by: Jason Dillaman <dillaman@redhat.com>

diff --git a/src/test/librbd/rbdrw.py b/src/test/librbd/rbdrw.py
old mode 100755 (executable)
new mode 100644 (file)
index 6034ad0..7179551
--- a/src/test/librbd/rbdrw.py
+++ b/src/test/librbd/rbdrw.py
@@ -18,14 +18,13 @@ import rados, rbd, sys
 
 with rados.Rados(conffile='') as r:
     with r.open_ioctx('rbd') as ioctx:
-        with rbd.Image(ioctx, sys.argv[1]) as image:
-            image.lock_exclusive(sys.argv[2])
-            while True:
-                try:
+        try:
+            with rbd.Image(ioctx, sys.argv[1]) as image:
+                image.lock_exclusive(sys.argv[2])
+                while True:
                     image.write('A' * 4096, 0)
                     r = image.read(0, 4096)
-                except rbd.ConnectionShutdown:
-                    # it so happens that the errno here is 108, but
-                    # anything recognizable would do
-                    exit(108)
-
+        except rbd.ConnectionShutdown:
+            # it so happens that the errno here is 108, but
+            # anything recognizable would do
+            exit(108)