utils/run_fleet_provisioning_sample.py

# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0.

import argparse
import uuid
import os
import sys
import run_in_ci
import ci_iot_thing


def main():
    argument_parser = argparse.ArgumentParser(
        description="Run Fleet Provisioning sample in CI")
    argument_parser.add_argument("--file", required=True, help="Configuration file to pull CI data from")
    argument_parser.add_argument(
        "--input-uuid", required=False,
        help="Use specified UUID for thing name instead of a random one")
    argument_parser.add_argument(
        "--thing-name-prefix", required=False, default="",
        help="Prefix for a thing name, should be the same that Fleet Provisioning template uses")
    argument_parser.add_argument(
        "--region", required=False, default="us-east-1", help="The name of the region to use")
    parsed_commands = argument_parser.parse_args()

    cfg_file = parsed_commands.file
    thing_uuid = parsed_commands.input_uuid if parsed_commands.input_uuid else str(uuid.uuid4())

    try:
        # Perform fleet provisioning. If it's successful, a newly created thing should appear.
        test_result = run_in_ci.setup_and_launch(cfg_file, thing_uuid)
    except Exception as e:
        print(f"Fleet provisioning failed: {e}")
        test_result = -1

    try:
        # Delete a thing created by fleet provisioning.
        # NOTE We want to try to delete thing even if test was unsuccessful.
        thing_name = parsed_commands.thing_name_prefix + thing_uuid
        ci_iot_thing.delete_iot_thing(thing_name, parsed_commands.region)
    except Exception as e:
        print(f"Deleting IoT thing failed, probably it was not created by Fleet Provisioning: {e}")
        test_result = -1

    if test_result != 0:
        sys.exit(-1)

if __name__ == "__main__":
    main()