t/perf/perf-lib.sh

   1 # Performance testing framework.  Each perf script starts much like
   2 # a normal test script, except it sources this library instead of
   3 # test-lib.sh.  See t/perf/README for documentation.
   4 #
   5 # Copyright (c) 2011 Thomas Rast
   6 #
   7 # This program is free software: you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation, either version 2 of the License, or
  10 # (at your option) any later version.
  11 #
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16 #
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program.  If not, see http://www.gnu.org/licenses/ .
  19
  20 # These variables must be set before the inclusion of test-lib.sh below,
  21 # because it will change our working directory.
  22 TEST_DIRECTORY=$(pwd)/..
  23 TEST_OUTPUT_DIRECTORY=$(pwd)
  24
  25 TEST_NO_CREATE_REPO=t
  26 TEST_NO_MALLOC_CHECK=t
  27
  28 . ../test-lib.sh
  29
  30 unset GIT_CONFIG_NOSYSTEM
  31 GIT_CONFIG_SYSTEM="$TEST_DIRECTORY/perf/config"
  32 export GIT_CONFIG_SYSTEM
  33
  34 if test -n "$GIT_TEST_INSTALLED" -a -z "$PERF_SET_GIT_TEST_INSTALLED"
  35 then
  36         error "Do not use GIT_TEST_INSTALLED with the perf tests.
  37
  38 Instead use:
  39
  40     ./run <path-to-git> -- <tests>
  41
  42 See t/perf/README for details."
  43 fi
  44
  45 # Variables from test-lib that are normally internal to the tests; we
  46 # need to export them for test_perf subshells
  47 export TEST_DIRECTORY TRASH_DIRECTORY GIT_BUILD_DIR GIT_TEST_CMP
  48
  49 MODERN_GIT=$GIT_BUILD_DIR/bin-wrappers/git
  50 export MODERN_GIT
  51
  52 MODERN_SCALAR=$GIT_BUILD_DIR/bin-wrappers/scalar
  53 export MODERN_SCALAR
  54
  55 perf_results_dir=$TEST_RESULTS_DIR
  56 test -n "$GIT_PERF_SUBSECTION" && perf_results_dir="$perf_results_dir/$GIT_PERF_SUBSECTION"
  57 mkdir -p "$perf_results_dir"
  58 rm -f "$perf_results_dir"/$(basename "$0" .sh).subtests
  59
  60 die_if_build_dir_not_repo () {
  61         if ! ( cd "$TEST_DIRECTORY/.." &&
  62                     git rev-parse --build-dir >/dev/null 2>&1 ); then
  63                 error "No $1 defined, and your build directory is not a repo"
  64         fi
  65 }
  66
  67 if test -z "$GIT_PERF_REPO"; then
  68         die_if_build_dir_not_repo '$GIT_PERF_REPO'
  69         GIT_PERF_REPO=$TEST_DIRECTORY/..
  70 fi
  71 if test -z "$GIT_PERF_LARGE_REPO"; then
  72         die_if_build_dir_not_repo '$GIT_PERF_LARGE_REPO'
  73         GIT_PERF_LARGE_REPO=$TEST_DIRECTORY/..
  74 fi
  75
  76 test_perf_do_repo_symlink_config_ () {
  77         test_have_prereq SYMLINKS || git config core.symlinks false
  78 }
  79
  80 test_perf_copy_repo_contents () {
  81         for stuff in "$1"/*
  82         do
  83                 case "$stuff" in
  84                 */objects|*/hooks|*/config|*/commondir|*/gitdir|*/worktrees|*/fsmonitor--daemon*)
  85                         ;;
  86                 *)
  87                         cp -R "$stuff" "$repo/.git/" || exit 1
  88                         ;;
  89                 esac
  90         done
  91 }
  92
  93 test_perf_create_repo_from () {
  94         test "$#" = 2 ||
  95         BUG "not 2 parameters to test-create-repo"
  96         repo="$1"
  97         source="$2"
  98         source_git="$("$MODERN_GIT" -C "$source" rev-parse --git-dir)"
  99         objects_dir="$("$MODERN_GIT" -C "$source" rev-parse --git-path objects)"
 100         common_dir="$("$MODERN_GIT" -C "$source" rev-parse --git-common-dir)"
 101         mkdir -p "$repo/.git"
 102         (
 103                 cd "$source" &&
 104                 { cp -Rl "$objects_dir" "$repo/.git/" 2>/dev/null ||
 105                         cp -R "$objects_dir" "$repo/.git/"; } &&
 106
 107                 # common_dir must come first here, since we want source_git to
 108                 # take precedence and overwrite any overlapping files
 109                 test_perf_copy_repo_contents "$common_dir"
 110                 if test "$source_git" != "$common_dir"
 111                 then
 112                         test_perf_copy_repo_contents "$source_git"
 113                 fi
 114         ) &&
 115         (
 116                 cd "$repo" &&
 117                 "$MODERN_GIT" init -q &&
 118                 test_perf_do_repo_symlink_config_ &&
 119                 mv .git/hooks .git/hooks-disabled 2>/dev/null &&
 120                 if test -f .git/index.lock
 121                 then
 122                         # We may be copying a repo that can't run "git
 123                         # status" due to a locked index. Since we have
 124                         # a copy it's fine to remove the lock.
 125                         rm .git/index.lock
 126                 fi &&
 127                 if test_bool_env GIT_PERF_USE_SCALAR false
 128                 then
 129                         "$MODERN_SCALAR" register
 130                 fi
 131         ) || error "failed to copy repository '$source' to '$repo'"
 132 }
 133
 134 # call at least one of these to establish an appropriately-sized repository
 135 test_perf_fresh_repo () {
 136         repo="${1:-$TRASH_DIRECTORY}"
 137         "$MODERN_GIT" init -q "$repo" &&
 138         (
 139                 cd "$repo" &&
 140                 test_perf_do_repo_symlink_config_ &&
 141                 if test_bool_env GIT_PERF_USE_SCALAR false
 142                 then
 143                         "$MODERN_SCALAR" register
 144                 fi
 145         )
 146 }
 147
 148 test_perf_default_repo () {
 149         test_perf_create_repo_from "${1:-$TRASH_DIRECTORY}" "$GIT_PERF_REPO"
 150 }
 151 test_perf_large_repo () {
 152         if test "$GIT_PERF_LARGE_REPO" = "$GIT_BUILD_DIR"; then
 153                 echo "warning: \$GIT_PERF_LARGE_REPO is \$GIT_BUILD_DIR." >&2
 154                 echo "warning: This will work, but may not be a sufficiently large repo" >&2
 155                 echo "warning: for representative measurements." >&2
 156         fi
 157         test_perf_create_repo_from "${1:-$TRASH_DIRECTORY}" "$GIT_PERF_LARGE_REPO"
 158 }
 159 test_checkout_worktree () {
 160         git checkout-index -u -a ||
 161         error "git checkout-index failed"
 162 }
 163
 164 # Performance tests should never fail.  If they do, stop immediately
 165 immediate=t
 166
 167 # Perf tests require GNU time
 168 case "$(uname -s)" in Darwin) GTIME="${GTIME:-gtime}";; esac
 169 GTIME="${GTIME:-/usr/bin/time}"
 170
 171 test_run_perf_ () {
 172         test_cleanup=:
 173         test_export_="test_cleanup"
 174         export test_cleanup test_export_
 175         "$GTIME" -f "%E %U %S" -o test_time.$i "$TEST_SHELL_PATH" -c '
 176 . '"$TEST_DIRECTORY"/test-lib-functions.sh'
 177 test_export () {
 178         test_export_="$test_export_ $*"
 179 }
 180 '"$1"'
 181 ret=$?
 182 needles=
 183 for v in $test_export_
 184 do
 185         needles="$needles;s/^$v=/export $v=/p"
 186 done
 187 set | sed -n "s'"/'/'\\\\''/g"'$needles" >test_vars
 188 exit $ret' >&3 2>&4
 189         eval_ret=$?
 190
 191         if test $eval_ret = 0 || test -n "$expecting_failure"
 192         then
 193                 test_eval_ "$test_cleanup"
 194                 . ./test_vars || error "failed to load updated environment"
 195         fi
 196         if test "$verbose" = "t" && test -n "$HARNESS_ACTIVE"; then
 197                 echo ""
 198         fi
 199         return "$eval_ret"
 200 }
 201
 202 test_wrapper_ () {
 203         local test_wrapper_func_="$1"; shift
 204         local test_title_="$1"; shift
 205         test_start_
 206         test_prereq=
 207         test_perf_setup_=
 208         while test $# != 0
 209         do
 210                 case $1 in
 211                 --prereq)
 212                         test_prereq=$2
 213                         shift
 214                         ;;
 215                 --setup)
 216                         test_perf_setup_=$2
 217                         shift
 218                         ;;
 219                 *)
 220                         break
 221                         ;;
 222                 esac
 223                 shift
 224         done
 225         test "$#" = 1 || BUG "test_wrapper_ needs 2 positional parameters"
 226         export test_prereq
 227         export test_perf_setup_
 228
 229         if ! test_skip "$test_title_" "$@"
 230         then
 231                 base=$(basename "$0" .sh)
 232                 echo "$test_count" >>"$perf_results_dir"/$base.subtests
 233                 echo "$test_title_" >"$perf_results_dir"/$base.$test_count.descr
 234                 base="$perf_results_dir"/"$PERF_RESULTS_PREFIX$(basename "$0" .sh)"."$test_count"
 235                 "$test_wrapper_func_" "$test_title_" "$@"
 236         fi
 237
 238         test_finish_
 239 }
 240
 241 test_perf_ () {
 242         if test -z "$verbose"; then
 243                 printf "%s" "perf $test_count - $1:"
 244         else
 245                 echo "perf $test_count - $1:"
 246         fi
 247         for i in $(test_seq 1 $GIT_PERF_REPEAT_COUNT); do
 248                 if test -n "$test_perf_setup_"
 249                 then
 250                         say >&3 "setup: $test_perf_setup_"
 251                         if ! test_eval_ $test_perf_setup_
 252                         then
 253                                 test_failure_ "$test_perf_setup_"
 254                                 break
 255                         fi
 256
 257                 fi
 258                 say >&3 "running: $2"
 259                 if test_run_perf_ "$2"
 260                 then
 261                         if test -z "$verbose"; then
 262                                 printf " %s" "$i"
 263                         else
 264                                 echo "* timing run $i/$GIT_PERF_REPEAT_COUNT:"
 265                         fi
 266                 else
 267                         test -z "$verbose" && echo
 268                         test_failure_ "$@"
 269                         break
 270                 fi
 271         done
 272         if test -z "$verbose"; then
 273                 echo " ok"
 274         else
 275                 test_ok_ "$1"
 276         fi
 277         "$TEST_DIRECTORY"/perf/min_time.perl test_time.* >"$base".result
 278         rm test_time.*
 279 }
 280
 281 # Usage: test_perf 'title' [options] 'perf-test'
 282 #       Run the performance test script specified in perf-test with
 283 #       optional prerequisite and setup steps.
 284 # Options:
 285 #       --prereq prerequisites: Skip the test if prequisites aren't met
 286 #       --setup "setup-steps": Run setup steps prior to each measured iteration
 287 #
 288 test_perf () {
 289         test_wrapper_ test_perf_ "$@"
 290 }
 291
 292 test_size_ () {
 293         if test -n "$test_perf_setup_"
 294         then
 295                 say >&3 "setup: $test_perf_setup_"
 296                 test_eval_ $test_perf_setup_
 297         fi
 298
 299         say >&3 "running: $2"
 300         if test_eval_ "$2" 3>"$base".result; then
 301                 test_ok_ "$1"
 302         else
 303                 test_failure_ "$@"
 304         fi
 305 }
 306
 307 # Usage: test_size 'title' [options] 'size-test'
 308 #       Run the size test script specified in size-test with optional
 309 #       prerequisites and setup steps. Returns the numeric value
 310 #       returned by size-test.
 311 # Options:
 312 #       --prereq prerequisites: Skip the test if prequisites aren't met
 313 #       --setup "setup-steps": Run setup steps prior to the size measurement
 314
 315 test_size () {
 316         test_wrapper_ test_size_ "$@"
 317 }
 318
 319 # We extend test_done to print timings at the end (./run disables this
 320 # and does it after running everything)
 321 test_at_end_hook_ () {
 322         if test -z "$GIT_PERF_AGGREGATING_LATER"; then
 323                 (
 324                         cd "$TEST_DIRECTORY"/perf &&
 325                         ./aggregate.perl --results-dir="$TEST_RESULTS_DIR" $(basename "$0")
 326                 )
 327         fi
 328 }
 329
 330 test_export () {
 331         export "$@"
 332 }
 333
 334 test_lazy_prereq PERF_EXTRA 'test_bool_env GIT_PERF_EXTRA false'