pybind/mgr: Add mgr/progress to mon_command reference

Signed-off-by: Sebastian Wagner <sebastian.wagner@suse.com>

diff --git a/src/pybind/mgr/progress/__init__.py b/src/pybind/mgr/progress/__init__.py
index e3379608af763aaea15df615d9ee398eb9b7d7ed..3f333e01b276c66a42b9f167355e47ae1861c25f 100644 (file)
--- a/src/pybind/mgr/progress/__init__.py
+++ b/src/pybind/mgr/progress/__init__.py
@@ -1,7 +1,7 @@
 import os
-if 'UNITTEST' not in os.environ:
-    from .module import *
-else:
+
+if 'UNITTEST' in os.environ:
     import tests
+from .module import Module
 
 

diff --git a/src/pybind/mgr/progress/module.py b/src/pybind/mgr/progress/module.py
index c37d7f2ce8fd2f96d1210593a3dd8d306f185aa0..4e2b301890db27467be8e329ade89faad9b1ff19 100644 (file)
--- a/src/pybind/mgr/progress/module.py
+++ b/src/pybind/mgr/progress/module.py
@@ -21,9 +21,6 @@ ENCODING_VERSION = 2
 # keep a global reference to the module so we can use it from Event methods
 _module = None  # type: Optional["Module"]
 
-# if unit test we want MgrModule to be blank
-if 'UNITTEST' in os.environ:
-    MgrModule = object  # type: ignore
 
 class Event(object):
     """

diff --git a/src/pybind/mgr/progress/test_progress.py b/src/pybind/mgr/progress/test_progress.py
index 52f2ae716637b4bc722ae0b8cd50b4d3f3a28228..12a23f4a7baedd20be5f5a2dd32ba7fae363af41 100644 (file)
--- a/src/pybind/mgr/progress/test_progress.py
+++ b/src/pybind/mgr/progress/test_progress.py
@@ -126,8 +126,9 @@ class TestModule(object):
         # bunch of attributes for testing
 
         module.PgRecoveryEvent.pg_update = mock.Mock()
-        self.test_module = module.Module() # so we can see if an event gets created
-        self.test_module.log = mock.Mock() # we don't need to log anything
+        module.Module._ceph_get_option = mock.Mock()  # .__init__
+        module.Module._configure_logging = lambda *args: ...  # .__init__
+        self.test_module = module.Module('module_name', 0, 0)  # so we can see if an event gets created
         self.test_module.get = mock.Mock() # so we can call pg_update
         self.test_module._complete = mock.Mock() # we want just to see if this event gets called
         self.test_module.get_osdmap = mock.Mock() # so that self.get_osdmap().get_epoch() works