mongo/buildscripts/resmokelib/testing/testcases/pytest.py

"""The unittest.TestCase for Python unittests."""

import os
import sys

from buildscripts.resmokelib import core, logging
from buildscripts.resmokelib.testing.testcases import interface


class PyTestCase(interface.ProcessTestCase):
    """A python test to execute."""

    REGISTERED_NAME = "py_test"

    def __init__(self, logger: logging.Logger, py_filenames: list[str]):
        """Initialize PyTestCase."""
        assert len(py_filenames) == 1
        interface.ProcessTestCase.__init__(self, logger, "PyTest", py_filenames[0])

    def _make_process(self):
        program_options = {}
        interface.append_process_tracking_options(program_options, self._id)
        return core.programs.generic_program(
            self.logger, [sys.executable, "-m", "unittest", self.test_module_name], program_options
        )

    @property
    def test_module_name(self):
        """Get the dotted module name from the path."""
        return os.path.splitext(self.test_name)[0].replace(os.sep, ".")