brillo_aue2e_test_setup.py: Validate mandatory flag by the parser.
BUG=b:25395612
TEST=Script fails if flag not provided
Change-Id: Id4a1e404c0dd218dbff7fb87d8f3e73b89987328
Reviewed-on: https://chromium-review.googlesource.com/310552
Commit-Ready: Gilad Arnold <garnold@chromium.org>
Tested-by: Gilad Arnold <garnold@chromium.org>
Reviewed-by: Simran Basi <sbasi@chromium.org>
diff --git a/site_utils/brillo_aue2e_test_setup.py b/site_utils/brillo_aue2e_test_setup.py
index ff7d1aa..bce950d 100755
--- a/site_utils/brillo_aue2e_test_setup.py
+++ b/site_utils/brillo_aue2e_test_setup.py
@@ -60,7 +60,7 @@
@param parser: argparse.ArgumentParser of the script.
"""
- parser.add_argument('-t', '--target_payload', metavar='SPEC',
+ parser.add_argument('-t', '--target_payload', metavar='SPEC', required=True,
help='Stage a target payload. This can either be a '
'path to a local payload file, or take the form '
'"%s:DST_IMAGE[:SRC_IMAGE]", in which case a '
@@ -76,16 +76,6 @@
brillo_common.setup_test_action_parser(parser)
-def validate_args(parser, args):
- """Validate command-line arguments.
-
- @param parser: argparse.ArgumentParser of the script.
- @param args: The parsed arguments.
- """
- if not args.target_payload:
- parser.error('Must provide --target_payload.')
-
-
def get_stage_rel_path(stage_file):
"""Returns the relative stage path for remote file.
@@ -270,7 +260,7 @@
args = brillo_common.parse_args(
'Set up Moblab for running Brillo AU end-to-end test, then launch '
'the test (unless otherwise requested).',
- setup_parser=setup_parser, validate_args=validate_args)
+ setup_parser=setup_parser)
moblab, devserver_port = brillo_common.get_moblab_and_devserver_port(
args.moblab_host)