tap: improve syncing between awk+shell and perl implementations
[platform/upstream/automake.git] / lib / tap-driver.sh
1 #! /bin/sh
2 # Copyright (C) 2011 Free Software Foundation, Inc.
3 #
4 # This program is free software; you can redistribute it and/or modify
5 # it under the terms of the GNU General Public License as published by
6 # the Free Software Foundation; either version 2, or (at your option)
7 # any later version.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 # GNU General Public License for more details.
13 #
14 # You should have received a copy of the GNU General Public License
15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
16
17 # As a special exception to the GNU General Public License, if you
18 # distribute this file as part of a program that contains a
19 # configuration script generated by Autoconf, you may include it under
20 # the same distribution terms that you use for the rest of that program.
21
22 # This file is maintained in Automake, please report
23 # bugs to <bug-automake@gnu.org> or send patches to
24 # <automake-patches@gnu.org>.
25
26 scriptversion=2011-08-25.10; # UTC
27
28 # Make unconditional expansion of undefined variables an error.  This
29 # helps a lot in preventing typo-related bugs.
30 set -u
31
32 me=tap-driver.sh
33
34 fatal ()
35 {
36   echo "$me: fatal: $*" >&2
37   exit 1
38 }
39
40 usage_error ()
41 {
42   echo "$me: $*" >&2
43   print_usage >&2
44   exit 2
45 }
46
47 print_usage ()
48 {
49   cat <<END
50 Usage:
51   tap-driver.sh --test-name=NAME --log-file=PATH --trs-file=PATH
52                 [--expect-failure={yes|no}] [--color-tests={yes|no}]
53                 [--enable-hard-errors={yes|no}] [--ignore-exit]
54                 [--diagnostic-string=STRING] [--merge|--no-merge]
55                 [--comments|--no-comments] [--] TEST-COMMAND
56 The \`--test-name', \`--log-file' and \`--trs-file' options are mandatory.
57 END
58 }
59
60 # TODO: better error handling in option parsing (in particular, ensure
61 # TODO: $log_file, $trs_file and $test_name are defined).
62 test_name= # Used for reporting.
63 log_file=  # Where to save the result and output of the test script.
64 trs_file=  # Where to save the metadata of the test run.
65 expect_failure=0
66 color_tests=0
67 merge=0
68 ignore_exit=0
69 comments=0
70 diag_string='#'
71 while test $# -gt 0; do
72   case $1 in
73   --help) print_usage; exit $?;;
74   --version) echo "$me $scriptversion"; exit $?;;
75   --test-name) test_name=$2; shift;;
76   --log-file) log_file=$2; shift;;
77   --trs-file) trs_file=$2; shift;;
78   --color-tests) color_tests=$2; shift;;
79   --expect-failure) expect_failure=$2; shift;;
80   --enable-hard-errors) shift;; # No-op.
81   --merge) merge=1;;
82   --no-merge) merge=0;;
83   --ignore-exit) ignore_exit=1;;
84   --comments) comments=1;;
85   --no-comments) comments=0;;
86   --diagnostic-string) diag_string=$2; shift;;
87   --) shift; break;;
88   -*) usage_error "invalid option: '$1'";;
89   esac
90   shift
91 done
92
93 test $# -gt 0 || usage_error "missing test command"
94
95 case $expect_failure in
96   yes) expect_failure=1;;
97     *) expect_failure=0;;
98 esac
99
100 if test $color_tests = yes; then
101   init_colors='
102     color_map["red"]="\e[0;31m" # Red.
103     color_map["grn"]="\e[0;32m" # Green.
104     color_map["lgn"]="\e[1;32m" # Light green.
105     color_map["blu"]="\e[1;34m" # Blue.
106     color_map["mgn"]="\e[0;35m" # Magenta.
107     color_map["std"]="\e[m"     # No color.
108     color_for_result["ERROR"] = "mgn"
109     color_for_result["PASS"]  = "grn"
110     color_for_result["XPASS"] = "red"
111     color_for_result["FAIL"]  = "red"
112     color_for_result["XFAIL"] = "lgn"
113     color_for_result["SKIP"]  = "blu"'
114 else
115   init_colors=''
116 fi
117
118 {
119   { if test $merge -gt 0; then
120       exec 2>&1
121     else
122       exec 2>&3
123     fi
124     "$@"
125     echo $?
126   } | LC_ALL=C ${AM_TAP_AWK-awk} \
127         -v me="$me" \
128         -v test_script_name="$test_name" \
129         -v log_file="$log_file" \
130         -v trs_file="$trs_file" \
131         -v expect_failure="$expect_failure" \
132         -v merge="$merge" \
133         -v ignore_exit="$ignore_exit" \
134         -v comments="$comments" \
135         -v diag_string="$diag_string" \
136 '
137 # FIXME: the usages of "cat >&3" below could be optimized when using
138 # FIXME: GNU awk, and/on on systems that supports /dev/fd/.
139
140 # Implementation note: in what follows, `result_obj` will be an
141 # associative array that (partly) simulates a TAP result object
142 # from the `TAP::Parser` perl module.
143
144 ## ----------- ##
145 ##  FUNCTIONS  ##
146 ## ----------- ##
147
148 function fatal(msg)
149 {
150   print me ": " msg | "cat >&2"
151   exit 1
152 }
153
154 function abort(where)
155 {
156   fatal("internal error " where)
157 }
158
159 # Convert a boolean to a "yes"/"no" string.
160 function yn(bool)
161 {
162   return bool ? "yes" : "no";
163 }
164
165 function add_test_result(result)
166 {
167   if (!test_results_index)
168     test_results_index = 0
169   test_results_list[test_results_index] = result
170   test_results_index += 1
171   test_results_seen[result] = 1;
172 }
173
174 # Whether the test script should be re-run by "make recheck".
175 function must_recheck()
176 {
177   for (k in test_results_seen)
178     if (k != "XFAIL" && k != "PASS" && k != "SKIP")
179       return 1
180   return 0
181 }
182
183 # Whether the content of the log file associated to this test should
184 # be copied into the "global" test-suite.log.
185 function copy_in_global_log()
186 {
187   for (k in test_results_seen)
188     if (k != "PASS")
189       return 1
190   return 0
191 }
192
193 # FIXME: this can certainly be improved ...
194 function get_global_test_result()
195 {
196     if ("ERROR" in test_results_seen)
197       return "ERROR"
198     if ("FAIL" in test_results_seen || "XPASS" in test_results_seen)
199       return "FAIL"
200     all_skipped = 1
201     for (k in test_results_seen)
202       if (k != "SKIP")
203         all_skipped = 0
204     if (all_skipped)
205       return "SKIP"
206     return "PASS";
207 }
208
209 function stringify_result_obj(result_obj)
210 {
211   if (result_obj["is_unplanned"] || result_obj["number"] != testno)
212     return "ERROR"
213
214   if (plan_seen == LATE_PLAN)
215     return "ERROR"
216
217   if (result_obj["directive"] == "TODO")
218     return result_obj["is_ok"] ? "XPASS" : "XFAIL"
219
220   if (result_obj["directive"] == "SKIP")
221     return result_obj["is_ok"] ? "SKIP" : COOKED_FAIL;
222
223   if (length(result_obj["directive"]))
224       abort("in function stringify_result_obj()")
225
226   return result_obj["is_ok"] ? COOKED_PASS : COOKED_FAIL
227 }
228
229 function decorate_result(result)
230 {
231   color_name = color_for_result[result]
232   if (color_name)
233     return color_map[color_name] "" result "" color_map["std"]
234   # If we are not using colorized output, or if we do not know how
235   # to colorize the given result, we should return it unchanged.
236   return result
237 }
238
239 function report(result, details)
240 {
241   if (result ~ /^(X?(PASS|FAIL)|SKIP|ERROR)/)
242     {
243       msg = ": " test_script_name
244       add_test_result(result)
245     }
246   else if (result == "#")
247     {
248       msg = " " test_script_name ":"
249     }
250   else
251     {
252       abort("in function report()")
253     }
254   if (length(details))
255     msg = msg " " details
256   # Output on console might be colorized.
257   print decorate_result(result) msg
258   # Log the result in the log file too, to help debugging (this is
259   # especially true when said result is a TAP error or "Bail out!").
260   print result msg | "cat >&3";
261 }
262
263 function testsuite_error(error_message)
264 {
265   report("ERROR", "- " error_message)
266 }
267
268 function handle_tap_result()
269 {
270   details = result_obj["number"];
271   if (length(result_obj["description"]))
272     details = details " " result_obj["description"]
273
274   if (plan_seen == LATE_PLAN)
275     {
276       details = details " # AFTER LATE PLAN";
277     }
278   else if (result_obj["is_unplanned"])
279     {
280        details = details " # UNPLANNED";
281     }
282   else if (result_obj["number"] != testno)
283     {
284        details = sprintf("%s # OUT-OF-ORDER (expecting %d)",
285                          details, testno);
286     }
287   else if (result_obj["directive"])
288     {
289       details = details " # " result_obj["directive"];
290       if (length(result_obj["explanation"]))
291         details = details " " result_obj["explanation"]
292     }
293
294   report(stringify_result_obj(result_obj), details)
295 }
296
297 # `skip_reason` should be empty whenever planned > 0.
298 function handle_tap_plan(planned, skip_reason)
299 {
300   planned += 0 # Avoid getting confused if, say, `planned` is "00"
301   if (length(skip_reason) && planned > 0)
302     abort("in function handle_tap_plan()")
303   if (plan_seen)
304     {
305       # Error, only one plan per stream is acceptable.
306       testsuite_error("multiple test plans")
307       return;
308     }
309   planned_tests = planned
310   # The TAP plan can come before or after *all* the TAP results; we speak
311   # respectively of an "early" or a "late" plan.  If we see the plan line
312   # after at least one TAP result has been seen, assume we have a late
313   # plan; in this case, any further test result seen after the plan will
314   # be flagged as an error.
315   plan_seen = (testno >= 1 ? LATE_PLAN : EARLY_PLAN)
316   # If testno > 0, we have an error ("too many tests run") that will be
317   # automatically dealt with later, so do not worry about it here.  If
318   # $plan_seen is true, we have an error due to a repeated plan, and that
319   # has already been dealt with above.  Otherwise, we have a valid "plan
320   # with SKIP" specification, and should report it as a particular kind
321   # of SKIP result.
322   if (planned == 0 && testno == 0)
323     {
324       if (length(skip_reason))
325         skip_reason = "- "  skip_reason;
326       report("SKIP", skip_reason);
327     }
328 }
329
330 function extract_tap_comment(line)
331 {
332   if (index(line, diag_string) == 1)
333     {
334       # Strip leading `diag_string` from `line`.
335       line = substr(line, length(diag_string) + 1)
336       # And strip any leading and trailing whitespace left.
337       sub("^[ \t]*", "", line)
338       sub("[ \t]*$", "", line)
339       # Return what is left (if any).
340       return line;
341     }
342   return "";
343 }
344
345 # When this function is called, we know that line is a TAP result line,
346 # so that it matches the (perl) RE "^(not )?ok\b".
347 function setup_result_obj(line)
348 {
349   # Get the result, and remove it from the line.
350   result_obj["is_ok"] = (substr(line, 1, 2) == "ok" ? 1 : 0)
351   sub("^(not )?ok[ \t]*", "", line)
352
353   # If the result has an explicit number, get it and strip it; otherwise,
354   # automatically assing the next progresive number to it.
355   if (line ~ /^[0-9]+$/ || line ~ /^[0-9]+[^a-zA-Z0-9_]/)
356     {
357       match(line, "^[0-9]+")
358       # The final `+ 0` is to normalize numbers with leading zeros.
359       result_obj["number"] = substr(line, 1, RLENGTH) + 0
360       line = substr(line, RLENGTH + 1)
361     }
362   else
363     {
364       result_obj["number"] = testno
365     }
366
367   if (plan_seen == LATE_PLAN)
368     # No further test results are acceptable after a "late" TAP plan
369     # has been seen.
370     result_obj["is_unplanned"] = 1
371   else if (plan_seen && testno > planned_tests)
372     result_obj["is_unplanned"] = 1
373   else
374     result_obj["is_unplanned"] = 0
375
376   # Strip trailing and leading whitespace.
377   sub("^[ \t]*", "", line)
378   sub("[ \t]*$", "", line)
379
380   # This will have to be corrected if we have a "TODO"/"SKIP" directive.
381   result_obj["description"] = line
382   result_obj["directive"] = ""
383   result_obj["explanation"] = ""
384
385   # TODO: maybe we should allow a way to escape "#"?
386   if (index(line, "#") == 0)
387     return # No possible directive, nothing more to do.
388
389   # Directives are case-insensitive.
390   rx = "[ \t]*#[ \t]*([tT][oO][dD][oO]|[sS][kK][iI][pP])[ \t]*"
391
392   # See whether we have the directive, and if yes, where.
393   pos = match(line, rx "$")
394   if (!pos)
395     pos = match(line, rx "[^a-zA-Z0-9_]")
396
397   # If there was no TAP directive, we have nothing more to do.
398   if (!pos)
399     return
400
401   # Strip the directive and its explanation (if any) from the test
402   # description.
403   result_obj["description"] = substr(line, 1, pos - 1)
404   # Now remove the test description from the line, that has been dealt
405   # with already.
406   line = substr(line, pos)
407   # Strip the directive, and save its value (normalized to upper case).
408   sub("^[ \t]*#[ \t]*", "", line)
409   result_obj["directive"] = toupper(substr(line, 1, 4))
410   line = substr(line, 5)
411   # Now get the explanation for the directive (if any), with leading
412   # and trailing whitespace removed.
413   sub("^[ \t]*", "", line)
414   sub("[ \t]*$", "", line)
415   result_obj["explanation"] = line
416 }
417
418 function get_test_exit_message(status)
419 {
420   if (status == 0)
421     return ""
422   if (status !~ /^[1-9][0-9]*$/)
423     abort("getting exit status")
424   if (status < 127)
425     exit_details = ""
426   else if (status == 127)
427     exit_details = " (command not found?)"
428   else if (status >= 128 && status <= 255)
429     exit_details = sprintf(" (terminated by signal %d?)", status - 128)
430   else if (status >= 256)
431     exit_details = " (abnormal termination)"
432   return sprintf("exited with status %d%s", status, exit_details)
433 }
434
435 function write_test_results()
436 {
437   print ":global-test-result: " get_global_test_result() > trs_file
438   print ":recheck: "  yn(must_recheck()) > trs_file
439   print ":copy-in-global-log: " yn(copy_in_global_log()) > trs_file
440   for (i = 0; i < test_results_index; i += 1)
441     print ":test-result: " test_results_list[i] > trs_file
442   close(trs_file);
443 }
444
445 BEGIN {
446
447 ## ------- ##
448 ##  SETUP  ##
449 ## ------- ##
450
451 '"$init_colors"'
452
453 # Properly initialized once the TAP plan is seen.
454 planned_tests = 0
455
456 COOKED_PASS = expect_failure ? "XPASS": "PASS";
457 COOKED_FAIL = expect_failure ? "XFAIL": "FAIL";
458
459 # Enumeration-like constants to remember which kind of plan (if any)
460 # has been seen.  It is important that NO_PLAN evaluates "false" as
461 # a boolean.
462 NO_PLAN = 0
463 EARLY_PLAN = 1
464 LATE_PLAN = 2
465
466 testno = 0     # Number of test results seen so far.
467 bailed_out = 0 # Whether a "Bail out!" directive has been seen.
468
469 # Whether the TAP plan has been seen or not, and if yes, which kind
470 # it is ("early" is seen before any test result, "late" otherwise).
471 plan_seen = NO_PLAN
472
473 ## --------- ##
474 ##  PARSING  ##
475 ## --------- ##
476
477 is_first_read = 1
478
479 while (1)
480   {
481     # Involutions required so that we are able to read the exit status
482     # from the last input line.
483     st = getline
484     if (st < 0) # I/O error.
485       fatal("I/O error while reading from input stream")
486     else if (st == 0) # End-of-input
487       {
488         if (is_first_read)
489           abort("in input loop: only one input line")
490         break
491       }
492     if (is_first_read)
493       {
494         is_first_read = 0
495         nextline = $0
496         continue
497       }
498     else
499       {
500         curline = nextline
501         nextline = $0
502         $0 = curline
503       }
504     # Copy any input line verbatim into the log file.
505     print | "cat >&3"
506     # Parsing of TAP input should stop after a "Bail out!" directive.
507     if (bailed_out)
508       continue
509
510     # TAP test result.
511     if ($0 ~ /^(not )?ok$/ || $0 ~ /^(not )?ok[^a-zA-Z0-9_]/)
512       {
513         testno += 1
514         setup_result_obj($0)
515         handle_tap_result()
516       }
517     # TAP plan (normal or "SKIP" without explanation).
518     else if ($0 ~ /^1\.\.[0-9]+[ \t]*$/)
519       {
520         # The next two lines will put the number of planned tests in $0.
521         sub("^1\\.\\.", "")
522         sub("[^0-9]*$", "")
523         handle_tap_plan($0, "")
524         continue
525       }
526     # TAP "SKIP" plan, with an explanation.
527     else if ($0 ~ /^1\.\.0+[ \t]*#/)
528       {
529         # The next lines will put the skip explanation in $0, stripping
530         # any leading and trailing whitespace.  This is a little more
531         # tricky in truth, since we want to also strip a potential leading
532         # "SKIP" string from the message.
533         sub("^[^#]*#[ \t]*(SKIP[: \t][ \t]*)?", "")
534         sub("[ \t]*$", "");
535         handle_tap_plan(0, $0)
536       }
537     # "Bail out!" magic.
538     else if ($0 ~ /^Bail out!/)
539       {
540         bailed_out = 1
541         # Get the bailout message (if any), with leading and trailing
542         # whitespace stripped.  The message remains stored in `$0`.
543         sub("^Bail out![ \t]*", "");
544         sub("[ \t]*$", "");
545         # Format the error message for the
546         bailout_message = "Bail out!"
547         if (length($0))
548           bailout_message = bailout_message " " $0
549         testsuite_error(bailout_message)
550       }
551     # Maybe we have too look for dianogtic comments too.
552     else if (comments != 0)
553       {
554         comment = extract_tap_comment($0);
555         if (length(comment))
556           report("#", comment);
557       }
558   }
559
560 ## -------- ##
561 ##  FINISH  ##
562 ## -------- ##
563
564 # A "Bail out!" directive should cause us to ignore any following TAP
565 # error, as well as a non-zero exit status from the TAP producer.
566 if (!bailed_out)
567   {
568     if (!plan_seen)
569       {
570         testsuite_error("missing test plan")
571       }
572     else if (planned_tests != testno)
573       {
574         bad_amount = testno > planned_tests ? "many" : "few"
575         testsuite_error(sprintf("too %s tests run (expected %d, got %d)",
576                                 bad_amount, planned_tests, testno))
577       }
578     if (!ignore_exit)
579       {
580         # Fetch exit status from the last line.
581         exit_message = get_test_exit_message(nextline)
582         if (exit_message)
583           testsuite_error(exit_message)
584       }
585   }
586
587 write_test_results()
588
589 exit 0
590
591 } # End of "BEGIN" block.
592 '
593
594 # TODO: document that we consume the file descriptor 3 :-(
595 } 3>"$log_file"
596
597 test $? -eq 0 || fatal "I/O or internal error"
598
599 # Local Variables:
600 # mode: shell-script
601 # sh-indentation: 2
602 # eval: (add-hook 'write-file-hooks 'time-stamp)
603 # time-stamp-start: "scriptversion="
604 # time-stamp-format: "%:y-%02m-%02d.%02H"
605 # time-stamp-time-zone: "UTC"
606 # time-stamp-end: "; # UTC"
607 # End: