#!/bin/bash
set -e

function integration_tests() {
    m -j citadel_integration_tests || return 1
    adb shell stop || return 1
    adb sync || return 1
    adb shell start || return 1
    adb exec-out \
        '/vendor/bin/hw/citadel_integration_tests --release-tests' || return 1
}

# TODO: add AVB / Weaver / Keymaster VTS / CTS tests with filters here.

function oem_lock_vts_tests() {
    atest VtsHalOemLockV1_0TargetTest || return 1
}

function keymaster_cts_tests() {
    return 0
}

function keymaster_vts_tests() {
    return 0
}

function weaver_cts_tests() {
  # These CTS tests make a lot of use of Weaver by enrolling and changing
  # credentials. Add omre if you come across them.
  atest com.android.cts.devicepolicy.ManagedProfileTest\#testLockNowWithKeyEviction || return 1
  atest com.android.cts.devicepolicy.DeviceAdminHostSideTestApi24 || return 1
}

function weaver_vts_tests() {
    atest VtsHalWeaverV1_0TargetTest || return 1
}

function auth_secret_vts_tests() {
    atest VtsHalAuthSecretV1_0TargetTest || return 1
}

function pay_cts_tests() {
  :
    # TODO(ngm): uncomment once these are working.
    # runtest --path \
    #     cts/tests/tests/keystore/src/android/keystore/cts/ImportWrappedKeyTest.java || return 1
}

# TODO: add any other tests

if [ -z "${TARGET_PRODUCT:-}" ]; then
  echo "You need to run the Android setup stuff first"
  exit 1
fi

for t in integration_tests \
	     oem_lock_vts_tests \
	     keymaster_cts_tests \
	     keymaster_vts_tests \
	     weaver_cts_tests \
	     weaver_vts_tests \
             auth_secret_vts_tests \
             pay_cts_tests; do
    if eval "${t}"; then
	echo "PASS: ${t}"
    else
	echo "FAIL: ${t}"
	exit 1
    fi
done