#!/bin/sh
set -e

TESTDIR="$(readlink -f "$(dirname "$0")")"
. "$TESTDIR/framework"
setupenvironment
configarchitecture 'amd64'

TESTFILE="$TESTDIR/framework"
cp "$TESTFILE" aptarchive/foo
APTARCHIVE="$(readlink -f ./aptarchive)"

filedown() {
	msgtest 'Downloading the same URI twice over file' "$1"
	testsuccess --nomsg apthelper download-file "file:///$APTARCHIVE/foo" './downloaded/foo1' '' \
		"file:///$APTARCHIVE/foo" './downloaded/foo2' '' -o Debug::pkgAcquire::Worker=1
	cp rootdir/tmp/testsuccess.output download.log
	testsuccess cmp "$TESTFILE" ./downloaded/foo1
	testsuccess cmp ./downloaded/foo1 ./downloaded/foo2
	testequal '1' grep -c '200%20URI%20Start' ./download.log
	testequal '1' grep -c '201%20URI%20Done' ./download.log
	rm -f ./downloaded/foo1 ./downloaded/foo2
}

testrun() {
	$1 'no partial'
	cp "$TESTFILE" ./downloaded/foo1
	$1 'complete partial 1'
	cp "$TESTFILE" ./downloaded/foo2
	$1 'complete partial 2'
	cp "$TESTFILE" ./downloaded/foo1
	cp "$TESTFILE" ./downloaded/foo2
	$1 'complete partial 1+2'
	dd if="$TESTFILE" of=./downloaded/foo1 bs=500 count=1 2>/dev/null
	$1 'partial partial 1'
	dd if="$TESTFILE" of=./downloaded/foo2 bs=500 count=1 2>/dev/null
	$1 'partial partial 2'
	dd if="$TESTFILE" of=./downloaded/foo1 bs=500 count=1 2>/dev/null
	dd if="$TESTFILE" of=./downloaded/foo2 bs=500 count=1 2>/dev/null
	$1 'partial partial 1+2'
}
testrun 'filedown'

changetowebserver -o aptwebserver::redirect::replace::/foo2=/foo

httpdown() {
	msgtest 'Downloading the same URI to different files' 'twice over http'
	testsuccess --nomsg apthelper download-file "http://localhost:${APTHTTPPORT}/foo" "./downloaded/foo1" '' \
		"http://localhost:${APTHTTPPORT}/foo" './downloaded/foo2' '' -o Debug::pkgAcquire::Worker=1 -o Debug::Acquire::http=1
	cp rootdir/tmp/testsuccess.output download.log
	testsuccess cmp "$TESTDIR/framework" ./downloaded/foo1
	testsuccess cmp ./downloaded/foo1 ./downloaded/foo2
	testequal '1' grep -c '200%20URI%20Start' ./download.log
	testequal '1' grep -c '201%20URI%20Done' ./download.log
	rm -f ./downloaded/foo1 ./downloaded/foo2
}
testrun 'httpdown'

httpredirectdown() {
	msgtest 'Redirect leads' 'first URI to the second URI'
	for i in 1 2 3 4 5 6 7 8 9 10; do
		rm -f ./downloaded/foo1 ./downloaded/foo2
		testsuccess --nomsg apthelper download-file "http://localhost:${APTHTTPPORT}/foo2" "./downloaded/foo1" '' \
			"http://localhost:${APTHTTPPORT}/foo" './downloaded/foo2' '' -o Debug::pkgAcquire::Worker=1 -o Debug::Acquire::http=1
		cp rootdir/tmp/testsuccess.output download.log
		testsuccess cmp "$TESTDIR/framework" ./downloaded/foo1
		testsuccess cmp ./downloaded/foo1 ./downloaded/foo2

		if grep -q '103%20Redirect' ./download.log; then
			break
		fi
	done
	testequal '1' grep -c '200%20URI%20Start' ./download.log
	testequal '1' grep -c '103%20Redirect' ./download.log
	testequal '1' grep -c '201%20URI%20Done' ./download.log
	rm -f ./downloaded/foo1 ./downloaded/foo2
}
testrun 'httpredirectdown'

httpsamedown() {
	msgtest 'Downloading two files via the same URI to' 'the same file'
	testsuccess --nomsg apthelper download-file "http://localhost:${APTHTTPPORT}/foo" "./downloaded/foo1" '' \
		"http://localhost:${APTHTTPPORT}/foo" './downloaded/foo1' '' -o Debug::pkgAcquire::Worker=1
	cp rootdir/tmp/testsuccess.output download.log
	testsuccess cmp "$TESTDIR/framework" ./downloaded/foo1
	testequal '1' grep -c '200%20URI%20Start' ./download.log
	testequal '1' grep -c '201%20URI%20Done' ./download.log
	rm -f ./downloaded/foo1
}
testrun 'httpsamedown'