# Copyright 2021 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from autotest_lib.server import utils

AUTHOR = "mruthven"
NAME = "firmware_Cr50Keygen.ecc"
PURPOSE = "Verify ecc key generation."
ATTRIBUTES = "suite:faft_cr50_prepvt, suite:faft_cr50_pvt"
DEPENDENCIES = "servo_state:WORKING"
TIME = "SHORT"
TEST_TYPE = "server"
PY_VERSION = 3

DOC = """Verify ecc keygen."""

if 'args_dict' not in locals():
    args_dict = {}

args_dict.update(utils.args_to_dict(args))
servo_args = hosts.CrosHost.get_servo_arguments(args_dict)

def run(machine):
    host = hosts.create_host(machine, servo_args=servo_args)

    iterations = int(args_dict.get("iterations", 1))

    job.run_test("firmware_Cr50Keygen", host=host, cmdline_args=args,
                 key_type="ecc", iterations=iterations)

parallel_simple(run, machines)