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.assertIn('retry_of:%s' % self.test_build.key.id(), build.tags)
+
+  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 646 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)