cascading-pr/tests/run.sh

#!/bin/bash
# SPDX-License-Identifier: MIT

set -e

SELF_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
TMPDIR=/tmp/cascading-pr-test
mkdir -p $TMPDIR
source $SELF_DIR/../cascading-pr-lib.sh

function push_self() {
    forgejo-test-helper.sh push_self_action http://user1:admin1234@${options[host_port]} user1 cascading-pr vTest
}

function user_login() {
    local username=$1
    (
	export DOT=$TMPDIR/$username
	forgejo-curl.sh logout
	forgejo-curl.sh --user $username --password "${options[password]}" login ${options[url]}
    )
}

function user_curl() {
    local username=$1
    shift
    DOT=$TMPDIR/$username forgejo-curl.sh "$@"
}

function user_token() {
    local username=$1 name=$2

    curl -sS -f -H Content-Type:application/json --user "$username:${options[password]}" --data-raw '{"name":"'$name'","scopes":["write:repository","write:issue","read:user"]}' ${options[url]}/api/v1/users/$username/tokens  | jq --raw-output .sha1 | tee $TMPDIR/$username/repo-token
}

function user_secret() {
    local username=$1 name=$2 token=$3

    user_curl $username api_json -X PUT --data-raw '{"data":"'$token'"}' ${options[url]}/api/v1/repos/user1/originrepo/actions/secrets/$name
}

function user_create() {
    local username="$1" email="$2"
    forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/admin/users/$username?purge=true >& /dev/null || true
    forgejo-curl.sh api_json --data-raw '{"username":"'$username'","email":"'$email'","password":"'${options[password]}'","must_change_password":false}' ${options[url]}/api/v1/admin/users
    user_login $username
}

function close_pull_request() {
    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user1/originrepo/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json -X PATCH --data-raw '{"state":"closed"}' ${options[url]}/api/v1/repos/user1/originrepo/issues/$pr
    done
}

function merge_pull_request() {
    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user1/originrepo/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json --data-raw '{"Do":"merge"}' ${options[url]}/api/v1/repos/user1/originrepo/pulls/$pr/merge
    done
}

function create_pull_request() {
    forgejo-curl.sh api_json ${options[url]}/api/v1/repos/user1/originrepo/pulls | jq --raw-output '.[] | .number' | while read pr ; do
	forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/repos/user1/originrepo/issues/$pr
    done
    forgejo-curl.sh api_json -X DELETE ${options[url]}/api/v1/repos/user1/originrepo/branches/branch1 >& /dev/null || true
    forgejo-curl.sh api_json --data-raw '{"new_branch_name":"branch1"}' ${options[url]}/api/v1/repos/user1/originrepo/branches
    (
	cd $TMPDIR
	rm -fr originrepo
	git clone -b branch1 http://user1:admin1234@${options[host_port]}/user1/originrepo
	cd originrepo
	echo CONTENT > README
	git config user.email root@example.com
	git config user.name username
	git add .
	git commit -m 'update'
	git push origin branch1
	git rev-parse HEAD > ../originrepo.sha
    )

    forgejo-curl.sh api_json --data-raw '{"title":"PR","base":"main","head":"branch1"}' ${options[url]}/api/v1/repos/user1/originrepo/pulls
}

function finalize_options() {
    if test -f forgejo-ip; then
	: ${options[host_port]:=$(cat forgejo-ip):3000}
    fi
    options[url]=http://${options[host_port]}
    if test -f forgejo-token; then
	: ${options[token]:=$(cat forgejo-token)}
    fi
    options[password]=admin1234
}

function unit_retry_fail() {
    local file=$1
    local value=$(cat $file)
    expr $value - 1 > $file
    echo RESULT
    if test $value -gt 0 ; then
	return 1
    else
	return 0
    fi
}

function unit_retry() {
    local two=$TMPDIR/unit_retry_two

    rm -f $TMPDIR/retry.*

    #
    # Succeeds after two tries
    #
    echo 2 > $TMPDIR/unit_retry_two
    RETRY_DELAYS='1 1 1 1' retry unit_retry_fail $two > $TMPDIR/retry.result 2> $TMPDIR/retry.log
    test "$(cat $TMPDIR/retry.result)" = "RESULT"
    test "$(grep -c 'waiting 1 unit_retry_fail' $TMPDIR/retry.log)" = 2

    #
    # Succeeds immediately
    #
    RETRY_DELAYS='1' retry unit_retry_fail $two > $TMPDIR/retry.result 2> $TMPDIR/retry.log
    test "$(cat $TMPDIR/retry.result)" = "RESULT"
    test "$(grep -c 'waiting 1 unit_retry_fail' $TMPDIR/retry.log)" = 0

    #
    # Verify the output is only the output of the last run and is not polluted by
    # unrelated output
    #
    echo 2 > $TMPDIR/unit_retry_two
    test "$(RETRY_DELAYS='1 1 1' retry unit_retry_fail $two)" = "RESULT"

    #
    # Fails after one try
    #
    echo 2 > $TMPDIR/unit_retry_two
    if RETRY_DELAYS='1' retry unit_retry_fail $two |& tee $TMPDIR/retry.log ; then
	return 1
    fi
    grep --quiet 'retry failed' $TMPDIR/retry.log
}

function integration() {
    user_create user2 user2@example.com
    forgejo-test-helper.sh push tests/destinationrepo http://user2:admin1234@${options[host_port]} user2 destinationrepo

    user_create user1 user1@example.com
    forgejo-test-helper.sh push tests/originrepo http://user1:admin1234@${options[host_port]} user1 originrepo cascading-pr
    user_secret user1 ORIGIN_TOKEN $(user_token user1 ORIGIN_TOKEN)
    user_secret user1 DESTINATION_TOKEN $(user_token user2 DESTINATION_TOKEN)

    push_self

    #
    # create & close
    #
    create_pull_request
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/originrepo.sha)
    close_pull_request
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/originrepo.sha)

    #
    # create & merge
    #
    create_pull_request
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/originrepo.sha)
    merge_pull_request
    wait_success ${options[url]}/api/v1/repos/user1/originrepo $(cat $TMPDIR/originrepo.sha)
}

function unit() {
    unit_retry
}

function run() {
    unit
    integration
}

function main() {
    local command=run

    while true; do
	case "$1" in
	    --verbose)
		shift
		verbose
		;;
	    --debug)
		shift
		debug
		;;
	    --host_port)
		shift
		options[host_port]=$1
		shift
		;;
	    --url)
		shift
		options[url]=$1
		shift
		;;
	    --token)
		shift
		options[token]=$1
		shift
		;;
	    *)
		finalize_options
		"${1:-run}"
		return 0
		;;
	esac
    done
}

dependencies

${MAIN:-main} "${@}"