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])