buildbucket: add retry API.

appengine/cr-buildbucket/test/service_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 contextlib
 import datetime
 
 from components import auth
 from components import utils
 from google.appengine.ext import ndb
@@ skipping 91 matching lines @@
 
   def test_add_with_auth_error(self):
     self.mock_cannot(acl.Action.ADD_BUILD)
     with self.assertRaises(auth.AuthorizationError):
       service.add(self.test_build.bucket)
 
   def test_add_with_bad_parameters(self):
     with self.assertRaises(errors.InvalidInputError):
       service.add('bucket', parameters=[])
 
+  ################################### RETRY ####################################
+
+  def test_retry(self):
+    self.test_build.put()
+    build = service.retry(self.test_build.key.id())
+    self.assertIsNotNone(build)
+    self.assertIsNotNone(build.key)
+    self.assertNotEqual(build.key.id(), self.test_build.key.id())
+    self.assertEqual(build.bucket, self.test_build.bucket)
+    self.assertEqual(build.parameters, self.test_build.parameters)
+    self.assertEqual(build.retry_of, self.test_build.key.id())
+
+  def test_retry_not_found(self):
+    with self.assertRaises(errors.BuildNotFoundError):
+      service.retry(2)
+
   #################################### GET #####################################
 
   def test_get(self):
     self.test_build.put()
     build = service.get(self.test_build.key.id())
     self.assertEqual(build, self.test_build)
 
   def test_get_nonexistent_build(self):
     self.assertIsNone(service.get(42))
 
@@ skipping 144 matching lines @@
     build2 = model.Build(
       bucket=self.test_build.bucket,
       created_by=auth.Identity.from_bytes('user:x@chromium.org')
     )
     build2.put()
 
     builds, _ = service.search(
       created_by='x@chromium.org', buckets=[self.test_build.bucket])
     self.assertEqual(builds, [build2])
 
+  def test_search_by_retry_of(self):
+    self.test_build.put()
+    build2 = model.Build(
+      bucket=self.test_build.bucket,
+      retry_of=42,
+    )
+    build2.put()
+
+    builds, _ = service.search(retry_of=42)
+    self.assertEqual(builds, [build2])
+
   def test_search_by_created_by_with_bad_string(self):
     with self.assertRaises(errors.InvalidInputError):
       service.search(created_by='blah')
 
   def test_search_with_paging(self):
     self.put_many_builds()
 
     first_page, next_cursor = service.search(
       buckets=[self.test_build.bucket],
       max_builds=10,
@@ skipping 482 matching lines @@
         self.test_build.bucket, model.BuildStatus.SCHEDULED)
 
   def test_delete_many_builds_schedule_task(self):
     service.delete_many_builds(
       self.test_build.bucket, model.BuildStatus.SCHEDULED)
 
   def test_delete_many_completed_builds(self):
     with self.assertRaises(errors.InvalidInputError):
       service.delete_many_builds(
         self.test_build.bucket, model.BuildStatus.COMPLETED)