ADD FORCE_COLOR environment variable

This is usefull if you want to force color output
even if out of a tty. For instance when running tests
in a gitlab-runner.

Author: pgrange

bash_unit

@@ -165,25 +165,20 @@ usage() {
 
 # Formating
 
-color() {
-  local color=$1
-  shift
-  if [ -t 1 ] ; then echo -en "$color" ; fi
-  if [ $# -gt 0 ]
-  then
-    echo $*
-  else
-    cat
-  fi
-  if [ -t 1 ] ; then echo -en "$NOCOLOR" ; fi
+pretty_success() {
+  pretty_format "$GREEN" "\u2713" "$1"
+}
+
+pretty_failure() {
+  pretty_format "$RED" "\u2717" "$1"
 }
 
 pretty_format() {
   local color="$1"
   local pretty_symbol="$2"
   local alt_symbol="$3"
   local term_utf8=false
-  if test -t 1 && [[ "$LANG" =~ .i*UTF-8.* ]]
+  if is_terminal && [[ "$LANG" =~ .*UTF-8.* ]]
   then
     term_utf8=true
   fi
@@ -198,12 +193,27 @@ pretty_format() {
   ) | color "$color"
 }
 
-pretty_success() {
-  pretty_format "$GREEN" "\u2713" "$1"
+color() {
+  _start_color() {
+    if is_terminal ; then echo -en "$color" ; fi
+  }
+  _stop_color() {
+    if is_terminal ; then echo -en "$NOCOLOR" ; fi
+  }
+  local color=$1
+  shift
+  _start_color
+  if [ $# -gt 0 ]
+  then
+    echo $*
+  else
+    cat
+  fi
+  _stop_color
 }
 
-pretty_failure() {
-  pretty_format "$RED" "\u2717" "$1"
+is_terminal() {
+  [ -t 1 ] || [[ "${FORCE_COLOR}" == true ]]
 }
 
 text_format() {

tests/test_bash_unit.sh

@@ -313,4 +313,4 @@ mute() {
   notify_stderr        () { echo -n ; }
 }
 
-BASH_UNIT=../bash_unit
+BASH_UNIT="eval FORCE_COLOR=false ../bash_unit"

tests/test_doc.sh

@@ -2,9 +2,9 @@
 
 TEST_PATTERN='```bash|```test'
 OUTPUT_PATTERN='```output'
-LANG=C
+LANG=C.UTF-8
 
-BASH_UNIT=./bash_unit
+BASH_UNIT="eval FORCE_COLOR=false ./bash_unit"
 
 prepare_tests() {
   mkdir /tmp/$$

tests/test_tap_format

@@ -91,4 +91,4 @@ bash_unit_out_for_code() {
   $BASH_UNIT -f tap <(cat) | sed -e 's:/dev/fd/[0-9]*:code:g'
 }
 
-BASH_UNIT=../bash_unit
+BASH_UNIT="eval FORCE_COLOR=false ../bash_unit"