#!/bin/sh

set -e

# Execute a command in "adb shell" and capture the result.
adb_shell_run() {
  local tmp_log="$OUT/mediadrmtest.log"
  local adb_error="[ADB SHELL] $1 failed"
  adb shell $1 \|\| echo "$adb_error" | tee "$tmp_log"
  ! grep -Fq "$adb_error" "$tmp_log"
}

if [ -z "$ANDROID_BUILD_TOP" ]; then
    echo "Android build environment not set"
    exit -1
fi

echo "waiting for device"
adb root && adb wait-for-device remount

adb_shell_run /system/bin/oemcrypto_test
adb_shell_run /system/bin/request_license_test
# cdm_extended_duration_test takes >30 minutes to run.
# adb_shell_run /system/bin/cdm_extended_duration_test
adb_shell_run /system/bin/max_res_engine_unittest
adb_shell_run /system/bin/policy_engine_unittest
adb_shell_run /system/bin/libwvdrmmediacrypto_test
adb_shell_run /system/bin/libwvdrmdrmplugin_test
adb_shell_run /system/bin/cdm_engine_test
adb_shell_run /system/bin/cdm_session_unittest
adb_shell_run /system/bin/file_store_unittest
adb_shell_run /system/bin/license_unittest
adb_shell_run /system/bin/initialization_data_unittest
adb_shell_run /system/bin/device_files_unittest
adb_shell_run /system/bin/timer_unittest

library_path="/system/vendor/lib/mediadrm/ "
adb_shell_run "LD_LIBRARY_PATH=$library_path /system/bin/libwvdrmengine_test"

adb_shell_run "am start com.widevine.test/com.widevine.test.MediaDrmAPITest"
echo "MediaDrm unittests completed successfully!"