import unittest

from unittest.mock import patch

import singer.metrics as metrics

import time

import copy

class DummyException(Exception):

pass

def logged_points(mock):

return [args[1] for args, _ in mock.call_args_list]

class TestRecordCounter(unittest.TestCase):

@patch('singer.metrics.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))

@patch('singer.metrics.log')

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))

class TestHttpRequestTimer(unittest.TestCase):

@patch('singer.metrics.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)

@patch('singer.metrics.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))

@patch('singer.metrics.log')

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))

class TestParse(unittest.TestCase):

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'}))

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_invalid_json_returns_none(self):

point = metrics.parse('INFO METRIC: something that is invalid }')

self.assertIsNone(point)

def test_parse_no_match(self):

point = metrics.parse('a line that is not a metric')

self.assertIsNone(point)