Function main

misc/python/materialize/cli/lint_test_flags.py:28–69

()

def main() -> int:
    parser = argparse.ArgumentParser(
        prog="lint-tests-flags",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        description="Check that new configs are used in parallel workload and other tests",
    )
    parser.parse_args()

    configs = []

    for path in Path("src").rglob("*.rs"):
        if path in [Path("src/dyncfg/src/lib.rs"), Path("src/dyncfg-file/src/lib.rs")]:
            continue  # contains tests
        with path.open(encoding="utf-8") as file:
            content = file.read()
            if matches := CONFIG_REGEX.findall(content):
                configs.extend(matches)

    action = FlipFlagsAction(random.Random(), None)
    parallel_workload_known_flags = set(action.flags_with_values).union(
        action.uninteresting_flags
    )
    mzcompose_known_flags = set(
        get_default_system_parameters(MzVersion.parse_cargo()).keys()
    ).union(UNINTERESTING_SYSTEM_PARAMETERS)
    found = False

    for config in configs:
        if config not in parallel_workload_known_flags:
            print(
                f'Configuration flag "{config}" seems to have been introduced/changed. Make sure parallel-workload\'s FlipFlagsAction in misc/python/materialize/parallel_workload/action.py knows about some valid values for it'
            )
            found = True

    for config in configs:
        if config not in mzcompose_known_flags:
            print(
                f'Configuration flag "{config}" seems to have been introduced/changed. Make sure get_variable_system_parameters/get_minimal_system_parameters/UNINTERESTING_SYSTEM_PARAMETERS in misc/python/materialize/mzcompose/__init__.py knows about some valid values for it'
            )
            found = True

    return int(found)


if __name__ == "__main__":