Merge remote branch 'origin/upstream' into tempbranch

frontend/planner/rpc_utils.py

 import common
 import os
 from autotest_lib.frontend.afe import models as afe_models, model_logic
 from autotest_lib.frontend.planner import models, model_attributes
+from autotest_lib.frontend.planner import failure_actions
 from autotest_lib.client.common_lib import global_config, utils
 
 
 PLANNER_LABEL_PREFIX = 'planner_'
 PLANNER_ATOMIC_GROUP_NAME = 'planner_global_atomic_group'
 SERVER = global_config.global_config.get_config_value('SERVER', 'hostname')
 LAZY_LOADED_FILES = {}
 
 
 def create_plan_label(plan):
@@ skipping 89 matching lines @@
     """
     if host.blocked:
         return None
 
     test_configs = plan.testconfig_set.order_by('execution_order')
     for test_config in test_configs:
         afe_jobs = plan.job_set.filter(test_config=test_config)
         afe_job_ids = afe_jobs.values_list('afe_job', flat=True)
         hqes = afe_models.HostQueueEntry.objects.filter(job__id__in=afe_job_ids,
                                                         host=host.host)
-        if not hqes:
-            return test_config.id
+        if not hqes and not bool(test_config.skipped_hosts.filter(host=host)):
+            return test_config
         for hqe in hqes:
             if not hqe.complete:
                 # HostQueueEntry still active for this host,
                 # should not run another test
                 return None
 
     # All HQEs related to this host are complete
     host.complete = True
     host.save()
     return None
@@ skipping 26 matching lines @@
     """
     host = afe_models.Host.objects.get(hostname=hostname)
 
     planner_host = models.Host.objects.get(plan=plan, host=host)
     test_run, _ = models.TestRun.objects.get_or_create(plan=plan,
                                                        test_job=planner_job,
                                                        tko_test=tko_test,
                                                        host=planner_host)
     test_run.status = status
     test_run.save()
+
+
+def _site_process_host_action_dummy(host, action):
+    return False
+
+
+def process_host_action(host, action):
+    """
+    Takes the specified action on the host
+    """
+    HostAction = failure_actions.HostAction
+    if action not in HostAction.values:
+        raise ValueError('Unexpected host action %s' % action)
+
+    site_process = utils.import_site_function(
+            __file__, 'autotest_lib.frontend.planner.site_rpc_utils',
+            'site_process_host_action', _site_process_host_action_dummy)
+
+    if not site_process(host, action):
+        # site_process_host_action returns True and and only if it matched a
+        # site-specific processing option
+        if action == HostAction.BLOCK:
+            host.blocked = True
+        elif action == HostAction.UNBLOCK:
+            host.blocked = False
+        else:
+            assert action == HostAction.REINSTALL
+            raise NotImplemented('TODO: implement reinstall')
+
+        host.save()
+
+
+def process_test_action(planner_job, action):
+    """
+    Takes the specified action for this planner job
+    """
+    TestAction = failure_actions.TestAction
+    if action not in TestAction.values:
+        raise ValueError('Unexpected test action %s' % action)
+
+    if action == TestAction.SKIP:
+        # Do nothing
+        pass
+    else:
+        assert action == TestAction.RERUN
+        planner_job.requires_rerun = True
+        planner_job.save()