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

"""Smoke test of tigertool binary."""

import argparse
import subprocess
import sys


# Script to control tigertail USB-C Mux board.
#
# optional arguments:
#  -h, --help            show this help message and exit
#  -s SERIALNO, --serialno SERIALNO
#                        serial number of board to use
#  -b BUS, --bus BUS     Which rail to log: [vbus|cc1|cc2]
#  --setserialno SETSERIALNO
#                        serial number to set on the board.
#  --check_serial        check serial number set on the board.
#  -m MUX, --mux MUX     mux selection
#  -p, --power           check VBUS
#  -l POWERLOG, --powerlog POWERLOG
#                        log VBUS
#  -r SYSJUMP, --sysjump SYSJUMP
#                        region selection
#  --reboot              reboot tigertail
#  --check_version       check tigertail version


def testCmd(cmd, expected_results):
    """Run command on console, check for success.

    Args:
      cmd: shell command to run.
      expected_results: a list object of strings expected in the result.

    Raises:
      Exception on fail.
    """
    print("run: " + cmd)
    try:
        p = subprocess.run(cmd, shell=True, check=False, capture_output=True)
        output = p.stdout.decode("utf-8")
        error = p.stderr.decode("utf-8")
        assert p.returncode == 0
        for result in expected_results:
            output.index(result)
    except Exception as e:
        print("FAIL")
        print("cmd: " + cmd)
        print("error: " + str(e))
        print("stdout:\n" + output)
        print("stderr:\n" + error)
        print("expected: " + str(expected_results))
        print("RC: " + str(p.returncode))
        raise e


def test_sequence():
    testCmd("./tigertool.py --reboot", ["PASS"])
    testCmd("./tigertool.py --setserialno test", ["PASS"])
    testCmd("./tigertool.py --check_serial", ["test", "PASS"])
    testCmd("./tigertool.py -s test --check_serial", ["test", "PASS"])
    testCmd("./tigertool.py -m A", ["Mux set to A", "PASS"])
    testCmd("./tigertool.py -m B", ["Mux set to B", "PASS"])
    testCmd("./tigertool.py -m off", ["Mux set to off", "PASS"])
    testCmd("./tigertool.py -p", ["PASS"])
    testCmd("./tigertool.py -r rw", ["PASS"])
    testCmd("./tigertool.py -r ro", ["PASS"])
    testCmd("./tigertool.py --check_version", ["RW", "RO", "PASS"])

    print("PASS")


def main(argv):
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
        "-c", "--count", type=int, default=1, help="loops to run"
    )

    opts = parser.parse_args(argv)

    for i in range(1, opts.count + 1):
        print("Iteration: %d" % i)
        test_sequence()


if __name__ == "__main__":
    main(sys.argv[1:])