site_utils/graphite/statsd_mock_unittest.py

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