autotest: allow tests to be run on all clients

teuthology/misc.py

@@ -375,3 +375,18 @@ def get_first_mon(ctx, config):
     firstmon = sorted(mons)[0]
     assert firstmon
     return firstmon
+
+def replace_all_with_clients(cluster, config):
+    """
+    Converts a dict containing a key all to one
+    mapping all clients to the value of config['all']
+    """
+    assert isinstance(config, dict), 'config must be a dict'
+    if 'all' not in config:
+        return config
+    norm_config = {}
+    assert len(config) == 1, \
+        "config cannot have 'all' and specific clients listed"
+    for client in all_roles_of_type(cluster, 'client'):
+        norm_config['client.{id}'.format(id=client)] = config['all']
+    return norm_config

teuthology/task/autotest.py

@@ -25,9 +25,17 @@ def task(ctx, config):
         - autotest:
             client.0: [dbench]
             client.1: [bonnie]
+
+    You can also specify a list of tests to run on all clients::
+
+        tasks:
+        - ceph:
+        - cfuse:
+        - autotest:
+            all: [dbench]
     """
     assert isinstance(config, dict)
-
+    config = teuthology.replace_all_with_clients(ctx.cluster, config)
     log.info('Setting up autotest...')
     with parallel() as p:
         for role in config.iterkeys():

teuthology/task/rbd.py

@@ -340,24 +340,6 @@ def mount(ctx, config):
                     ]
                 )
 
-def _normalize_config(cluster, config):
-    """
-    Returns a configuration that can be used by rbd subtasks.
-
-    This is either a list of clients or a dict mapping clients
-    to image options. "all" is converted to individual clients.
-    """
-    assert isinstance(config, list) or isinstance(config, dict), \
-        "task rbd only supports a list or dict for configuration"
-    if isinstance(config, list) or 'all' not in config:
-        return config
-    norm_config = {}
-    assert len(config) == 1, \
-        "rbd config cannot have 'all' and specific clients listed"
-    for client in teuthology.all_roles_of_type(cluster, 'client'):
-        norm_config['client.{id}'.format(id=client)] = config['all']
-    return norm_config
-
 @contextlib.contextmanager
 def task(ctx, config):
     """
@@ -396,7 +378,9 @@ def task(ctx, config):
               image_size: 20480
               fs_type: xfs
     """
-    norm_config = _normalize_config(ctx.cluster, config)
+    norm_config = config
+    if isinstance(config, dict):
+        norm_config = teuthology.replace_all_with_clients(ctx.cluster, config)
     if isinstance(norm_config, dict):
         role_images = {}
         for role, properties in norm_config.iteritems():