buildbucket: put_batch endpoint

appengine_module/testing_utils/testing.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.
 
 from contextlib import contextmanager
 import collections
 import time
 
 # appengine sdk is supposed to be on the path.
 import dev_appserver
@@ skipping 10 matching lines @@
 
 
 class AppengineTestCase(auto_stub.TestCase):  # pragma: no cover
   """Base class for Appengine test cases.
 
   Must set app_module to use self.test_app.
   """
 
   # To be set in tests that wants to use test_app
   app_module = None
+  # Datastore consistency policy.
+  consistency_policy = None

[Vadim Sh., 2015/04/16 00:12:21]

hm.. can you please commit it in a separate CL (verifying all tests pass)? Also,
why is it needed?

[nodir, 2015/04/16 00:36:20]

not needed anymore, since xg transactions are not used

 
   def setUp(self):
     super(AppengineTestCase, self).setUp()
     self.testbed = testbed.Testbed()
     # needed because endpoints expects a . in this value
     self.testbed.setup_env(current_version_id='testbed.version')
     self.testbed.activate()
     # Can't use init_all_stubs() because PIL isn't in wheel.
     self.testbed.init_app_identity_stub()
     self.testbed.init_blobstore_stub()
     self.testbed.init_capability_stub()
     self.testbed.init_channel_stub()
-    self.testbed.init_datastore_v3_stub()
+    self.testbed.init_datastore_v3_stub(
+        consistency_policy=self.consistency_policy)
     self.testbed.init_files_stub()
     self.testbed.init_logservice_stub()
     self.testbed.init_mail_stub()
     self.testbed.init_memcache_stub()
     self.testbed.init_taskqueue_stub()
     self.testbed.init_urlfetch_stub()
     self.testbed.init_user_stub()
     self.testbed.init_xmpp_stub()
     # Test app is lazily initialized on a first use from app_module.
     self._test_app = None
@@ skipping 126 matching lines @@
   @contextmanager
   def call_should_fail(self, _status):
     """Asserts that Endpoints call inside the guarded region of code fails."""
     # This should be a call_api(..., status=<something>). Unfortunately,
     # Cloud Endpoints doesn't interact with webtest properly. See
     # https://code.google.com/p/googleappengine/issues/detail?id=10544 and
     # http://stackoverflow.com/questions/24219654/content-length-error-in-
     #   google-cloud-endpoints-testing
     with self.assertRaises(AssertionError):
       yield