support users with gotest for coloured test output

1 files changed, 11 insertions, 1 deletions

diff --git a/test.sh b/test.sh
index a1ab4301f..429c4c381 100755
--- a/test.sh
+++ b/test.sh
@@ -3,9 +3,19 @@
 set -e
 echo "" > coverage.txt
 
+use_go_test=false
+if command -v gotest; then
+    use_go_test=true
+fi
+
 for d in $( find ./* -maxdepth 10 ! -path "./vendor*" ! -path "./.git*" ! -path "./scripts*" -type d); do
     if ls $d/*.go &> /dev/null; then
-        go test -v -race -coverprofile=profile.out -covermode=atomic $d
+        args="-v -race -coverprofile=profile.out -covermode=atomic $d"
+        if [ "$use_go_test" == true ]; then
+            gotest $args
+        else
+            go test $args
+        fi
         if [ -f profile.out ]; then
             cat profile.out >> coverage.txt
             rm profile.out