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 37 matching lines @@
     )
     self.assertTrue(build.is_leased)
     self.assertGreater(build.lease_expiration_date, utils.utcnow())
     self.assertIsNotNone(build.lease_key)
 
   def test_add_with_auth_error(self):
     self.mock_cannot(acl.Action.ADD_BUILD)
     with self.assertRaises(auth.AuthorizationError):
       self.service.add(self.test_build.bucket)
 
+  ################################# ADD_BATCH ##################################
+
+  def test_add_batch(self):
+    builds = self.service.add_batch([{'bucket': 'chromium'}, {'bucket': 'v8'}])
+    self.assertEqual(len(builds), 2)
+
+    self.assertIsNotNone(builds[0].key)
+    self.assertIsNotNone(builds[0].key.id())
+    self.assertEqual(builds[0].bucket, 'chromium')
+
+    self.assertIsNotNone(builds[1].key)
+    self.assertIsNotNone(builds[1].key.id())
+    self.assertEqual(builds[1].bucket, 'v8')
+
+  def test_add_batch_too_many(self):
+    build_reqs = [{'bucket': 'chromium'}] * 30
+    with self.assertRaises(errors.InvalidInputError):
+      self.service.add_batch(build_reqs)
+
+  def test_add_batch_with_bad_build(self):
+    with self.assertRaises(errors.InvalidInputError):
+      self.service.add_batch([{'bucket': 'chromium'}, {'bucket': 2}])
+    self.assertFalse(model.Build.query().fetch())
+
   #################################### GET #####################################
 
   def test_get(self):
     self.test_build.put()
     build = self.service.get(self.test_build.key.id())
     self.assertEqual(build, self.test_build)
 
   def test_get_nonexistent_build(self):
     self.assertIsNone(self.service.get(42))
 
@@ skipping 527 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)