]>
Commit | Line | Data |
---|---|---|
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() |