#!/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()