Home | History | Annotate | Download | only in control_segments
      1 # Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
      2 # Use of this source code is governed by a BSD-style license that can be
      3 # found in the LICENSE file.
      4 
      5 
      6 from autotest_lib.client.cros import constants
      7 from autotest_lib.server.cros import provision
      8 
      9 
     10 # job_labels should be a string like "name:setting,name:setting"
     11 # non-provisionable labels are currently skipped, so they're safe to pass in.
     12 job_labels = locals().get('job_labels') or ','.join(args)
     13 labels_list = [label.strip() for label in job_labels.split(',') if label]
     14 
     15 
     16 def provision_machine(machine):
     17     """
     18     Run the appropriate provisioning tests to make the machine's labels match
     19     those given in job_labels.
     20     """
     21     host = hosts.create_host(machine, try_lab_servo=True)
     22 
     23     fixed, provisionable = provision.filter_labels(labels_list)
     24 
     25     job.record('START', None, 'provision')
     26     try:
     27         job.sysinfo.add_logdir(constants.AUTOUPDATE_PRESERVE_LOG)
     28         provision.run_special_task_actions(job, host, labels_list,
     29                                            provision.Provision)
     30     except Exception as e:
     31         logging.exception(e)
     32         job.record('END FAIL', None, 'provision')
     33         # Raising a blank exception is done here because any message we can
     34         # give here would be less useful than whatever the failing test left as
     35         # its own exception message.
     36         #
     37         # The gory details of how raising a blank exception accomplishes this
     38         # is as follows:
     39         #
     40         # The scheduler only looks at the return code of autoserv to see if
     41         # the special task failed.  Therefore we need python to exit because
     42         # of an unhandled exception or because someone called sys.exit(1).
     43         #
     44         # We can't call sys.exit, since there's post-job-running logic (like
     45         # cleanup) that we'd be skipping out on.  So therefore, we need to
     46         # raise an exception.  However, if we raise an exception, this
     47         # exception ends up triggering server_job to write an INFO line with
     48         # job_abort_reason equal to str(e), which the tko parser then picks
     49         # up as the reason field for the job when the status.log we generate is
     50         # parsed as the job's results.
     51         #
     52         # So therefore, we raise a blank exception, which then generates an
     53         # empty job_abort_reason which the tko parser ignores just inserts as
     54         # a SERVER_JOB failure with no reason, which we then ignore at suite
     55         # results reporting time.
     56         raise Exception('')
     57     else:
     58         # If we finish successfully, nothing in autotest ever looks at the
     59         # status.log, so it's purely for human consumption and tracability.
     60         job.record('END GOOD', None, 'provision',
     61                    '%s provisioned successfully' % machine)
     62 
     63 
     64 job.parallel_simple(provision_machine, machines)
     65 
     66 # vim: set syntax=python :
     67