ceph/src/pybind/mgr/tests/__init__.py

   1 # type: ignore
   2 from __future__ import absolute_import
   3
   4
   5 import os
   6
   7 if 'UNITTEST' in os.environ:
   8
   9     # Mock ceph_module. Otherwise every module that is involved in a testcase and imports it will
  10     # raise an ImportError
  11
  12     import sys
  13
  14     try:
  15         from unittest import mock
  16     except ImportError:
  17         import mock
  18
  19     class M(object):
  20         def __init__(self, *args):
  21             super(M, self).__init__()
  22             self._ceph_get_version = mock.Mock()
  23             self._ceph_get = mock.MagicMock()
  24             self._ceph_get_module_option = mock.MagicMock()
  25             self._ceph_get_option = mock.MagicMock()
  26             self._validate_module_option = lambda _: True
  27             self._configure_logging = lambda *_: None
  28             self._unconfigure_logging = mock.MagicMock()
  29             self._ceph_log = mock.MagicMock()
  30             self._ceph_get_store = lambda _: ''
  31             self._ceph_get_store_prefix = lambda _: {}
  32             self._ceph_dispatch_remote = lambda *_: None
  33
  34
  35     cm = mock.Mock()
  36     cm.BaseMgrModule = M
  37     cm.BaseMgrStandbyModule = M
  38     sys.modules['ceph_module'] = cm
  39     sys.modules['rados'] = mock.Mock()