buildbucket: fix instrumentation

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 json
 
 from components import auth
 from components import utils
 from google.appengine.ext import ndb
 import mock
+import gae_ts_mon
 
 from testing_utils import testing
 import api
 import errors
 import model
 import service
 
 
 class BuildBucketApiTest(testing.EndpointsTestCase):
   api_service_cls = api.BuildBucketApi
 
   def setUp(self):
     super(BuildBucketApiTest, self).setUp()
+    gae_ts_mon.reset_for_unittest(disable=True)
     for a in dir(service):
       self.mock(service, a, mock.Mock())
 
     self.future_date = utils.utcnow() + datetime.timedelta(minutes=1)
     # future_ts is str because INT64 values are formatted as strings.
     self.future_ts = str(utils.datetime_to_timestamp(self.future_date))
     self.test_build = model.Build(
       id=1,
       bucket='chromium',
       parameters={
@@ skipping 518 matching lines @@
   def test_invalid_input_error(self):
     self.error_test(errors.InvalidInputError, 'INVALID_INPUT')
 
   def test_lease_expired_error(self):
     self.error_test(errors.LeaseExpiredError, 'LEASE_EXPIRED')
 
   def test_auth_error(self):
     with self.call_should_fail(403):
       service.get.side_effect = auth.AuthorizationError
       self.call_api('get', {'id': 123})