50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
"""The unittest.TestCase for Server Selection JSON tests."""
|
|
import os
|
|
import os.path
|
|
from . import interface
|
|
from ... import core
|
|
from ... import config
|
|
from ... import utils
|
|
from ...utils import globstar
|
|
from ... import errors
|
|
|
|
|
|
class ServerSelectionJsonTestCase(interface.ProcessTestCase):
|
|
"""Server Selection JSON test case."""
|
|
|
|
REGISTERED_NAME = "server_selection_json_test"
|
|
EXECUTABLE_BUILD_PATH = "build/**/mongo/client/sdam/server_selection_json_test"
|
|
TEST_DIR = os.path.normpath("src/mongo/client/sdam/json_tests/server_selection_tests")
|
|
|
|
def __init__(self, logger, json_test_file, program_options=None):
|
|
"""Initialize the TestCase with the executable to run."""
|
|
interface.ProcessTestCase.__init__(self, logger, "Server Selection Json Test",
|
|
json_test_file)
|
|
|
|
self.program_executable = self._find_executable()
|
|
self.json_test_file = os.path.normpath(json_test_file)
|
|
self.program_options = utils.default_if_none(program_options, {}).copy()
|
|
|
|
def _find_executable(self):
|
|
if config.INSTALL_DIR is not None:
|
|
binary = os.path.join(config.INSTALL_DIR, "server_selection_json_test")
|
|
if os.name == "nt":
|
|
binary += ".exe"
|
|
|
|
if os.path.isfile(binary):
|
|
return binary
|
|
|
|
execs = globstar.glob(self.EXECUTABLE_BUILD_PATH + '.exe')
|
|
if not execs:
|
|
execs = globstar.glob(self.EXECUTABLE_BUILD_PATH)
|
|
if len(execs) != 1:
|
|
raise errors.StopExecution(
|
|
"There must be a single server_selection_json_test binary in {}".format(execs))
|
|
return execs[0]
|
|
|
|
def _make_process(self):
|
|
command_line = [self.program_executable]
|
|
command_line += ["--source-dir", self.TEST_DIR]
|
|
command_line += ["-f", self.json_test_file]
|
|
return core.programs.make_process(self.logger, command_line)
|