gae_ts_mon: instrument Cloud Endpoint methods

appengine/chromium_try_flakes/handlers/test/flake_issues_test.py

 # Copyright 2015 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
 import mock
 import urllib2
 
 from google.appengine.datastore import datastore_stub_util
 from google.appengine.ext import ndb
 
 from handlers.flake_issues import ProcessIssue, CreateFlakyRun
 import main
+from infra_libs import ts_mon
 from model.flake import Flake, FlakyRun, FlakeOccurrence
 from model.build_run import PatchsetBuilderRuns, BuildRun
 from testing_utils import testing
 from time_functions.testing import mock_datetime_utc
 
 
 TEST_TEST_RESULTS_REPLY = json.dumps({
   'tests': {
     'test1': {
       'expected': 'PASS',
@@ skipping 111 matching lines @@
 
   # This is needed to be able to test handlers using cross-group transactions.
   datastore_stub_consistency_policy = (
       datastore_stub_util.PseudoRandomHRConsistencyPolicy(probability=1))
 
   # Needed to read queues from queue.yaml in the root of the app.
   taskqueue_stub_root_path = '.'
 
   def setUp(self):
     super(FlakeIssuesTestCase, self).setUp()
+    ts_mon.reset_for_unittest(disable=True)
     self.mock_api = MockIssueTrackerAPI()
     self.patchers = [
         mock.patch('issue_tracker.issue_tracker_api.IssueTrackerAPI',
                    lambda *args, **kwargs: self.mock_api),
         mock.patch('issue_tracker.issue.Issue', MockIssue),
     ]
     for patcher in self.patchers:
       patcher.start()
 
   def tearDown(self):
@@ skipping 584 matching lines @@
 
     for flake in flakes:
       self.assertEqual(flake.occurrences, [flaky_run.key])
 
   def test_flattens_tests_correctly(self):
     passed, failed, skipped = CreateFlakyRun._flatten_tests(
         json.loads(TEST_TEST_RESULTS_REPLY)['tests'], '/')
     self.assertEqual(set(passed), set(['test1']))
     self.assertEqual(set(failed), set(['test2/a', 'test2/d']))
     self.assertEqual(set(skipped), set(['test2/b']))