make connect work if no roles are specified

This is useful for -nuke.

diff --git a/teuthology/task/internal.py b/teuthology/task/internal.py
index 3782792aa9234a4a4e1bb4f196c7a38de9c910b7..337cf8075be6553b408a7f4a49a19b638f05e4a0 100644 (file)
--- a/teuthology/task/internal.py
+++ b/teuthology/task/internal.py
@@ -97,8 +97,13 @@ def connect(ctx, config):
     remotes = [remote.Remote(name=t, ssh=connection.connect(t))
                for t in ctx.config['targets']]
     ctx.cluster = orchestra.cluster.Cluster()
-    for rem, roles in zip(remotes, ctx.config['roles']):
-        ctx.cluster.add(rem, roles)
+    if 'roles' in ctx.config:
+        for rem, roles in zip(remotes, ctx.config['roles']):
+            ctx.cluster.add(rem, roles)
+    else:
+        for rem in remotes:
+            ctx.cluster.add(rem, rem.name)
+        
 
 def check_conflict(ctx, config):
     log.info('Checking for old test directory...')