summaryrefslogtreecommitdiff
path: root/buildscripts/resmokelib/testing/testcases/mongos_test.py
blob: ef29069a7ebf8190fc1f15f292d3bc1703a59bb7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
"""
unittest.TestCase for mongos --test.
"""

from __future__ import absolute_import

from . import interface
from ... import config
from ... import core
from ... import utils


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):
        """
        Initializes 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):
        """
        Ensures 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,
                                            executable=self.mongos_executable,
                                            **self.options)