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

"""The unittest.TestCase for C++ integration tests."""

from typing import Optional

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


class CPPIntegrationTestCase(interface.ProcessTestCase):
    """A C++ integration test to execute."""

    REGISTERED_NAME = "cpp_integration_test"

    def __init__(
        self,
        logger: logging.Logger,
        program_executables: list[str],
        program_options: Optional[dict] = None,
    ):
        """Initialize the CPPIntegrationTestCase with the executable to run."""

        assert len(program_executables) == 1
        interface.ProcessTestCase.__init__(
            self, logger, "C++ integration test", program_executables[0]
        )

        self.program_executable = program_executables[0]
        self.program_options = utils.default_if_none(program_options, {}).copy()

    def configure(self, fixture, *args, **kwargs):
        """Configure the test case."""
        interface.ProcessTestCase.configure(self, fixture, *args, **kwargs)

        self.program_options["connectionString"] = self.fixture.get_shell_connection_string()

        tls_mode = self.fixture.config.TLS_MODE
        ca_file = self.fixture.config.TLS_CA_FILE
        cert_file = self.fixture.config.SHELL_TLS_CERTIFICATE_KEY_FILE
        shell_grpc = self.fixture.config.SHELL_GRPC

        self.program_options["useEgressGRPC"] = shell_grpc

        if tls_mode:
            self.program_options["tlsMode"] = tls_mode
        if ca_file:
            self.program_options["tlsCAFile"] = ca_file
        if cert_file:
            self.program_options["tlsCertificateKeyFile"] = cert_file

    def _make_process(self):
        return core.programs.generic_program(
            self.logger, [self.program_executable], **self.program_options
        )