chore(test): Add test for coverage

python/lsst/ctrl/mpexec/cli/cmd/commands.py

@@ -29,6 +29,7 @@
 from collections.abc import Iterator, Sequence
 from contextlib import contextmanager
 from functools import partial
+from importlib import import_module
 from tempfile import NamedTemporaryFile
 from typing import Any
 
@@ -146,7 +147,7 @@ def coverage_context(kwargs: dict[str, Any]) -> Iterator[None]:
         return
     # Lazily import coverage only when we might need it
     try:
-        import coverage
+        coverage = import_module("coverage")
     except ModuleNotFoundError:
         raise click.ClickException("coverage was requested but the coverage package is not installed.")
     with NamedTemporaryFile("w") as rcfile:

tests/test_cliScript.py

@@ -28,10 +28,12 @@
 import os
 import tempfile
 import unittest
+import unittest.mock
 
 import click
 import lsst.utils.tests
 from lsst.ctrl.mpexec.cli import opt, script
+from lsst.ctrl.mpexec.cli.cmd.commands import coverage_context
 from lsst.ctrl.mpexec.cli.pipetask import cli as pipetaskCli
 from lsst.ctrl.mpexec.showInfo import ShowInfo
 from lsst.daf.butler.cli.utils import LogCliRunner, clickResultMsg
@@ -203,6 +205,25 @@ def cli(**kwargs):
         self.assertNotEqual(result.exit_code, 0)
 
 
+class CoverageTestCase(unittest.TestCase):
+    """Test coverage context manager."""
+
+    @unittest.mock.patch.dict("sys.modules", coverage=unittest.mock.MagicMock())
+    def testWithCoverage(self):
+        """Test that the coverage context manager runs when invoked."""
+        with coverage_context({"coverage": True}):
+            self.assertTrue(True)
+
+    @unittest.mock.patch("lsst.ctrl.mpexec.cli.cmd.commands.import_module", side_effect=ModuleNotFoundError())
+    def testWithMissingCoverage(self, mock_import):
+        """Test that the coverage context manager complains when coverage is
+        not available.
+        """
+        with self.assertRaises(click.exceptions.ClickException):
+            with coverage_context({"coverage": True}):
+                pass
+
+
 if __name__ == "__main__":
     lsst.utils.tests.init()
     unittest.main()