diff options
Diffstat (limited to 'buildscripts/resmokelib/testing/fixtures/masterslave.py')
| -rw-r--r-- | buildscripts/resmokelib/testing/fixtures/masterslave.py | 162 |
1 files changed, 0 insertions, 162 deletions
diff --git a/buildscripts/resmokelib/testing/fixtures/masterslave.py b/buildscripts/resmokelib/testing/fixtures/masterslave.py deleted file mode 100644 index d63205363ef..00000000000 --- a/buildscripts/resmokelib/testing/fixtures/masterslave.py +++ /dev/null @@ -1,162 +0,0 @@ -""" -Master/slave fixture for executing JSTests against. -""" - -from __future__ import absolute_import - -import os.path - -import pymongo -import pymongo.errors - -from . import interface -from . import standalone -from ... import config -from ... import utils - - -class MasterSlaveFixture(interface.ReplFixture): - """ - Fixture which provides JSTests with a master/slave deployment to - run against. - """ - - def __init__(self, - logger, - job_num, - mongod_executable=None, - mongod_options=None, - master_options=None, - slave_options=None, - dbpath_prefix=None, - preserve_dbpath=False): - - interface.ReplFixture.__init__(self, logger, job_num, dbpath_prefix=dbpath_prefix) - - if "dbpath" in mongod_options: - raise ValueError("Cannot specify mongod_options.dbpath") - - self.mongod_executable = mongod_executable - self.mongod_options = utils.default_if_none(mongod_options, {}) - self.master_options = utils.default_if_none(master_options, {}) - self.slave_options = utils.default_if_none(slave_options, {}) - self.preserve_dbpath = preserve_dbpath - - self._dbpath_prefix = os.path.join(self._dbpath_prefix, config.FIXTURE_SUBDIR) - - self.master = None - self.slave = None - - def setup(self): - if self.master is None: - self.master = self._new_mongod_master() - self.master.setup() - - if self.slave is None: - self.slave = self._new_mongod_slave() - self.slave.setup() - - def await_ready(self): - self.master.await_ready() - self.slave.await_ready() - - # Do a replicated write to ensure that the slave has finished with its initial sync before - # starting to run any tests. - client = self.master.mongo_client() - - # Keep retrying this until it times out waiting for replication. - def insert_fn(remaining_secs): - remaining_millis = int(round(remaining_secs * 1000)) - write_concern = pymongo.WriteConcern(w=2, wtimeout=remaining_millis) - coll = client.resmoke.get_collection("await_ready", write_concern=write_concern) - coll.insert_one({"awaiting": "ready"}) - - try: - self.retry_until_wtimeout(insert_fn) - except pymongo.errors.WTimeoutError: - self.logger.info("Replication of write operation timed out.") - raise - - def _do_teardown(self): - running_at_start = self.is_running() - success = True # Still a success if nothing is running. - - if not running_at_start: - self.logger.info( - "Master-slave deployment was expected to be running in _do_teardown(), but wasn't.") - - if self.slave is not None: - if running_at_start: - self.logger.info("Stopping slave...") - - success = self.slave.teardown() - - if running_at_start: - self.logger.info("Successfully stopped slave.") - - if self.master is not None: - if running_at_start: - self.logger.info("Stopping master...") - - success = self.master.teardown() and success - - if running_at_start: - self.logger.info("Successfully stopped master.") - - return success - - def is_running(self): - return (self.master is not None and self.master.is_running() and - self.slave is not None and self.slave.is_running()) - - def get_primary(self): - return self.master - - def get_secondaries(self): - return [self.slave] - - def _new_mongod(self, mongod_logger, mongod_options): - """ - Returns a standalone.MongoDFixture with the specified logger and - options. - """ - return standalone.MongoDFixture(mongod_logger, - self.job_num, - mongod_executable=self.mongod_executable, - mongod_options=mongod_options, - preserve_dbpath=self.preserve_dbpath) - - def _new_mongod_master(self): - """ - Returns a standalone.MongoDFixture configured to be used as the - master of a master-slave deployment. - """ - - mongod_logger = self.logger.new_fixture_node_logger("master") - - mongod_options = self.mongod_options.copy() - mongod_options.update(self.master_options) - mongod_options["master"] = "" - mongod_options["dbpath"] = os.path.join(self._dbpath_prefix, "master") - return self._new_mongod(mongod_logger, mongod_options) - - def _new_mongod_slave(self): - """ - Returns a standalone.MongoDFixture configured to be used as the - slave of a master-slave deployment. - """ - - mongod_logger = self.logger.new_fixture_node_logger("slave") - - mongod_options = self.mongod_options.copy() - mongod_options.update(self.slave_options) - mongod_options["slave"] = "" - mongod_options["source"] = self.master.get_internal_connection_string() - mongod_options["dbpath"] = os.path.join(self._dbpath_prefix, "slave") - return self._new_mongod(mongod_logger, mongod_options) - - def get_internal_connection_string(self): - return self.master.get_internal_connection_string() - - def get_driver_connection_url(self): - return self.master.get_driver_connection_url() |