Fix tests in response to breakage in #351

This commit is contained in:
Joseph Schorr 2015-08-17 16:26:20 -04:00
parent 2fd1d5969e
commit 4625ecf273
4 changed files with 14 additions and 7 deletions

View file

@ -19,7 +19,7 @@ class MetricQueue(object):
def put(self, name, value, **kwargs):
if self._queue is None:
logging.debug('No metric queue: %s %s %s', name, value, kwargs)
logger.debug('No metric queue %s %s %s', name, value, kwargs)
return
try:
@ -47,12 +47,15 @@ def time_after_request(name, metric_queue):
start = getattr(g, '_request_start_time', None)
if start is None:
return r
dur = time.time() - start
dims = {'endpoint': request.endpoint}
metric_queue.put('ResponseTime', dur, dimensions=dims, unit='Seconds')
metric_queue.put('ResponseCode', r.status_code, dimensions=dims)
if r.status_code < 200 or r.status_code >= 300:
metric_queue.put('Non200Response', 1, dimensions={'name': name})
return r
return f