a24a7548b8
This makes the sorting behavior more uniform: functions and macros are always sorted (separately), not only when merging. This changes the sort order used for functions and other things sorted by their symbol names. Symbols are sorted alphabetically without regard to leading underscores, and then for identifiers that differ only in the number of leading underscores, the fewer underscores the earlier in the sort order. For the functions declared in a generated header, adjacent names with and without underscores will be grouped together without blank lines. This is implemented by factoring the name field, equality, and sorting support out of the various entity classes into a new common superclass (hdrgen.Symbol). This uncovered YAML's requirement to quote the string "NULL" to avoid pyyaml parsing it as None (equivalent to Javascript null) rather than a string.
105 lines
3.5 KiB
Python
105 lines
3.5 KiB
Python
import argparse
|
|
import subprocess
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
class TestHeaderGenIntegration(unittest.TestCase):
|
|
def setUp(self):
|
|
self.output_dir = TestHeaderGenIntegration.output_dir
|
|
self.source_dir = Path(__file__).parent
|
|
self.main_script = self.source_dir.parent / "main.py"
|
|
self.maxDiff = 80 * 100
|
|
|
|
def run_script(self, yaml_file, output_file, entry_points=[], switches=[]):
|
|
command = [
|
|
"python3",
|
|
str(self.main_script),
|
|
str(yaml_file),
|
|
"--output",
|
|
str(output_file),
|
|
] + switches
|
|
|
|
for entry_point in entry_points:
|
|
command.extend(["--entry-point", entry_point])
|
|
|
|
result = subprocess.run(
|
|
command,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
print("STDOUT:", result.stdout)
|
|
print("STDERR:", result.stderr)
|
|
result.check_returncode()
|
|
|
|
def compare_files(self, generated_file, expected_file):
|
|
with generated_file.open("r") as gen_file:
|
|
gen_content = gen_file.read()
|
|
with expected_file.open("r") as exp_file:
|
|
exp_content = exp_file.read()
|
|
|
|
self.assertEqual(gen_content, exp_content)
|
|
|
|
def test_generate_header(self):
|
|
yaml_file = self.source_dir / "input/test_small.yaml"
|
|
expected_output_file = self.source_dir / "expected_output/test_header.h"
|
|
output_file = self.output_dir / "test_small.h"
|
|
entry_points = {"func_b", "func_a", "func_c", "func_d", "func_e"}
|
|
|
|
self.run_script(yaml_file, output_file, entry_points)
|
|
|
|
self.compare_files(output_file, expected_output_file)
|
|
|
|
def test_generate_subdir_header(self):
|
|
yaml_file = self.source_dir / "input" / "subdir" / "test.yaml"
|
|
expected_output_file = self.source_dir / "expected_output" / "subdir" / "test.h"
|
|
output_file = self.output_dir / "subdir" / "test.h"
|
|
self.run_script(yaml_file, output_file)
|
|
self.compare_files(output_file, expected_output_file)
|
|
|
|
def test_custom_license_and_standards(self):
|
|
yaml_file = self.source_dir / "input" / "custom.yaml"
|
|
expected_output_file = self.source_dir / "expected_output" / "custom.h"
|
|
output_file = self.output_dir / "custom.h"
|
|
self.run_script(yaml_file, output_file)
|
|
self.compare_files(output_file, expected_output_file)
|
|
|
|
def test_generate_json(self):
|
|
yaml_file = self.source_dir / "input/test_small.yaml"
|
|
expected_output_file = self.source_dir / "expected_output/test_small.json"
|
|
output_file = self.output_dir / "test_small.json"
|
|
|
|
self.run_script(yaml_file, output_file, switches=["--json"])
|
|
|
|
self.compare_files(output_file, expected_output_file)
|
|
|
|
def test_sorting(self):
|
|
yaml_file = self.source_dir / "input" / "sorting.yaml"
|
|
expected_output_file = self.source_dir / "expected_output" / "sorting.h"
|
|
output_file = self.output_dir / "sorting.h"
|
|
self.run_script(yaml_file, output_file)
|
|
self.compare_files(output_file, expected_output_file)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="TestHeaderGenIntegration arguments")
|
|
parser.add_argument(
|
|
"--output_dir",
|
|
type=Path,
|
|
help="Output directory for generated headers",
|
|
required=True,
|
|
)
|
|
args, remaining_argv = parser.parse_known_args()
|
|
|
|
TestHeaderGenIntegration.output_dir = args.output_dir
|
|
|
|
sys.argv[1:] = remaining_argv
|
|
|
|
unittest.main()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|