def setup_cli()

in antlir/vm/vmtest.py [0:0]

• 45 lines of code
• 1 McCabe index (conditional complexity)

    def setup_cli(cls, parser):
        super(VMTestExecOpts, cls).setup_cli(parser)

        parser.add_argument(
            "--devel-layer",
            action="store_true",
            default=False,
            help="Provide the kernel devel layer as a mount to the booted VM",
        )
        parser.add_argument(
            "--setenv",
            action="append",
            default=[],
            help="Specify an environment variable to pass to the test "
            "in the form NAME=VALUE",
        )
        parser.add_argument(
            "--test-binary",
            type=normalize_buck_path,
            help="Path to the actual test binary that will be invoked.  This "
            "is used to discover tests before they are executed inside the VM",
            required=True,
        )
        parser.add_argument(
            "--test-binary-wrapper",
            type=normalize_buck_path,
            help="Path to the test binary wrapper",
            required=True,
        )
        parser.add_argument(
            "--test-type",
            help="The type of test being executed, this is populated "
            "by the .bzl that wraps the test.",
            required=True,
            choices=_TEST_TYPE_TO_WRAP_CMD.keys(),
        )

        list_group = parser.add_mutually_exclusive_group()
        # For gtest
        list_group.add_argument(
            "--gtest_list_tests",
            action="store_true",
        )
        # For python tests
        list_group.add_argument(
            "--list-tests",
            type=Path.from_argparse,
        )
        # For rust tests
        list_group.add_argument("--list", action="store_true", dest="list_rust")