mgr/dashboard_v2: Added coverage tracking to module code

This will allow to collect code coverage information when the module
is running within the ceph-mgr daemon.

Signed-off-by: Ricardo Dias <rdias@suse.com>

diff --git a/src/pybind/mgr/dashboard_v2/.coveragerc b/src/pybind/mgr/dashboard_v2/.coveragerc
index 3832b81dbe00bcbb87ea38ba65c64cd1861ceb7f..29a63192c3f836587bdc82ce4c3671d16d9c4c81 100644 (file)
--- a/src/pybind/mgr/dashboard_v2/.coveragerc
+++ b/src/pybind/mgr/dashboard_v2/.coveragerc
@@ -3,4 +3,5 @@ omit = tests/*
        */python*/*
        ceph_module_mock.py
        __init__.py
+       */mgr_module.py
 

diff --git a/src/pybind/mgr/dashboard_v2/module.py b/src/pybind/mgr/dashboard_v2/module.py
index 5643e1bb3282c15200cc5c6c36e0f2572c2fac18..37cd99323a40d46c299efe0ec907f364fa289bb1 100644 (file)
--- a/src/pybind/mgr/dashboard_v2/module.py
+++ b/src/pybind/mgr/dashboard_v2/module.py
@@ -14,6 +14,12 @@ except ImportError:
 import cherrypy
 from mgr_module import MgrModule
 
+if 'COVERAGE_ENABLED' in os.environ:
+    import coverage
+    _cov = coverage.Coverage(config_file="{}/.coveragerc".format(os.path.dirname(__file__)))
+    _cov.start()
+
+# pylint: disable=wrong-import-position
 from .controllers.auth import Auth
 from .tools import load_controllers, json_error_page, SessionExpireAtBrowserCloseTool, \
                    NotificationQueue
@@ -105,6 +111,8 @@ class Module(MgrModule):
         cherrypy.tree.mount(Module.StaticRoot(), '/', config=config)
 
     def serve(self):
+        if 'COVERAGE_ENABLED' in os.environ:
+            _cov.start()
         self.configure_cherrypy()
 
         cherrypy.engine.start()
@@ -112,6 +120,9 @@ class Module(MgrModule):
         logger.info('Waiting for engine...')
         self.log.info('Waiting for engine...')
         cherrypy.engine.block()
+        if 'COVERAGE_ENABLED' in os.environ:
+            _cov.stop()
+            _cov.save()
         logger.info('Engine done')
 
     def shutdown(self):