lib/graphite_lib/stats_es_mock.py

# Copyright (c) 2014 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.

# pylint: disable=deprecated-pragma

"""ElasticSearch mock."""

from __future__ import print_function

class mock_class_base(object):
  """Base class for a mock statsd/es class."""
  # pylint: disable=unused-argument
  def __init__(self, *args, **kwargs):
    pass


  def __getattribute__(self, name):
    def any_call(*args, **kwargs):
      pass

    def decorate(f):
      return f

    # TODO (dshi) crbug.com/256111 - Find better solution for mocking
    # statsd.
    def get_client(*args, **kwargs):
      return self

    if name == 'decorate':
      return decorate
    elif name == 'get_client':
      return get_client
    elif name == 'indices':
      return mock_class_base()

    return any_call

class Stats(object):
  """Stubbed out Stats class to replace stats.Statsd in unit tests."""
  class Average(mock_class_base):
    """Dummy Average class."""
  class Counter(mock_class_base):
    """Dummy Counter class."""
  class Gauge(mock_class_base):
    """Dummy Gauge class."""
  class Timer(mock_class_base):
    """Dummy Timer class."""
  class Raw(mock_class_base):
    """Dummy Raw class."""