#!/bin/sh
set -e

TESTDIR="$(readlink -f "$(dirname "$0")")"
. "$TESTDIR/framework"

setupenvironment
configarchitecture 'i386'
configcompression '.' 'gz'

insertpackage 'unstable' 'apt' 'i386' '1.0'

export APT_DONT_SIGN=''
setupaptarchive --no-update
cp -a aptarchive/dists aptarchive/dists.good

test_inreleasetoobig() {
	# make InRelease really big to trigger fallback
	dd if=/dev/zero of=aptarchive/dists/unstable/InRelease bs=1M count=2 2>/dev/null
	touch -d '+1hour' aptarchive/dists/unstable/InRelease
	testsuccess aptget update -o Apt::Get::List-Cleanup=0  -o acquire::MaxReleaseFileSize=$((1*1000*1000)) -o Debug::pkgAcquire::worker=0
	msgtest 'Check that the max write warning is triggered'
	cp rootdir/tmp/testsuccess.output update.output
	testsuccess --nomsg grep -q 'File has unexpected size' update.output
	rm -f update.output
	# ensure the failed InRelease file got renamed
	testsuccess ls rootdir/var/lib/apt/lists/partial/*InRelease.FAILED
	testfailure test -e rootdir/var/lib/apt/lists/partial/Old.FAILED
}

test_packagestoobig() {
	insertpackage 'unstable' 'foo' 'i386' '1.0'
	buildaptarchivefromfiles '+1 hour'
	signreleasefiles
	# append junk at the end of the Packages.gz/Packages
	SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages.gz)"
	find aptarchive/dists -name 'Packages*' | while read pkg; do
		echo "1234567890" >> "$pkg"
		touch -d '+1hour' "$pkg"
	done
	NEW_SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages.gz)"
	testfailuremsg "E: Failed to fetch ${1}/dists/unstable/main/binary-i386/Packages.gz  File has unexpected size ($NEW_SIZE != $SIZE). Mirror sync in progress?
E: Some index files failed to download. They have been ignored, or old ones used instead." aptget update -o Debug::pkgAcquire::Worker=1 -o Debug::Acquire::Transaction=0
	testsuccess ls rootdir/var/lib/apt/lists/partial/*Packages*.FAILED
	testfailure test -e rootdir/var/lib/apt/lists/partial/Old.FAILED
}

test_packagestoosmall() {
	insertpackage 'unstable' 'foo' 'i386' '1.0'
	buildaptarchivefromfiles '+1 hour'
	signreleasefiles
	# replace Packages.gz/Packages with short junk
	SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages.gz)"
	find aptarchive/dists -name 'Packages*' | while read pkg; do
		echo "1234567890" > "$pkg"
		touch -d '+1hour' "$pkg"
	done
	NEW_SIZE="$(stat --printf=%s aptarchive/dists/unstable/main/binary-i386/Packages.gz)"
	testfailuremsg "E: Failed to fetch ${1}/dists/unstable/main/binary-i386/Packages.gz  File is smaller than expected ($NEW_SIZE < $SIZE). Mirror sync in progress?
E: Some index files failed to download. They have been ignored, or old ones used instead." aptget update -o Debug::pkgAcquire::Worker=1 -o Debug::Acquire::Transaction=0
	testsuccess ls rootdir/var/lib/apt/lists/partial/*Packages*.FAILED
	testfailure test -e rootdir/var/lib/apt/lists/partial/Old.FAILED
}


methodtest() {
	# less complicated test setup this way
	webserverconfig 'aptwebserver::support::modified-since' 'false' "$1"
	webserverconfig 'aptwebserver::support::last-modified' 'false' "$1"  # curl is clever and sees hits here also

	msgmsg 'Test with' "$1" 'and clean start'
	rm -rf rootdir/var/lib/apt/lists rootdir/var/lib/apt/lists.good aptarchive/dists
	cp -a aptarchive/dists.good aptarchive/dists
	# normal update works fine
	testsuccess aptget update
	touch rootdir/var/lib/apt/lists/partial/Old.FAILED
	mv rootdir/var/lib/apt/lists rootdir/var/lib/apt/lists.good
	# starting fresh works
	test_inreleasetoobig "$1"
	rm -rf aptarchive/dists rootdir/var/lib/apt/lists
	cp -a aptarchive/dists.good aptarchive/dists
	test_packagestoobig "$1"

	msgmsg 'Test with' "$1" 'and existing old data'
	rm -rf aptarchive/dists rootdir/var/lib/apt/lists
	cp -a rootdir/var/lib/apt/lists.good rootdir/var/lib/apt/lists
	cp -a aptarchive/dists.good aptarchive/dists
	test_inreleasetoobig "$1"
	rm -rf aptarchive/dists rootdir/var/lib/apt/lists
	cp -a rootdir/var/lib/apt/lists.good rootdir/var/lib/apt/lists
	cp -a aptarchive/dists.good aptarchive/dists
	test_packagestoobig "$1"
}

changetowebserver
methodtest "http://localhost:${APTHTTPPORT}"

changetohttpswebserver
methodtest "https://localhost:${APTHTTPSPORT}"