def test_generate_evg_buildvariant_one_base_variant(self):
evg_conf_mock = get_evergreen_config()
base_variant = "enterprise-rhel-62-64-bit-inmem"
generated_variant = "enterprise-rhel-62-64-bit-inmem-required"
burn_in_tags_gen_variant = "enterprise-rhel-62-64-bit"
shrub_config = Configuration()
under_test._generate_evg_build_variant(shrub_config, base_variant,
generated_variant,
burn_in_tags_gen_variant,
evg_conf_mock)
expected_variant_data = get_evergreen_config().get_variant(
base_variant)
generated_buildvariants = shrub_config.to_map()["buildvariants"]
self.assertEqual(len(generated_buildvariants), 1)
generated_build_variant = generated_buildvariants[0]
self.assertEqual(generated_build_variant["name"], generated_variant)
self.assertEqual(generated_build_variant["modules"],
expected_variant_data.modules)
generated_expansions = generated_build_variant["expansions"]
burn_in_bypass_expansion_value = generated_expansions.pop(
"burn_in_bypass")
self.assertEqual(burn_in_bypass_expansion_value,
burn_in_tags_gen_variant)
self.assertEqual(generated_expansions,
expected_variant_data.expansions)
def test_generate_evg_buildvariant_one_base_variant(self):
evg_conf_mock = get_evergreen_config()
base_variant = "enterprise-rhel-62-64-bit-inmem"
generated_variant = "enterprise-rhel-62-64-bit-inmem-required"
burn_in_tags_gen_variant = "enterprise-rhel-62-64-bit"
variant = evg_conf_mock.get_variant(base_variant)
build_variant = under_test._generate_evg_build_variant(variant, generated_variant,
burn_in_tags_gen_variant)
generated_build_variant = build_variant.as_dict()
self.assertEqual(generated_build_variant["name"], generated_variant)
self.assertEqual(generated_build_variant["modules"], variant.modules)
generated_expansions = generated_build_variant["expansions"]
burn_in_bypass_expansion_value = generated_expansions.pop("burn_in_bypass")
self.assertEqual(burn_in_bypass_expansion_value, burn_in_tags_gen_variant)
self.assertEqual(generated_expansions, variant.expansions)
