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
+from google.appengine.datastore import datastore_stub_util
 from google.appengine.ext import testbed
 from testing_utils import testing
 import mock
 
 import acl
 import errors
 import model
 import service
 
 
 class BuildBucketServiceTest(testing.AppengineTestCase):
   def __init__(self, *args, **kwargs):
     super(BuildBucketServiceTest, self).__init__(*args, **kwargs)
     self.test_build = None
 
   def mock_cannot(self, action):
     def can(_bucket, requested_action, _identity=None):
       return action != requested_action
     self.mock(acl, 'can', can)
 
   def setUp(self):
+    # Create a consistency policy that will simulate the High Replication
+    # consistency model.
+    self.consistency_policy = (
+        datastore_stub_util.PseudoRandomHRConsistencyPolicy(probability=1.0))
+
     super(BuildBucketServiceTest, self).setUp()
     self.service = service.BuildBucketService()
     self.test_build = model.Build(
         bucket='chromium',
         parameters={
             'buildername': 'infra',
             'changes': [{
                 'author': 'nodir@google.com',
                 'message': 'buildbucket: initial commit'
             }]
@@ skipping 17 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)