lib/graphite_lib/statsd_mock_unittest.py

# Copyright 2015 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittest for statsd mock."""

from __future__ import print_function

import unittest

from chromite.lib.graphite_lib 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"""
    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()