Add rbd modprobe method.

diff --git a/teuthology/task/rbd.py b/teuthology/task/rbd.py
index 5c9289b71b9f2663bf62565dd6ec8d2b8f9c77e8..0bae7f9accd49b2737ddbf86cf4cb78a267a3e3e 100644 (file)
--- a/teuthology/task/rbd.py
+++ b/teuthology/task/rbd.py
@@ -72,6 +72,42 @@ def create_image(ctx, config):
                     ],
                 )
 
+@contextlib.contextmanager
+def modprobe(ctx, config):
+    """
+    Load the rbd kernel module..
+
+    For example::
+
+        tasks:
+        - ceph:
+        - rbd.create_image: [client.0]
+        - rbd.modprobe: [client.0]
+    """
+    for role in config:
+        (remote,) = ctx.cluster.only(role).remotes.keys()
+        remote.run(
+            args=[
+                'sudo',
+                'modprobe',
+                'rbd',
+                ],
+            )
+    try:
+        yield
+    finally:
+        log.info('Unloading rbd kernel module...')
+        for role in config:
+            (remote,) = ctx.cluster.only(role).remotes.keys()
+            remote.run(
+                args=[
+                    'sudo',
+                    'modprobe',
+                    '-r',
+                    'rbd',
+                    ],
+                )
+
 @contextlib.contextmanager
 def task(ctx, config):
     create_image(ctx, config)