blob: cb7bc670ebd577f1730727dfb1e041de8f4228c0 [file] [log] [blame]
#!/bin/bash
# ---- JSON PARSING ----
json_val_by() { # json index|'key' > value
echo "$1" | python -c "import json,sys;print json.load(sys.stdin)[$2]"
}
json_jval_by() { # json index|'key' > json_value
echo "$1" |\
python -c "import json,sys;print json.dumps(json.load(sys.stdin)[$2])"
}
json_val_by_key() { json_val_by "$1" "'$2'" ; } # json key > value
json_jval_by_key() { json_jval_by "$1" "'$2'" ; } # json key > json_value
# ---- TEST RESULTS ----
result() { # test [error_message]
local result=$?
if [ $result -eq 0 ] ; then
echo "PASSED - $1 test"
else
echo "*** FAILED *** - $1 test"
RESULT=$result
[ $# -gt 1 ] && echo "$2"
fi
}
# output must match expected to pass
result_out() { # test expected output
local disp=$(echo "Expected Output:" ;\
echo " $2" ;\
echo "Actual Output:" ;\
echo " $3")
[ "$2" = "$3" ]
result "$1" "$disp"
}
# ---- Low level execution helpers ----
q() { "$@" > /dev/null 2>&1 ; } # cmd [args...] # quiet a command
gssh() { ssh -p 29418 -x "$SERVER" "$@" 2>&1 ; } # run a gerrit ssh command
batchssh() { # run a batch ssh command
local out rtn
out=$(gssh "$PLUGIN" "$@") ; rtn=$? ; echo "$out"
[ -n "$VERBOSE" ] && echo "$out" >&2
return $rtn
}
query() { gssh gerrit query "$@" ; }
remote_show() { # remote_ref > sha
git ls-remote "$GITURL" 2>/dev/null | awk '$2 == "'"$1"'" {print $1}'
}
mygit() { git --work-tree="$REPO_DIR" --git-dir="$GIT_DIR" "$@" ; } # [args...]
# ---- Custom batch getters -----
b_id() { json_val_by_key "$bjson" id ; } # > batch_id
b_state() { json_val_by_key "$bjson" state ; } # > batch_state
b_destination() { # index # > batch_destination[index]
json_jval_by "$(json_jval_by_key "$bjson" destinations)" "$1"
}
# ---- Custom batch destination getters -----
d_ref() { json_val_by_key "$1" ref ; } # destination > ref
d_project() { json_val_by_key "$1" project ; } # destination > project
d_sha1() { json_val_by_key "$1" sha1 ; } # destination > sha1
d_download() { json_val_by_key "$1" download_ref ; } # destination > download_ref
# ---- Parsers ----
query_by() { echo "$1" | awk '/^ *'"$2"':/{print $2}' ; } # qchange key > val
get_change_num() { # < gerrit_push_response > changenum
local url=$(awk '/New Changes:/ { getline; print $2 }')
echo "${url##*\/}" | tr -d -c '[:digit:]'
}
nn() { # change_num > nn
local nn=$(($1 % 100))
[ "$nn" -lt 10 ] && nn=0$nn
echo "$nn"
}
#----
get_ref_parents() { # ref > p1 p2
q mygit fetch "$GITURL" "$1" || exit 1
mygit rev-list -1 FETCH_HEAD --parents | cut -d' ' -f2,3
}
create_change() { # [--dependent] branch file [file_content] > changenum
local opt_d
[ "$1" = "--dependent" ] && { opt_d=$1 ; shift ; }
local branch=$1 tmpfile=$2 content=$3 out rtn
[ -n "$content" ] || content=$RANDOM
if [ -z "$opt_d" ] ; then
out=$(mygit fetch "$GITURL" "$branch" 2>&1) ||\
cleanup "Failed to fetch $branch: $out"
out=$(mygit checkout FETCH_HEAD 2>&1) ||\
cleanup "Failed to checkout $branch: $out"
fi
echo -e "$content" > "$tmpfile"
out=$(mygit add "$tmpfile" 2>&1) || cleanup "Failed to git add: $out"
out=$(scp -p -P 29418 "$SERVER":hooks/commit-msg $HOOK_DIR 2>&1) || cleanup "Failed to fetch commit_msg hook: $out"
out=$(mygit commit -m "Add $tmpfile" 2>&1) ||\
cleanup "Failed to commit change: $out"
out=$(mygit push "$GITURL" "HEAD:refs/for/$branch" 2>&1) ||\
cleanup "Failed to push change: $out"
out=$(echo "$out" | get_change_num) ; rtn=$? ; echo "$out"
[ -n "$VERBOSE" ] && echo " change:$out" >&2
return $rtn
}
setupGroup() { # shortname longname
GROUP=$1
echo
echo "$2"
echo "----------------------------------------------"
}
cleanup() { # [error_message]
rm -rf "$REPO_DIR"
if [ -n "$1" ] ; then
echo "$1, unable to perform batch tests" >&2
exit 1
fi
}
usage() { # [error_message]
local prog=$(basename "$0")
cat <<-EOF
Usage: $prog [-s|--server <server>] [-p|--project <project>]
[-r|--ref <ref branch>] [-g|--plugin <plugin>] [-h|--help]
-h|--help usage/help
-g|--plugin <plugin> plugin to use for the test (default: batch)
-s|--server <server> server to use for the test (default: localhost)
-p|--project <project> git project to use (default: project0)
-r|--ref <ref branch> reference branch used to create branches (default: master)
EOF
[ -n "$1" ] && echo -e '\n'"ERROR: $1"
exit 1
}
parseArgs() {
PLUGIN="batch"
SERVER="localhost"
PROJECT="tools/test/project0"
REF_BRANCH="master"
while (( "$#" )); do
case "$1" in
--plugin|-g) shift; PLUGIN=$1 ;;
--server|-s) shift; SERVER=$1 ;;
--project|-p) shift; PROJECT=$1 ;;
--ref|-r) shift; REF_BRANCH=$1 ;;
--help|-h) usage ;;
--verbose|-v) VERBOSE=$1 ;;
*) usage "invalid argument '$1'" ;;
esac
shift
done
[ -n "$SERVER" ] || usage "server not set"
[ -n "$PROJECT" ] || usage "project not set"
[ -n "$REF_BRANCH" ] || usage "ref branch not set"
}
parseArgs "$@"
GITURL=ssh://$SERVER:29418/$PROJECT
git ls-remote --heads "$GITURL" >/dev/null || usage "invalid project/server"
DEST_REF=refs/heads/$REF_BRANCH
REPO_DIR=$(mktemp -d)
trap cleanup EXIT
q git init "$REPO_DIR"
GIT_DIR="$REPO_DIR/.git"
HOOK_DIR="$GIT_DIR/hooks"
FILE_A="$REPO_DIR/fileA"
FILE_B="$REPO_DIR/fileB"
RESULT=0
setupGroup "merge-change" "Merge Change" # -------------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
bjson=$(batchssh merge-change --close "$ch1",1)
result "$GROUP" "$bjson"
id=$(b_id)
dest1=$(b_destination 0)
sha1=$(d_sha1 "$dest1")
qchange=$(query "$ch1" --current-patch-set)
result_out "$GROUP project" "$PROJECT" "$(d_project "$dest1")"
result_out "$GROUP ref" "$DEST_REF" "$(d_ref "$dest1")"
result_out "$GROUP sha1" "$(query_by "$qchange" "revision")" "$sha1"
result_out "$GROUP state" "CLOSED" "$(b_state)"
result_out "$GROUP change_state" "NEW" "$(query_by "$qchange" "status")"
setupGroup "delete" "Batch Delete" # -------------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
bjson=$(batchssh merge-change --close "$ch1",1)
id=$(b_id)
delete=$(batchssh delete "$id")
result "$GROUP" "$delete"
! delete=$(batchssh delete "$id")
result "$GROUP retry" "$delete"
setupGroup "ls-batches" "List Batches" # -------------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
bjson=$(batchssh merge-change --close "$ch1",1)
list=$(batchssh ls-batches)
echo "$list"| grep '"id"'| grep -q "$(b_id)"
result "$GROUP" "$list"
listinfo=$(batchssh ls-batches --include-batch-info)
echo "$listinfo"| grep -q '"last_modified'
result "$GROUP last_modified" "listResult: $listinfo"
setupGroup "independent clean" "Independent changes, clean merge" # ------------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
ch2=$(create_change "$REF_BRANCH" "$FILE_B") || exit
bjson=$(batchssh merge-change --close "$ch1",1 "$ch2",1)
result "$GROUP" "$bjson"
setupGroup "independent conflict" "Independent changes, merge conflict" # ------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
ch2=$(create_change "$REF_BRANCH" "$FILE_A") || exit
bjson=$(batchssh merge-change --close "$ch1",1 "$ch2",1)
echo "$bjson"| grep -q "Couldn't merge change"
result "$GROUP" "$bjson"
setupGroup "dependent changes ff" "Dependent changes, fast forward" # ---------
ch1=$(create_change "$REF_BRANCH" "$FILE_A") || exit
ch2=$(create_change --dependent "$REF_BRANCH" "$FILE_B") || exit
bjson=$(batchssh merge-change --close "$ch1",1 "$ch2",1)
id=$(b_id)
dest1=$(b_destination 0)
sha1=$(d_sha1 "$dest1")
qchange=$(query "$ch2" --current-patch-set)
result_out "$GROUP sha1" "$(query_by "$qchange" "revision")" "$sha1"
bjson=$(batchssh merge-change --close "$ch2",1 "$ch1",1)
result "$GROUP reverse" "$bjson"
! bjson=$(batchssh merge-change --close "$ch2",1)
result "$GROUP missing" "$bjson"
setupGroup "submit" "Batch Submit" # -------------
bjson=$(batchssh submit --force "$id")
result "$GROUP submit" "$bjson"
result_out "$GROUP submit dest_commit" "$sha1" "$(remote_show "$DEST_REF")"
result_out "$GROUP submit state" "DELETED" "$(b_state)"
result_out "$GROUP submit change_state" "MERGED" \
"$(query_by "$(query "$ch1")" "status")"
exit $RESULT