From ffd14be3bd79baec5d9cafd8972d592085e0e099 Mon Sep 17 00:00:00 2001
From: Zack Cerza <zack@redhat.com>
Date: Tue, 20 Sep 2016 10:03:46 -0600
Subject: [PATCH] Set ctx.cluster even when a job has no roles

This is to allow for teuthology to run self-tests that don't use
resources.

Signed-off-by: Zack Cerza <zack@redhat.com>
---
 teuthology/run.py           |  9 +++++----
 teuthology/task/internal.py | 11 +++++++++--
 2 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/teuthology/run.py b/teuthology/run.py
index e1f79a127..22cb1bc7f 100644
--- a/teuthology/run.py
+++ b/teuthology/run.py
@@ -184,13 +184,14 @@ def get_initial_tasks(lock, config, machine_type):
         init_tasks.append({'internal.lock_machines': (
             len(config['roles']), machine_type)})
 
-
-    init_tasks.append({'internal.save_config': None})
+    init_tasks.extend([
+        {'internal.save_config': None},
+        {'internal.check_lock': None},
+        {'internal.add_remotes': None},
+    ])
 
     if 'roles' in config:
         init_tasks.extend([
-            {'internal.check_lock': None},
-            {'internal.add_remotes': None},
             {'console_log': None},
             {'internal.connect': None},
             {'internal.push_inventory': None},
diff --git a/teuthology/task/internal.py b/teuthology/task/internal.py
index 742f6fd07..7849fac4e 100644
--- a/teuthology/task/internal.py
+++ b/teuthology/task/internal.py
@@ -213,7 +213,11 @@ def check_lock(ctx, config, check_up=True):
     """
     Check lock status of remote machines.
     """
-    if not teuth_config.lock_server or ctx.config.get('check-locks') is False:
+    if (
+        not teuth_config.lock_server or
+        ctx.config.get('check-locks') is False or
+        'roles' not in ctx.config
+    ):
         log.info('Lock checking disabled.')
         return
     log.info('Checking locks...')
@@ -308,6 +312,10 @@ def add_remotes(ctx, config):
     """
     Create a ctx.cluster object populated with remotes mapped to roles
     """
+    ctx.cluster = cluster.Cluster()
+    # Allow jobs to run without using nodes, for self-testing
+    if 'roles' not in ctx.config:
+        return
     remotes = []
     machs = []
     for name in ctx.config['targets'].iterkeys():
@@ -321,7 +329,6 @@ def add_remotes(ctx, config):
             pass
         rem = remote.Remote(name=t, host_key=key, keep_alive=True)
         remotes.append(rem)
-    ctx.cluster = cluster.Cluster()
     if 'roles' in ctx.config:
         for rem, roles in zip(remotes, ctx.config['roles']):
             assert all(isinstance(role, str) for role in roles), \
-- 
2.47.3