From 7234481bae5e23cd7ed27bd935413b2df49ecbf2 Mon Sep 17 00:00:00 2001 From: Loic Dachary Date: Thu, 20 Nov 2014 13:54:11 +0100 Subject: [PATCH] ceph_manager: define pool context To create a pool before running a code bloc and remove it after. with manager.pool("mypool"): mytest.. Signed-off-by: Loic Dachary --- tasks/ceph_manager.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tasks/ceph_manager.py b/tasks/ceph_manager.py index b1ffc80ec77cf..05e94111f882e 100644 --- a/tasks/ceph_manager.py +++ b/tasks/ceph_manager.py @@ -2,6 +2,7 @@ ceph manager -- Thrasher and CephManager objects """ from cStringIO import StringIO +import contextlib import random import time import gevent @@ -898,6 +899,12 @@ class CephManager: erasure_code_profile_name=erasure_code_profile_name) return name + @contextlib.contextmanager + def pool(self, pool_name, pg_num=16, erasure_code_profile_name=None): + self.create_pool(pool_name, pg_num, erasure_code_profile_name) + yield + self.remove_pool(pool_name) + def create_pool(self, pool_name, pg_num=16, erasure_code_profile_name=None): """ Create a pool named from the pool_name parameter. -- 2.39.5