Skip to content

Commit 0c6a7d8

Browse files
committed
tools/psoc6/run_test_plan.py: WIP. FIXUP.
Signed-off-by: jaenrig-ifx <enriquezgarcia.external@infineon.com>
1 parent 9d98a2b commit 0c6a7d8

1 file changed

Lines changed: 22 additions & 25 deletions

File tree

tools/psoc6/run_test_plan.py

Lines changed: 22 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -660,70 +660,68 @@ def get_test_list(self, test_name_list=[]):
660660
return test_list
661661

662662

663-
class TestPlanParser:
663+
class TestPlanCLI:
664664
def __init__(self):
665-
self.tests_parser = argparse.ArgumentParser(description="MicroPython test suites runner.")
666-
self.tests_parser.add_argument(
667-
"test_suite", nargs="*", type=str, help="Test suite to run."
668-
)
669-
self.tests_parser.add_argument(
665+
self.parser = argparse.ArgumentParser(description="MicroPython test suites runner.")
666+
self.parser.add_argument("test_suite", nargs="*", type=str, help="Test suite to run.")
667+
self.parser.add_argument(
670668
"--test-plan", type=str, default=None, help="Path to the test plan file."
671669
)
672-
self.tests_parser.add_argument(
670+
self.parser.add_argument(
673671
"--hil-devs", type=str, default=None, help="Path to the HIL devices file."
674672
)
675-
self.tests_parser.add_argument(
673+
self.parser.add_argument(
676674
"-b",
677675
"--board",
678676
type=str,
679677
default=None,
680678
help="Test board name (only used with --hil-devs).",
681679
)
682-
self.tests_parser.add_argument(
680+
self.parser.add_argument(
683681
"-d",
684682
"--dut-port",
685683
type=str,
686684
default=None,
687685
help="Device under test port. Default is /dev/ttyACM0.",
688686
)
689-
self.tests_parser.add_argument(
687+
self.parser.add_argument(
690688
"-s",
691689
"--stub-port",
692690
type=str,
693691
default=None,
694692
help="Stub device port. Default is /dev/ttyACM1.",
695693
)
696-
self.tests_parser.add_argument(
694+
self.parser.add_argument(
697695
"--max-retries",
698696
type=int,
699697
default=0,
700698
help="Maximum number of retries for failed tests.",
701699
)
702-
self.tests_parser.add_argument(
700+
self.parser.add_argument(
703701
"--mpy-root-dir",
704702
type=str,
705703
default=None,
706704
help="Path to the root of the MicroPython repository. Default is two levels up from this script.",
707705
)
708706

709707
def parse(self):
710-
args = self.tests_parser.parse_args()
708+
args = self.parser.parse_args()
711709
args = self.set_validate_args(args)
712710
return args
713711

714712
def set_validate_args(self, args):
715713
if args.hil_devs:
716714
args.hil_devs = os.path.abspath(args.hil_devs)
717715
if args.board is None:
718-
self.tests_parser.error("--board is required when --hil-devs is provided")
716+
self.parser.error("--board is required when --hil-devs is provided")
719717

720718
if args.dut_port or args.stub_port:
721-
self.tests_parser.error(
719+
self.parser.error(
722720
"--dut-port and --stub-port are not supported when --hil-devs is provided"
723721
)
724722
else:
725723
if args.board is not None:
726-
self.tests_parser.error("--hil-devs is required when --board is provided")
724+
self.parser.error("--hil-devs is required when --board is provided")
727725

728726
# If the ports are not provide, the default values are set.
729727
if args.dut_port is None:
@@ -749,14 +747,13 @@ def set_validate_args(self, args):
749747

750748

751749
if __name__ == "__main__":
752-
test_parser = TestPlanParser()
753-
parser_args = test_parser.parse()
754-
print(parser_args)
750+
test_plan_cli = TestPlanCLI()
751+
cli_args = test_plan_cli.parse()
755752
test_plan_runner = TestPlanRunner(
756-
parser_args.test_plan,
757-
parser_args.hil_devs,
758-
parser_args.max_retries,
759-
parser_args.dut_port,
760-
parser_args.stub_port,
753+
cli_args.test_plan,
754+
cli_args.hil_devs,
755+
cli_args.max_retries,
756+
cli_args.dut_port,
757+
cli_args.stub_port,
761758
)
762-
test_plan_runner.run(parser_args.board, parser_args.test_suite)
759+
test_plan_runner.run(cli_args.board, cli_args.test_suite)

0 commit comments

Comments
 (0)