buildbucket: put_batch endpoint

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
 import json
 
 from components import auth
 from components import utils
@@ skipping 48 matching lines @@
     build = self.service.add(
         bucket='chromium',
         parameters=params,
     )
     self.assertIsNotNone(build.key)
     self.assertIsNotNone(build.key.id())
     self.assertEqual(build.bucket, 'chromium')
     self.assertEqual(build.parameters, params)
     self.assertEqual(build.created_by, auth.get_current_identity())
 
+  def test_add_with_client_operation_id(self):
+    build = self.service.add(
+        bucket='chromium',
+        parameters={'buildername': 'linux_rel'},
+        client_operation_id='1',
+    )
+    build2 = self.service.add(
+        bucket='chromium',
+        parameters={'buildername': 'linux_rel'},
+        client_operation_id='1',
+    )
+    self.assertIsNotNone(build.key)
+    self.assertEqual(build, build2)
+
   def test_add_with_bad_bucket_name(self):
     with self.assertRaises(errors.InvalidInputError):
       self.service.add(bucket='chromium as')
     with self.assertRaises(errors.InvalidInputError):
       self.service.add(bucket='')
 
   def test_add_with_leasing(self):
     build = self.service.add(
         bucket='chromium',
         lease_expiration_date=utils.utcnow () + datetime.timedelta(seconds=10),
@@ skipping 547 matching lines @@
     self.assertEqual(build.status, model.BuildStatus.SCHEDULED)
     self.assertIsNone(build.lease_key)
 
   def test_completed_builds_are_not_reset(self):
     self.test_build.status = model.BuildStatus.COMPLETED
     self.test_build.result = model.BuildResult.SUCCESS
     self.test_build.put()
     self.service.reset_expired_builds()
     build = self.test_build.key.get()
     self.assertEqual(build.status, model.BuildStatus.COMPLETED)