gae_ts_mon: instrument Cloud Endpoint methods

infra_libs/ts_mon/common/interface.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.
 
 """Classes representing the monitoring interface for tasks or devices.
 
 Usage:
   import argparse
   from infra_libs import ts_mon
 
@@ skipping 60 matching lines @@
     # --ts-mon-flush-interval-secs seconds.  May be None if
     # --ts-mon-flush != 'auto' or --ts-mon-flush-interval-secs == 0.
     self.flush_thread = None
     # All metrics created by this application.
     self.metrics = {}
     # The MetricStore object that holds the actual metric values.
     self.store = store_ctor(self)
     # Cached time of the last flush. Useful mostly in AppEngine apps.
     self.last_flushed = datetime.datetime.utcfromtimestamp(0)
 
+  def reset_for_unittest(self):
+    self.metrics = {}
+    self.last_flushed = datetime.datetime.utcfromtimestamp(0)
+    self.store.reset_for_unittest()
+
 state = State()
 
 
 def flush():
   """Send all metrics that are registered in the application."""
   if not state.global_monitor or not state.target:
     raise errors.MonitoringNoConfiguredMonitorError(None)
 
   if not state.flush_enabled_fn():
     logging.debug('ts_mon: sending metrics is disabled.')
@@ skipping 34 matching lines @@
   """Removes the metric from the list of metrics sent by flush()."""
   del state.metrics[metric.name]
 
 
 def close():
   """Stops any background threads and waits for them to exit."""
   if state.flush_thread is not None:
     state.flush_thread.stop()
 
 
-def reset_for_unittest():
-  state.store.reset_for_unittest()
+def reset_for_unittest(disable=False):
+  state.reset_for_unittest()
+  if disable:
+    state.flush_enabled_fn = lambda: False
 
 
 class _FlushThread(threading.Thread):
   """Background thread that flushes metrics on an interval."""
 
   def __init__(self, interval_secs, stop_event=None):
     super(_FlushThread, self).__init__(name='ts_mon')
 
     if stop_event is None:
       stop_event = threading.Event()
@@ skipping 30 matching lines @@
             'Last monitoring flush took %f seconds (longer than '
             '--ts-mon-flush-interval-secs = %f seconds)',
             flush_duration, self.interval_secs)
         next_timeout = 0
 
   def stop(self):
     """Stops the background thread and performs a final flush."""
 
     self.stop_event.set()
     self.join()