buildbucket: put_batch endpoint

appengine/cr-buildbucket/test/api_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 datetime
 import httplib
 import json
 import mock
 
 from components import auth
@@ skipping 60 matching lines @@
     self.assertEqual(
         resp['build']['parameters_json'], '{"buildername": "linux_rel"}')
 
   def test_get_nonexistent_build(self):
     self.service.get.return_value = None
     self.expect_error('get', {'id': 1}, 'BUILD_NOT_FOUND')
 
   ##################################### PUT ####################################
 
   def test_put(self):
-    self.test_build.tags = ['owner=ivan']
+    self.test_build.tags = ['owner:ivan']
     self.service.add.return_value = self.test_build
     req = {
         'bucket': self.test_build.bucket,
         'tags': self.test_build.tags,
     }
     resp = self.call_api('put', req).json_body
     self.service.add.assert_called_once_with(
         bucket=self.test_build.bucket,
         tags=req['tags'],
         parameters=None,
@@ skipping 32 matching lines @@
   def test_put_with_empty_bucket(self):
     self.expect_error('put', {'bucket': ''}, 'INVALID_INPUT')
 
   def test_put_with_malformed_parameters_json(self):
     req = {
         'bucket':'chromium',
         'parameters_json': '}non-json',
     }
     self.expect_error('put', req, 'INVALID_INPUT')
 
+  ################################## PUT_BATCH #################################
+
+  def test_put_batch(self):
+    self.test_build.tags = ['owner:ivan']
+    build2 = model.Build(bucket='v8')
+    build2.put()
+    self.service.add_batch.return_value = [self.test_build, build2]
+    req = {
+        'builds': [
+            {
+                'bucket': self.test_build.bucket,
+                'tags': self.test_build.tags,
+            },
+            {
+                'bucket': build2.bucket,
+            },
+        ],
+    }
+    resp = self.call_api('put_batch', req).json_body
+    self.service.add_batch.assert_called_once_with([
+        {
+            'bucket': self.test_build.bucket,
+            'tags': self.test_build.tags,
+            'parameters': None,
+            'lease_expiration_date': None,
+        },
+        {
+            'bucket': build2.bucket,
+            'tags': [],
+            'parameters': None,
+            'lease_expiration_date': None,
+        },
+    ])
+
+    res_build0 = resp['builds'][0]
+    self.assertEqual(res_build0['id'], str(self.test_build.key.id()))
+    self.assertEqual(res_build0['bucket'], self.test_build.bucket)
+    self.assertEqual(res_build0['tags'], self.test_build.tags)
+
+    res_build1 = resp['builds'][1]
+    self.assertEqual(res_build1['id'], str(self.test_build.key.id()))
+    self.assertEqual(res_build1['bucket'], build2.bucket)
+
   #################################### SEARCH ##################################
 
   def test_search(self):
     self.test_build.put()
     self.service.search.return_value = ([self.test_build], 'the cursor')
     req = {
         'bucket': ['chromium'],
         'cancelation_reason': 'CANCELED_EXPLICITLY',
         'created_by': 'user:x@chromium.org',
         'result': 'CANCELED',
@@ skipping 343 matching lines @@
     self.error_test(errors.BuildNotFoundError, 'BUILD_NOT_FOUND')
 
   def test_invalid_input_error(self):
     self.error_test(errors.InvalidInputError, 'INVALID_INPUT')
 
   def test_invalid_build_state_error(self):
     self.error_test(errors.InvalidBuildStateError, 'INVALID_BUILD_STATE')
 
   def test_lease_expired_error(self):
     self.error_test(errors.LeaseExpiredError, 'LEASE_EXPIRED')