"""Unit tests for NewRelic connector."""

from analytics.connectors import statsd


def test_metric():
    """Test the metric decorator."""
    @statsd.metric('foo', count=2)
    def test():
        pass

    test()
    call_args = statsd.send_metric.call_args
    assert call_args[0][0] == 'ows.analytics.foo'
    assert call_args[1].get('count') == 2
    statsd.send_metric.reset()


def test_timing():
    """Test the timing decoration."""
    @statsd.timing('foo')
    def test():
        for i in range(2000):
            continue

    test()
    call_args = statsd.send_timer.call_args
    assert call_args[0][0] == 'ows.analytics.foo'
    assert call_args[1].get('delta') > 0
    statsd.send_timer.reset()