bazel_rules_fuchsia/fuchsia/tools/run_test_component.py

#!/usr/bin/env python3
# Copyright 2022 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import subprocess

from fuchsia_task_lib import *


class FuchsiaTaskRunTestComponent(FuchsiaTask):
    def parse_args(self, parser: ScopedArgumentParser) -> argparse.Namespace:
        '''Parses arguments.'''

        parser.add_argument(
            '--ffx',
            type=parser.path_arg(),
            help='A path to the ffx tool.',
            required=True,
        )
        parser.add_argument(
            '--url',
            type=str,
            help='The full component url.',
            required=True,
        )
        parser.add_argument(
            '--target',
            help='Optionally specify the target fuchsia device.',
            required=False,
            scope=ArgumentScope.GLOBAL,
        )
        return parser.parse_args()


    def run(self, parser: ScopedArgumentParser) -> None:
        args = self.parse_args(parser)
        ffx = [args.ffx] + (['--target', args.target] if args.target else [])

        try:
            subprocess.check_call([
                *ffx,
                'test',
                'run',
                args.url,
            ])
        except subprocess.CalledProcessError as e:
            if e.returncode != 1:
                raise e
            raise TaskExecutionException(f'Test Failures!', is_caught_failure=True)


if __name__ == '__main__':
    FuchsiaTaskRunTestComponent.main()