From 4f9253e9d549aae04e8ca311140d342d95f28e89 Mon Sep 17 00:00:00 2001
From: Francois Laupretre <francois@tekwire.net>
Date: Mon, 21 May 2018 12:49:44 +0200
Subject: [PATCH] Add bats_test_succeeds() function

---
 libexec/bats-exec-test  | 32 ++++++++++++++++++++++++++++++++
 libexec/bats-preprocess | 14 +++++++++-----
 2 files changed, 41 insertions(+), 5 deletions(-)

diff --git a/libexec/bats-exec-test b/libexec/bats-exec-test
index 8f3bd51..f2dab89 100755
--- a/libexec/bats-exec-test
+++ b/libexec/bats-exec-test
@@ -28,6 +28,7 @@ fi
 
 BATS_TEST_DIRNAME="$(dirname "$BATS_TEST_FILENAME")"
 BATS_TEST_NAMES=()
+BATS_TEST_RESULTS=()
 
 load() {
   local name="$1"
@@ -267,12 +268,43 @@ bats_exit_trap() {
   exit "$status"
 }
 
+bats_test_succeeds() {
+  local name status index i ret
+
+  ret=0
+
+  for name in $@
+    do
+    index=''
+    i=0
+    while [ $i -lt ${#BATS_TEST_NAMES[@]} ; do
+      if [ "${BATS_TEST_NAMES[$i]}" = "test_$name" ] ; then
+        index=$i
+        break
+      fi
+    done
+    if [ -z $index ] ; then
+      echo "Error: $name: this test name does not exist" >&2
+    else
+      status=${BATS_TEST_RESULTS[$index]}
+      if [ -z $status ] ; then
+        echo "Warning: $name: Cannot check status as test didn't run yet"
+      else
+        let ret+=$status
+      fi
+    fi
+  done
+
+  return $ret
+}
+
 bats_perform_tests() {
   echo "1..$#"
   test_number=1
   status=0
   for test_name in "$@"; do
     "$0" $BATS_EXTENDED_SYNTAX "$BATS_TEST_FILENAME" "$test_name" "$test_number" || status=1
+    BATS_TEST_RESULTS[$test_number]=$status
     let test_number+=1
   done
   exit "$status"
diff --git a/libexec/bats-preprocess b/libexec/bats-preprocess
index 04297ed..ef6546f 100755
--- a/libexec/bats-preprocess
+++ b/libexec/bats-preprocess
@@ -31,15 +31,19 @@ encode_name() {
 
 tests=()
 index=0
-pattern='^ *@test  *([^ ].*)  *\{ *(.*)$'
+pattern='^ *@test(\[([^ ])[^ ]*\])?  *([^ ].*)  *\{ *(.*)$'
 
 while IFS= read -r line; do
   let index+=1
   if [[ "$line" =~ $pattern ]]; then
-    quoted_name="${BASH_REMATCH[1]}"
-    body="${BASH_REMATCH[2]}"
-    name="$(eval echo "$quoted_name")"
-    encoded_name="$(encode_name "$name")"
+    quoted_name="${BASH_REMATCH[3]}"
+    if [ -n "${BASH_REMATCH[2]}" ; then
+      encoded_name="test_${BASH_REMATCH[2]}"
+    else
+      name="$(eval echo "$quoted_name")"
+      encoded_name="$(encode_name "$name")"
+    fi
+    body="${BASH_REMATCH[4]}"
     tests["${#tests[@]}"]="$encoded_name"
     echo "${encoded_name}() { bats_test_begin ${quoted_name} ${index}; ${body}"
   else