58bc053e2e
Co-authored-by: Nick Jefferies <nick.jefferies@mongodb.com> GitOrigin-RevId: 852adc50c1e8956bb1f5d0fc7a41cfd03b889753
61 lines
2.7 KiB
Python
61 lines
2.7 KiB
Python
import logging
|
|
import os
|
|
import unittest
|
|
|
|
from buildscripts.resmokelib import config, configure_resmoke, suitesconfig
|
|
from buildscripts.resmokelib.logging import loggers
|
|
|
|
|
|
class ValidateGeneratedSuites(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
# make sure we have modules set up for the test
|
|
config.MODULES_CONFIG_PATH = os.path.join(
|
|
"buildscripts", "resmokeconfig", "resmoke_modules.yml"
|
|
)
|
|
modules = configure_resmoke._get_module_configs()
|
|
config.MODULES = modules.keys()
|
|
configure_resmoke._set_up_modules()
|
|
|
|
config.CONFIG_DIR = "buildscripts/resmokeconfig"
|
|
cls.matrix_suite_config = suitesconfig.MatrixSuiteConfig()
|
|
loggers.ROOT_EXECUTOR_LOGGER = logging
|
|
|
|
def test_generated_suites(self):
|
|
suite_names = self.matrix_suite_config.get_named_suites()
|
|
|
|
for suite_name in suite_names:
|
|
try:
|
|
suite = self.matrix_suite_config.get_config_obj_and_verify(suite_name)
|
|
self.assertIsNotNone(
|
|
suite,
|
|
msg=f"{suite_name} was not found. This means either MatrixSuiteConfig.get_named_suites() "
|
|
+ "or MatrixSuiteConfig.get_config_obj_and_verify() are not working as intended.",
|
|
)
|
|
except Exception as ex:
|
|
self.fail(repr(ex))
|
|
|
|
def test_stray_generated_files(self):
|
|
suite_names = set(self.matrix_suite_config.get_named_suites())
|
|
for suites_dir in self.matrix_suite_config.get_suites_dirs():
|
|
gen_dir = os.path.join(suites_dir, "generated_suites")
|
|
generated_files = os.listdir(gen_dir)
|
|
for filename in generated_files:
|
|
(suite_name, ext) = os.path.splitext(filename)
|
|
self.assertEqual(
|
|
ext, ".yml", msg=f"{filename} has the wrong file extension, expected `.yml`"
|
|
)
|
|
expected_mapping_file = os.path.join(suites_dir, f"mappings/{suite_name}.yml")
|
|
self.assertIn(
|
|
suite_name,
|
|
suite_names,
|
|
msg=f"{filename} does not have a correlated mapping file. Make a mapping file or delete it."
|
|
f"You have a generated file {filename} that does not have a corresponding mapping file {expected_mapping_file}. "
|
|
+ "If you have added a non matrix suite to resmokeconfig/matrix_suites/generated_suites, move it to the resmokeconfig/suites."
|
|
+ " If you have removed the mapping file be sure to rerun buildscripts/resmoke.py generate-matrix-suites.",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|