@@ -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
751749if __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