#!/bin/bash
#/usr/bin/expect
if [ -z $(which android) ] ; then
	echo "android binary not in your user's path."
	exit
fi
expect -c '
set timeout -1 ;
spawn android update sdk --no-ui;
expect {
    "Do you accept the license" { exp_send "y\r" ; exp_continue }
    eof
}
'

#!/bin/bash
#************************************************#
#              test_automator.sh                 #
#           written by Matt Miller               #
#                May 9, 2014                     #
#                                                #
#          Run robotium tests on connected       #
#	         android devices and emulators         #
#************************************************#

usage() {
	echo "Usage: $0 -p package-name [-t testcase] [-a] [-e]"
	echo "-a Run all tests in package"
	echo "-t testcase or testcases delimited by commas"
	echo "-p required: package name of your application's tests"
	echo "-e start emulators"
}

# Function to start avds we have created
function startEmulators {
	echo "Looking for AVDs..."
	avds=( $(android list avd | awk '/Name:/ {print $2}') )

	echo "Available AVDs:"

	for avd in "${avds[@]}"
	do
		echo "Starting emulator ", $avd
		emulator -avd $avd &
		avd_processes+=($!)
	done

	#wait just a bit for ADB to catch up
	sleep 5

	# Get all emulator ids that are waiting to boot
	# We must wait until the lock screen is showing to begin the testing.
	avd_to_boot=( $(adb devices | awk '/emulator-/ {print $1}') )

	#set the timeout form the config
	if [ -z $timeout ]; then
		echo "Timeout config not set, defaulting to 120 second boot timeout"
		TIMEOUT=120 #2 minute boot time for the emulators
	else
		TIMEOUT=$timeout
	fi

	#If we are booted add to the current booted devices
	declare -A booted_devices

	starttime=$(date +%s)

	while [ "${#booted_devices[@]}" -lt "${#avd_to_boot[@]}" ]; do
		currenttime=$(date +%s)
		diff=`expr $currenttime - $starttime`
		echo "Diff = $diff"
		if [ $diff -ge $TIMEOUT ]; then
			break
		fi
		for avd in "${avd_to_boot[@]}"
		do
			ret=$(adb -s $avd shell getprop init.svc.bootanim)
			echo "here: $ret"
			if [[ "$ret" == *"stopped"* ]]; then
				echo "$avd is booted!"
				booted_devices[$avd]=$avd
			fi
		done
		sleep 15
	done

	#kill the rest of the devices that didn't boot
	for avd in "${avd_to_boot[@]}"
	do
		if [ -z "$booted_devices[$avd]" ]; then
			adb -s $avd emu kill
		fi
	done
}

# Function to execute test on a device
function executeTest {
	device=$1

	# Make sure the app APK and the test APK are installed
	# And install them if not from the config or args

	res=$(adb -s $device shell pm list instrumentation $package)
	if [ -z  "$res" ]; then
		echo "Test apk not installed, installing from config"
		adb -s $device install $test_apk_location
   	fi

	res=$(adb -s $device shell pm list packages $package)
	if [ -z "$res" ]; then
		echo "App Apk not installed, installing from config"
		adb -s $device install $apk_location
	fi

	if [ "$runall" = true ]; then
		echo "Running all tests for device $device"
		#run all tests
		cmd="adb -s $1 shell am instrument -w -e package $package.test $package.test/android.test.InstrumentationTestRunner"
		echo $cmd
		adb -s $device shell am instrument -w -e package $package.test $package.test/android.test.InstrumentationTestRunner > $device.results &
	else
		echo "Running testcases $testcases for device $device"
		#run all tests
		cmd="adb -s $1 shell am instrument -w -e class $testcases $package.test/android.test.InstrumentationTestRunner"
		adb -s $device shell am instrument -w -e class $testcases $package.test/android.test.InstrumentationTestRunner >> $device.results &
	fi
}

function checkDependencies {
	# Dependency ADB (Android Debug Bridge)
	if [ -z $(which adb) ] ; then
		echo "adb binary not in your user's path. It is a dependency of this script."
		exit
	fi

	if [ -z $(which android) ] ; then
		echo "android binary not in your user's path. It is a dependency of this script."
		exit
	fi
}

# Catch ctrl-c and kill all subprocesses that the script started.
trap ctrl_c INT
function ctrl_c() {
	echo "Script terminating..."
	pkill -P $$
	exit
}

# our vars
package=
runall=true
testcases=
avd_processes=()

source test_config
echo "APK location = $apk_location"
echo "Test APK location = $test_apk_location"
# parse cmd line args
while getopts :p:aet: opt; do
	case $opt in
		p)
			package=$OPTARG
			;;
		a)
			runall=true
			;;
		t)
			runall=false
			testcases=$OPTARG
			;;
		e)
			startEmulators
			;;
		*)
			usage
			exit
			;;
	esac
done

#required
if [ -z "$package" ]; then
	echo "Missing package."
	usage
	exit
fi

checkDependencies

#Get all devices currently running on adb and store their ids into an array
output=$(adb devices)
devices=( $(printf "%s","$output" | awk '{if (NR!=1){print $1}}') )

#iterate over each device, running tests we've specified.
for device in "${devices[@]}"
do
	echo "Running test on device $device"
	executeTest $device
done
wait
echo
echo "Tests finished for all devices"

usage() {
	echo "Usage: $0 -p package-name [-t testcase] [-a] [-e]"
	echo "-a Run all tests in package"
	echo "-t testcase or testcases delimited by commas"
	echo "-p required: package name of your application's tests"
	echo "-e start emulators"
}

function checkDependencies {
	# Dependency ADB (Android Debug Bridge)
	if [ -z $(which adb) ] ; then
		echo "adb binary not in your user's path. It is a dependency of this script."
		exit
	fi

	if [ -z $(which android) ] ; then
		echo "android binary not in your user's path. It is a dependency of this script."
		exit
	fi
}