perf-lib.sh: make "./run <revisions>" use the correct gits
[git] / 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 ABSOLUTE_GIT_TEST_INSTALLED=$(
25         test -n "$GIT_TEST_INSTALLED" && cd "$GIT_TEST_INSTALLED" && pwd)
26
27 TEST_NO_CREATE_REPO=t
28 TEST_NO_MALLOC_CHECK=t
29
30 . ../test-lib.sh
31
32 if test -z "$GIT_TEST_INSTALLED"; then
33         perf_results_prefix=
34 else
35         if test -n "$GIT_PERF_DIR_MYDIR_REL"
36         then
37                 GIT_TEST_INSTALLED=$GIT_PERF_DIR_MYDIR_REL
38         fi
39         perf_results_prefix=$(printf "%s" "${GIT_TEST_INSTALLED%/bin-wrappers}" | tr -c "[a-zA-Z0-9]" "[_*]")"."
40         GIT_TEST_INSTALLED=$ABSOLUTE_GIT_TEST_INSTALLED
41 fi
42
43 # Variables from test-lib that are normally internal to the tests; we
44 # need to export them for test_perf subshells
45 export TEST_DIRECTORY TRASH_DIRECTORY GIT_BUILD_DIR GIT_TEST_CMP
46
47 MODERN_GIT=$GIT_BUILD_DIR/bin-wrappers/git
48 export MODERN_GIT
49
50 perf_results_dir=$TEST_OUTPUT_DIRECTORY/test-results
51 test -n "$GIT_PERF_SUBSECTION" && perf_results_dir="$perf_results_dir/$GIT_PERF_SUBSECTION"
52 mkdir -p "$perf_results_dir"
53 rm -f "$perf_results_dir"/$(basename "$0" .sh).subtests
54
55 die_if_build_dir_not_repo () {
56         if ! ( cd "$TEST_DIRECTORY/.." &&
57                     git rev-parse --build-dir >/dev/null 2>&1 ); then
58                 error "No $1 defined, and your build directory is not a repo"
59         fi
60 }
61
62 if test -z "$GIT_PERF_REPO"; then
63         die_if_build_dir_not_repo '$GIT_PERF_REPO'
64         GIT_PERF_REPO=$TEST_DIRECTORY/..
65 fi
66 if test -z "$GIT_PERF_LARGE_REPO"; then
67         die_if_build_dir_not_repo '$GIT_PERF_LARGE_REPO'
68         GIT_PERF_LARGE_REPO=$TEST_DIRECTORY/..
69 fi
70
71 test_perf_do_repo_symlink_config_ () {
72         test_have_prereq SYMLINKS || git config core.symlinks false
73 }
74
75 test_perf_create_repo_from () {
76         test "$#" = 2 ||
77         BUG "not 2 parameters to test-create-repo"
78         repo="$1"
79         source="$2"
80         source_git="$("$MODERN_GIT" -C "$source" rev-parse --git-dir)"
81         objects_dir="$("$MODERN_GIT" -C "$source" rev-parse --git-path objects)"
82         mkdir -p "$repo/.git"
83         (
84                 cd "$source" &&
85                 { cp -Rl "$objects_dir" "$repo/.git/" 2>/dev/null ||
86                         cp -R "$objects_dir" "$repo/.git/"; } &&
87                 for stuff in "$source_git"/*; do
88                         case "$stuff" in
89                                 */objects|*/hooks|*/config|*/commondir)
90                                         ;;
91                                 *)
92                                         cp -R "$stuff" "$repo/.git/" || exit 1
93                                         ;;
94                         esac
95                 done
96         ) &&
97         (
98                 cd "$repo" &&
99                 "$MODERN_GIT" init -q &&
100                 test_perf_do_repo_symlink_config_ &&
101                 mv .git/hooks .git/hooks-disabled 2>/dev/null &&
102                 if test -f .git/index.lock
103                 then
104                         # We may be copying a repo that can't run "git
105                         # status" due to a locked index. Since we have
106                         # a copy it's fine to remove the lock.
107                         rm .git/index.lock
108                 fi
109         ) || error "failed to copy repository '$source' to '$repo'"
110 }
111
112 # call at least one of these to establish an appropriately-sized repository
113 test_perf_fresh_repo () {
114         repo="${1:-$TRASH_DIRECTORY}"
115         "$MODERN_GIT" init -q "$repo" &&
116         (
117                 cd "$repo" &&
118                 test_perf_do_repo_symlink_config_
119         )
120 }
121
122 test_perf_default_repo () {
123         test_perf_create_repo_from "${1:-$TRASH_DIRECTORY}" "$GIT_PERF_REPO"
124 }
125 test_perf_large_repo () {
126         if test "$GIT_PERF_LARGE_REPO" = "$GIT_BUILD_DIR"; then
127                 echo "warning: \$GIT_PERF_LARGE_REPO is \$GIT_BUILD_DIR." >&2
128                 echo "warning: This will work, but may not be a sufficiently large repo" >&2
129                 echo "warning: for representative measurements." >&2
130         fi
131         test_perf_create_repo_from "${1:-$TRASH_DIRECTORY}" "$GIT_PERF_LARGE_REPO"
132 }
133 test_checkout_worktree () {
134         git checkout-index -u -a ||
135         error "git checkout-index failed"
136 }
137
138 # Performance tests should never fail.  If they do, stop immediately
139 immediate=t
140
141 # Perf tests require GNU time
142 case "$(uname -s)" in Darwin) GTIME="${GTIME:-gtime}";; esac
143 GTIME="${GTIME:-/usr/bin/time}"
144
145 test_run_perf_ () {
146         test_cleanup=:
147         test_export_="test_cleanup"
148         export test_cleanup test_export_
149         "$GTIME" -f "%E %U %S" -o test_time.$i "$SHELL" -c '
150 . '"$TEST_DIRECTORY"/test-lib-functions.sh'
151 test_export () {
152         [ $# != 0 ] || return 0
153         test_export_="$test_export_\\|$1"
154         shift
155         test_export "$@"
156 }
157 '"$1"'
158 ret=$?
159 set | sed -n "s'"/'/'\\\\''/g"';s/^\\($test_export_\\)/export '"'&'"'/p" >test_vars
160 exit $ret' >&3 2>&4
161         eval_ret=$?
162
163         if test $eval_ret = 0 || test -n "$expecting_failure"
164         then
165                 test_eval_ "$test_cleanup"
166                 . ./test_vars || error "failed to load updated environment"
167         fi
168         if test "$verbose" = "t" && test -n "$HARNESS_ACTIVE"; then
169                 echo ""
170         fi
171         return "$eval_ret"
172 }
173
174 test_wrapper_ () {
175         test_wrapper_func_=$1; shift
176         test_start_
177         test "$#" = 3 && { test_prereq=$1; shift; } || test_prereq=
178         test "$#" = 2 ||
179         BUG "not 2 or 3 parameters to test-expect-success"
180         export test_prereq
181         if ! test_skip "$@"
182         then
183                 base=$(basename "$0" .sh)
184                 echo "$test_count" >>"$perf_results_dir"/$base.subtests
185                 echo "$1" >"$perf_results_dir"/$base.$test_count.descr
186                 base="$perf_results_dir"/"$perf_results_prefix$(basename "$0" .sh)"."$test_count"
187                 "$test_wrapper_func_" "$@"
188         fi
189
190         test_finish_
191 }
192
193 test_perf_ () {
194         if test -z "$verbose"; then
195                 printf "%s" "perf $test_count - $1:"
196         else
197                 echo "perf $test_count - $1:"
198         fi
199         for i in $(test_seq 1 $GIT_PERF_REPEAT_COUNT); do
200                 say >&3 "running: $2"
201                 if test_run_perf_ "$2"
202                 then
203                         if test -z "$verbose"; then
204                                 printf " %s" "$i"
205                         else
206                                 echo "* timing run $i/$GIT_PERF_REPEAT_COUNT:"
207                         fi
208                 else
209                         test -z "$verbose" && echo
210                         test_failure_ "$@"
211                         break
212                 fi
213         done
214         if test -z "$verbose"; then
215                 echo " ok"
216         else
217                 test_ok_ "$1"
218         fi
219         "$TEST_DIRECTORY"/perf/min_time.perl test_time.* >"$base".times
220 }
221
222 test_perf () {
223         test_wrapper_ test_perf_ "$@"
224 }
225
226 test_size_ () {
227         say >&3 "running: $2"
228         if test_eval_ "$2" 3>"$base".size; then
229                 test_ok_ "$1"
230         else
231                 test_failure_ "$@"
232         fi
233 }
234
235 test_size () {
236         test_wrapper_ test_size_ "$@"
237 }
238
239 # We extend test_done to print timings at the end (./run disables this
240 # and does it after running everything)
241 test_at_end_hook_ () {
242         if test -z "$GIT_PERF_AGGREGATING_LATER"; then
243                 ( cd "$TEST_DIRECTORY"/perf && ./aggregate.perl $(basename "$0") )
244         fi
245 }
246
247 test_export () {
248         export "$@"
249 }