Add support for an --image flag to atest.

scheduler/scheduler_models.py

 """Database model classes for the scheduler.
 
 Contains model classes abstracting the various DB tables used by the scheduler.
 These overlap the Django models in basic functionality, but were written before
 the Django models existed and have not yet been phased out.  Some of them
 (particularly HostQueueEntry and Job) have considerable scheduler-specific logic
 which would probably be ill-suited for inclusion in the general Django model
 classes.
 
 Globals:
@@ skipping 809 matching lines @@
     _delay_ready_task = None
 
     # TODO(gps): On scheduler start/recovery we need to call HQE.on_pending() on
     # all status='Pending' atomic group HQEs incase a delay was running when the
     # scheduler was restarted and no more hosts ever successfully exit Verify.
 
     def __init__(self, id=None, row=None, **kwargs):
         assert id or row
         super(Job, self).__init__(id=id, row=row, **kwargs)
         self._owner_model = None # caches model instance of owner
+        self.update_image_path = None # path of OS image to install
 
 
     def model(self):
         return models.Job.objects.get(id=self.id)
 
 
     def owner_model(self):
         # work around the fact that the Job owner field is a string, not a
         # foreign key
         if not self._owner_model:
@@ skipping 14 matching lines @@
                 SELECT * FROM afe_host_queue_entries
                 WHERE job_id= %s
         """, (self.id,))
         entries = [HostQueueEntry(row=i) for i in rows]
 
         assert len(entries)>0
 
         return entries
 
 
+    def is_image_update_job(self):
+        """
+        Discover if the current job requires an OS update.
+
+        @return: True/False if OS should be updated before job is run.
+        """
+        # All image update jobs have the parameterized_job_id set.
+        if not self.parameterized_job_id:
+            return False
+
+        # Retrieve the ID of the ParameterizedJob this job is an instance of.
+        rows = _db.execute("""
+                SELECT test_id
+                FROM afe_parameterized_jobs
+                WHERE id = %s
+                """, (self.parameterized_job_id,))
+        if not rows:
+            return False
+        test_id = rows[0][0]
+
+        # Retrieve the ID of the known autoupdate_ParameterizedJob.
+        rows = _db.execute("""
+                SELECT id
+                FROM afe_autotests
+                WHERE name = 'autoupdate_ParameterizedJob'
+                """)
+        if not rows:
+            return False
+        update_id = rows[0][0]
+
+        # If the IDs are the same we've found an image update job.
+        if test_id == update_id:
+            # Finally, get the path to the OS image to install.
+            rows = _db.execute("""
+                    SELECT parameter_value
+                    FROM afe_parameterized_job_parameters
+                    WHERE parameterized_job_id = %s
+                    """, (self.parameterized_job_id,))
+            if rows:
+                # Save the path in update_image_path to use later as a command
+                # line parameter to autoserv.
+                self.update_image_path = rows[0][0]
+                return True
+
+        return False
+
+
     def get_execution_details(self):
         """
         Get test execution details for this job.
 
         @return: Dictionary with test execution details
         """
         def _find_test_jobs(rows):
             """
             Here we are looking for tests such as SERVER_JOB and CLIENT_JOB.*
             Those are autotest 'internal job' tests, so they should not be
@@ skipping 435 matching lines @@
     def abort_delay_ready_task(self):
         """Abort the delayed task associated with this job, if any."""
         if self._delay_ready_task:
             # Cancel any pending callback that would try to run again
             # as we are already running.
             self._delay_ready_task.abort()
 
 
     def __str__(self):
         return '%s-%s' % (self.id, self.owner)