You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.1 KiB
87 lines
2.1 KiB
7 months ago
|
# Lint as: python2, python3
|
||
|
|
# pylint: disable=missing-docstring
|
||
|
|
# 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.
|
||
|
|
|
||
|
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import six
|
||
|
|
|
||
|
|
|
||
|
|
def any_call(*args, **kwargs):
|
||
|
|
"""An empty method to handle any call.
|
||
|
|
"""
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def decorate(f):
|
||
|
|
"""A noop decorator.
|
||
|
|
"""
|
||
|
|
return f
|
||
|
|
|
||
|
|
|
||
|
|
def decorate_wrapper(f):
|
||
|
|
"""Wrapper of the noop decorator.
|
||
|
|
|
||
|
|
Calling this method with any args will return the noop decorator function.
|
||
|
|
"""
|
||
|
|
return decorate
|
||
|
|
|
||
|
|
|
||
|
|
class mock_class_type(type):
|
||
|
|
"""Type class for the mock class to handle any class methods."""
|
||
|
|
|
||
|
|
def __getattr__(self, attr):
|
||
|
|
# This is to support decorators like "@metrics.SecondsTimerDecorator"
|
||
|
|
# In this case, the call returns a function which returns a noop
|
||
|
|
# decorator function ("decorate").
|
||
|
|
if 'Decorator' in attr:
|
||
|
|
return decorate_wrapper
|
||
|
|
else:
|
||
|
|
return mock_class_base
|
||
|
|
|
||
|
|
|
||
|
|
class mock_class_base(six.with_metaclass(mock_class_type, object)):
|
||
|
|
"""Base class for a mock es class."""
|
||
|
|
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def __getattribute__(self, name):
|
||
|
|
|
||
|
|
# TODO(dshi): Remove this method after all reference of timer.get_client
|
||
|
|
# is removed.
|
||
|
|
def get_client(*args, **kwargs):
|
||
|
|
return self
|
||
|
|
|
||
|
|
# get_client is to support call like "timer.get_client", which returns
|
||
|
|
# a class supporting Context when being called.
|
||
|
|
if name == 'get_client':
|
||
|
|
return get_client
|
||
|
|
elif name == 'indices':
|
||
|
|
return mock_class_base()
|
||
|
|
|
||
|
|
return any_call
|
||
|
|
|
||
|
|
|
||
|
|
def __enter__(self, *args, **kwargs):
|
||
|
|
"""Method to support Context class."""
|
||
|
|
return self
|
||
|
|
|
||
|
|
|
||
|
|
def __exit__(self, *args, **kwargs):
|
||
|
|
"""Method to support Context class."""
|
||
|
|
|
||
|
|
|
||
|
|
def __getitem__(self, key):
|
||
|
|
"""Method to override __getitem__."""
|
||
|
|
return self
|
||
|
|
|
||
|
|
|
||
|
|
def __setitem__(self, key, value):
|
||
|
|
"""Method to override __setitem__."""
|