#!/usr/bin/python
import unittest
import statsd_mock as statsd
class statsd_mock_test(unittest.TestCase):
"""Test statsd_mock"""
def test_average_mock(self):
"""Test mock class Average"""
statsd.Average('average').send('name', 1)
def test_connection_mock(self):
"""Test mock class Connection"""
connection = statsd.Connection(host='host', port=1)
statsd.Connection.set_defaults(host='host', port=1)
def test_counter_mock(self):
"""Test mock class Counter"""
counter = statsd.Counter('counter')
counter.increment(subname='name', delta=1)
counter.decrement(subname='name', delta=1)
def test_gauge_mock(self):
"""Test mock class Gauge"""
statsd.Gauge('gauge').send('name', 1)
def test_raw_mock(self):
"""Test mock class Raw"""
statsd.Raw('raw').send(subname='name', value=1, timestamp=None)
def test_timer_mock(self):
"""Test mock class Timer"""
timer = statsd.Timer('timer')
timer.start()
timer.stop()
class decorate_test(object):
"""Test class to test timer decorator."""
test_timer = statsd.Timer('test')
@test_timer.decorate
def f(self):
"""Test function to apply timer decorator to."""
return True
dt = decorate_test()
self.assertTrue(dt.f(), 'timer decorator failed.')
if __name__ == '__main__':
unittest.main()