chrome/test/ispy/server/update_mask_handler.py

   1 # Copyright 2013 The Chromium 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 """Request Handler to allow test mask updates."""
   6
   7 import webapp2
   8 import re
   9 import sys
  10 import os
  11
  12 from common import constants
  13 from common import image_tools
  14 from common import ispy_utils
  15
  16 import gs_bucket
  17
  18
  19 class UpdateMaskHandler(webapp2.RequestHandler):
  20   """Request handler to allow test mask updates."""
  21
  22   def post(self):
  23     """Accepts post requests.
  24
  25     This method will accept a post request containing device, site and
  26     device_id parameters. This method takes the diff of the run
  27     indicated by it's parameters and adds it to the mask of the run's
  28     test. It will then delete the run it is applied to and redirect
  29     to the device list view.
  30     """
  31     test_run = self.request.get('test_run')
  32     expectation = self.request.get('expectation')
  33
  34     # Short-circuit if a parameter is missing.
  35     if not (test_run and expectation):
  36       self.response.headers['Content-Type'] = 'json/application'
  37       self.response.write(json.dumps(
  38           {'error': '\'test_run\' and \'expectation\' must be '
  39                     'supplied to update a mask.'}))
  40       return
  41     # Otherwise, set up the utilities.
  42     self.bucket = gs_bucket.GoogleCloudStorageBucket(constants.BUCKET)
  43     self.ispy = ispy_utils.ISpyUtils(self.bucket)
  44     # Short-circuit if the failure does not exist.
  45     if not self.ispy.FailureExists(test_run, expectation):
  46       self.response.headers['Content-Type'] = 'json/application'
  47       self.response.write(json.dumps(
  48         {'error': 'Could not update mask because failure does not exist.'}))
  49       return
  50     # Get the failure namedtuple (which also computes the diff).
  51     failure = self.ispy.GetFailure(test_run, expectation)
  52     # Upload the new mask in place of the original.
  53     self.ispy.UpdateImage(
  54         ispy_utils.GetExpectationPath(expectation, 'mask.png'),
  55         image_tools.ConvertDiffToMask(failure.diff))
  56     # Now that there is no diff for the two images, remove the failure.
  57     self.ispy.RemoveFailure(test_run, expectation)
  58     # Redirect back to the sites list for the test run.
  59     self.redirect('/?test_run=%s' % test_run)