def test_build_sharded_cluster_multiversion(self): parser.set_run_options() fixture_config = { "mongod_options": { SET_PARAMS: { "enableTestCommands": 1 } }, "configsvr_options": { "num_nodes": 2 }, "num_shards": 2, "num_rs_nodes_per_shard": 2, "mixed_bin_versions": "new_old_old_new", "old_bin_version": "last_lts", } sharded_cluster = under_test.make_fixture(self.fixture_class_name, self.mock_logger, self.job_num, **fixture_config) self.assertEqual(len(sharded_cluster.configsvr.nodes), 2) self.assertEqual(len(sharded_cluster.shards), 2) self.assertEqual(len(sharded_cluster.shards[0].nodes), 2) self.assertEqual(len(sharded_cluster.shards[1].nodes), 2) self.assertEqual(len(sharded_cluster.mongos), 1) from buildscripts.resmokelib import multiversionconstants # configsvr nodes are always latest self.assertEqual(sharded_cluster.configsvr.nodes[0].mongod_executable, config.DEFAULT_MONGOD_EXECUTABLE) self.assertEqual(sharded_cluster.configsvr.nodes[1].mongod_executable, config.DEFAULT_MONGOD_EXECUTABLE) # 1st repl set nodes are latest and last-lts (new_old) self.assertEqual(sharded_cluster.shards[0].nodes[0].mongod_executable, config.DEFAULT_MONGOD_EXECUTABLE) self.assertEqual(sharded_cluster.shards[0].nodes[1].mongod_executable, multiversionconstants.LAST_LTS_MONGOD_BINARY) self.assertEqual(sharded_cluster.shards[0].fcv, multiversionconstants.LAST_LTS_FCV) # 2st repl set nodes are last-lts and latest (old_new) self.assertEqual(sharded_cluster.shards[1].nodes[0].mongod_executable, multiversionconstants.LAST_LTS_MONGOD_BINARY) self.assertEqual(sharded_cluster.shards[1].nodes[1].mongod_executable, config.DEFAULT_MONGOD_EXECUTABLE) self.assertEqual(sharded_cluster.shards[0].fcv, multiversionconstants.LAST_LTS_FCV) # mongos is last-lts self.assertEqual(sharded_cluster.mongos[0].mongos_executable, multiversionconstants.LAST_LTS_MONGOS_BINARY)
def test_build_sharded_cluster_simple(self): parser.set_run_options() fixture_config = { "mongod_options": { SET_PARAMS: { "enableTestCommands": 1 } } } sharded_cluster = under_test.make_fixture(self.fixture_class_name, self.mock_logger, self.job_num, **fixture_config) self.assertEqual(len(sharded_cluster.configsvr.nodes), 1) self.assertEqual(len(sharded_cluster.shards), 1) self.assertEqual(len(sharded_cluster.shards[0].nodes), 1) self.assertEqual(len(sharded_cluster.mongos), 1) from buildscripts.resmokelib import multiversionconstants self.assertEqual(sharded_cluster.shards[0].fcv, multiversionconstants.LATEST_FCV)
def test_build_sharded_cluster_with_feature_flags(self): ff_name = "featureFlagDummy" parser.set_run_options(f"--additionalFeatureFlags={ff_name}") fixture_config = { "mongod_options": { SET_PARAMS: { "enableTestCommands": 1 } } } sharded_cluster = under_test.make_fixture(self.fixture_class_name, self.mock_logger, self.job_num, **fixture_config) self.assertEqual(len(sharded_cluster.configsvr.nodes), 1) self.assertEqual(len(sharded_cluster.shards), 1) self.assertEqual(len(sharded_cluster.shards[0].nodes), 1) self.assertEqual(len(sharded_cluster.mongos), 1) from buildscripts.resmokelib import multiversionconstants self.assertEqual(sharded_cluster.shards[0].fcv, multiversionconstants.LATEST_FCV) # feature flags are set self.assertIn( ff_name, sharded_cluster.configsvr.nodes[0].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.configsvr.nodes[0].mongod_options[SET_PARAMS] [ff_name]) self.assertIn( ff_name, sharded_cluster.shards[0].nodes[0].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.shards[0].nodes[0].mongod_options[SET_PARAMS] [ff_name]) self.assertIn(ff_name, sharded_cluster.mongos[0].mongos_options[SET_PARAMS]) self.assertTrue( sharded_cluster.mongos[0].mongos_options[SET_PARAMS][ff_name])
def make_fixture(self, class_name, logger, job_num, *args, **kwargs): """Build fixtures by calling builder API.""" return _builder.make_fixture(class_name, logger, job_num, *args, **kwargs)
def test_build_sharded_cluster_multiversion_with_feature_flags(self): ff_name = "featureFlagDummy" parser.set_run_options(f"--additionalFeatureFlags={ff_name}") fixture_config = { "mongod_options": { SET_PARAMS: { "enableTestCommands": 1 } }, "configsvr_options": { "num_nodes": 2 }, "num_shards": 2, "num_rs_nodes_per_shard": 2, "mixed_bin_versions": "new_old_old_new", "old_bin_version": "last_lts", } sharded_cluster = under_test.make_fixture(self.fixture_class_name, self.mock_logger, self.job_num, **fixture_config) self.assertEqual(len(sharded_cluster.configsvr.nodes), 2) self.assertEqual(len(sharded_cluster.shards), 2) self.assertEqual(len(sharded_cluster.shards[0].nodes), 2) self.assertEqual(len(sharded_cluster.shards[1].nodes), 2) self.assertEqual(len(sharded_cluster.mongos), 1) # feature flags are set on new versions self.assertIn( ff_name, sharded_cluster.configsvr.nodes[0].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.configsvr.nodes[0].mongod_options[SET_PARAMS] [ff_name]) self.assertIn( ff_name, sharded_cluster.configsvr.nodes[1].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.configsvr.nodes[1].mongod_options[SET_PARAMS] [ff_name]) self.assertIn( ff_name, sharded_cluster.shards[0].nodes[0].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.shards[0].nodes[0].mongod_options[SET_PARAMS] [ff_name]) self.assertIn( ff_name, sharded_cluster.shards[1].nodes[1].mongod_options[SET_PARAMS]) self.assertTrue( sharded_cluster.shards[1].nodes[1].mongod_options[SET_PARAMS] [ff_name]) # feature flags are NOT set on old versions self.assertNotIn( ff_name, sharded_cluster.shards[0].nodes[1].mongod_options[SET_PARAMS]) self.assertNotIn( ff_name, sharded_cluster.shards[1].nodes[0].mongod_options[SET_PARAMS]) self.assertNotIn(ff_name, sharded_cluster.mongos[0].mongos_options[SET_PARAMS])