def test_incremental(self, log):
with metrics.record_counter(endpoint='users') as counter:
counter.increment(1)
counter._ready_to_log = lambda: True
counter.increment(2)
counter._ready_to_log = lambda: False
counter.increment(5)
self.assertEqual(
[metrics.Point('counter', 'record_count', 3, {'endpoint': 'users'}),
metrics.Point('counter', 'record_count', 5, {'endpoint': 'users'})],
logged_points(log))
def test_success_with_http_status_code(self, log):
with metrics.http_request_timer('users') as timer:
timer.elapsed = lambda: 0
timer.tags[metrics.Tag.http_status_code] = 200
self.assertEqual(
[metrics.Point('timer', 'http_request_duration', 0, {'endpoint': 'users', 'status': 'succeeded', 'http_status_code': 200})],
logged_points(log))
def test_log_on_exit(self, log):
with metrics.record_counter('users') as counter:
counter.increment()
counter.increment()
self.assertEqual(
[metrics.Point('counter', 'record_count', 2, {'endpoint': 'users'})],
logged_points(log))
def test_success(self, log):
timer = metrics.http_request_timer('users')
timer.elapsed = lambda: 0
with timer:
pass
got = logged_points(log)
self.assertEqual(
[metrics.Point('timer', 'http_request_duration', 0, {'endpoint': 'users', 'status': 'succeeded'})],
got)
def test_failure(self, log):
try:
with metrics.http_request_timer('users') as timer:
timer.elapsed = lambda: 0
timer.tags[metrics.Tag.http_status_code] = 400
raise ValueError('foo is not bar')
except ValueError:
pass
self.assertEqual(
[metrics.Point('timer', 'http_request_duration', 0, {'endpoint': 'users', 'status': 'failed', 'http_status_code': 400})],
logged_points(log))
def test_parse_without_tags(self):
point = metrics.parse('INFO METRIC: {"type": "counter", "metric": "record_count", "value": 10}')
self.assertEqual(
point,
metrics.Point('counter', 'record_count', 10, None))
def test_parse_with_everything(self):
point = metrics.parse('INFO METRIC: {"type": "counter", "metric": "record_count", "value": 10, "tags": {"endpoint": "users"}}')
self.assertEqual(
point,
metrics.Point('counter', 'record_count', 10, {'endpoint': 'users'}))