test_parse_platform

Complexity

Total Complexity

7

Size/Duplication

Total Lines	54
Duplicated Lines	0 %

Importance

Changes

0

1 Function

Rating	Name	Duplication	Size	Complexity
B	main()	0	34	7

#!/usr/bin/env python

import os
import sys

import ssg.build_remediations
import ssg.constants
import ssg.fixes
import ssg.rules
import ssg.utils
import ssg.yaml

REMEDIATION_LANGS = list(ssg.build_remediations.REMEDIATION_TO_EXT_MAP)


def main():
    """
    Walk through all known products in the ssg root specified in sys.argv[1],
    and ensure that all fixes in all rule directories are parsable under
    ssg.fixes.parse_platform(...).
    """

    if len(sys.argv) != 2:
        print("Error! Must supply only path to root of ssg directory",
              file=sys.stderr)
        sys.exit(1)

    ssg_root = sys.argv[1]

    guide_dirs = set()
    for product in ssg.constants.product_directories:
        product_dir = os.path.join(ssg_root, product)
        product_yaml_path = os.path.join(product_dir, "product.yml")
        product_yaml = ssg.yaml.open_raw(product_yaml_path)

        guide_dir = os.path.join(product_dir, product_yaml['benchmark_root'])
        if guide_dir in guide_dirs:
            continue

        for rule_dir in ssg.rules.find_rule_dirs(guide_dir):
            for lang in REMEDIATION_LANGS:
                for fix in ssg.rules.get_rule_dir_remediations(rule_dir, lang):
                    fix_contents = ssg.utils.read_file_list(fix)
                    results = ssg.fixes.parse_platform(fix_contents)

                    assert results is not None
                    assert isinstance(results, int)

        guide_dirs.add(guide_dir)


if __name__ == "__main__":
    main()