"""The unittest.TestCase for mongos --test."""

from buildscripts.resmokelib import config
from buildscripts.resmokelib import core
from buildscripts.resmokelib import utils
from buildscripts.resmokelib.testing.testcases import interface


class MongosTestCase(interface.ProcessTestCase):
    """A TestCase which runs a mongos binary with the given parameters."""

    REGISTERED_NAME = "mongos_test"

    def __init__(self, logger, mongos_options):
        """Initialize the mongos test and saves the options."""

        self.mongos_executable = utils.default_if_none(config.MONGOS_EXECUTABLE,
                                                       config.DEFAULT_MONGOS_EXECUTABLE)
        # Use the executable as the test name.
        interface.ProcessTestCase.__init__(self, logger, "mongos test", self.mongos_executable)
        self.options = mongos_options.copy()

    def configure(self, fixture, *args, **kwargs):
        """Ensure the --test option is present in the mongos options."""

        interface.ProcessTestCase.configure(self, fixture, *args, **kwargs)
        # Always specify test option to ensure the mongos will terminate.
        if "test" not in self.options:
            self.options["test"] = ""

    def _make_process(self):
        return core.programs.mongos_program(self.logger, self.fixture.job_num,
                                            executable=self.mongos_executable, **self.options)