buildbucket: add retry API.

appengine/cr-buildbucket/test/api_test.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import datetime
 import json
 
 from components import auth
 from components import utils
 from google.appengine.ext import ndb
@@ skipping 58 matching lines @@
   def test_get_nonexistent_build(self):
     service.get.return_value = None
     self.expect_error('get', {'id': 1}, 'BUILD_NOT_FOUND')
 
   ##################################### PUT ####################################
 
   def test_put(self):
     self.test_build.tags = ['owner:ivan']
     service.add.return_value = self.test_build
     req = {
+      'client_operation_id': '42',
       'bucket': self.test_build.bucket,
       'tags': self.test_build.tags,
       'pubsub_callback': {
         'topic': 'projects/foo/topic/bar',
         'user_data': 'hello',
         'auth_token': 'secret',
       }
     }
     resp = self.call_api('put', req).json_body
     service.add.assert_called_once_with(
       bucket=self.test_build.bucket,
       tags=req['tags'],
       parameters=None,
       lease_expiration_date=None,
-      client_operation_id=None,
+      client_operation_id='42',
       pubsub_callback=model.PubSubCallback(
         topic='projects/foo/topic/bar',
         user_data='hello',
         auth_token='secret',
       ),
     )
     self.assertEqual(resp['build']['id'], str(self.test_build.key.id()))
     self.assertEqual(resp['build']['bucket'], req['bucket'])
     self.assertEqual(resp['build']['tags'], req['tags'])
 
@@ skipping 25 matching lines @@
     self.assertEqual(
       resp['build']['lease_expiration_ts'], req['lease_expiration_ts'])
 
   def test_put_with_malformed_parameters_json(self):
     req = {
       'bucket': 'chromium',
       'parameters_json': '}non-json',
     }
     self.expect_error('put', req, 'INVALID_INPUT')
 
+  #################################### RETRY ###################################
+
+  def test_retry(self):
+    build = model.Build(
+      bucket='chromium',
+      parameters={'builder_name': 'debug'},
+      tags = ['a:b'],
+      retry_of=2,
+    )
+    build.put()
+    service.retry.return_value = build
+
+    req = {
+      'id': build.key.id(),
+      'client_operation_id': '42',
+      'pubsub_callback': {
+        'topic': 'projects/foo/topic/bar',
+        'user_data': 'hello',
+        'auth_token': 'secret',
+      }
+    }
+    resp = self.call_api('retry', req).json_body
+    service.retry.assert_called_once_with(
+      build.key.id(),
+      client_operation_id='42',
+      lease_expiration_date=None,
+      pubsub_callback=model.PubSubCallback(
+        topic='projects/foo/topic/bar',
+        user_data='hello',
+        auth_token='secret',
+      ),
+    )
+    self.assertEqual(resp['build']['id'], str(build.key.id()))
+    self.assertEqual(resp['build']['bucket'], build.bucket)
+    self.assertEqual(
+      json.loads(resp['build']['parameters_json']), build.parameters)
+    self.assertEqual(resp['build']['retry_of'], '2')
+
+  def test_retry_not_found(self):
+    service.retry.side_effect = errors.BuildNotFoundError
+    self.expect_error('retry', {'id': 42}, 'BUILD_NOT_FOUND')
+
   ################################## PUT_BATCH #################################
 
   def test_put_batch(self):
     self.test_build.tags = ['owner:ivan']
     build1_future = ndb.Future()
     build1_future.set_result(self.test_build)
 
     build2 = model.Build(id=2, bucket='v8')
     build2_future = ndb.Future()
     build2_future.set_result(build2)
@@ skipping 60 matching lines @@
   def test_search(self):
     self.test_build.put()
     service.search.return_value = ([self.test_build], 'the cursor')
     req = {
       'bucket': ['chromium'],
       'cancelation_reason': 'CANCELED_EXPLICITLY',
       'created_by': 'user:x@chromium.org',
       'result': 'CANCELED',
       'status': 'COMPLETED',
       'tag': ['important'],
+      'retry_of': '42',
     }
 
     res = self.call_api('search', req).json_body
 
     service.search.assert_called_once_with(
       buckets=req['bucket'],
       tags=req['tag'],
       status=model.BuildStatus.COMPLETED,
       result=model.BuildResult.CANCELED,
       failure_reason=None,
       cancelation_reason=model.CancelationReason.CANCELED_EXPLICITLY,
       created_by='user:x@chromium.org',
       max_builds=None,
-      start_cursor=None)
+      start_cursor=None,
+      retry_of=42,
+    )
     self.assertEqual(len(res['builds']), 1)
     self.assertEqual(res['builds'][0]['id'], str(self.test_build.key.id()))
     self.assertEqual(res['next_cursor'], 'the cursor')
 
   ##################################### PEEK ###################################
 
   def test_peek(self):
     self.test_build.put()
     service.peek.return_value = ([self.test_build], 'the cursor')
     req = {'bucket': [self.test_build.bucket]}
@@ skipping 264 matching lines @@
   def test_invalid_input_error(self):
     self.error_test(errors.InvalidInputError, 'INVALID_INPUT')
 
   def test_lease_expired_error(self):
     self.error_test(errors.LeaseExpiredError, 'LEASE_EXPIRED')
 
   def test_auth_error(self):
     with self.call_should_fail(403):
       service.get.side_effect = auth.AuthorizationError
       self.call_api('get', {'id': 123})