feat: handler IS_TTY env variable to display color (#322)

* feat: handler IS_TTY env variable to display color Signed-off-by: francois samin <francois.samin@corp.ovh.com>
ovh · Nov 20, 2020 · 3825176 · 3825176
1 parent 7b1a450
commit 3825176
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -339,6 +339,10 @@ $ venom run test.yml
 	  [info] the value of result.systemoutjson is map[foo:bar] (exec.yml:34)
 ```
 
+# Use venom in CI
+
+Venom can be use on dev environement or your CI server.
+To display correctly the venom output, you probably will have to export the environment variable `IS_TTY=true` before running venom.
 
 # Hacking
 

diff --git a/venom_output.go b/venom_output.go
@@ -6,14 +6,23 @@ import (
 	"encoding/xml"
 	"fmt"
 	"io/ioutil"
+	"os"
 	"path"
+	"strings"
 	"time"
 
+	"github.com/fatih/color"
 	tap "github.com/mndrix/tap-go"
 	log "github.com/sirupsen/logrus"
 	yaml "gopkg.in/yaml.v2"
 )
 
+func init() {
+	if strings.ToLower(os.Getenv("IS_TTY")) == "true" || os.Getenv("IS_TTY") == "1" {
+		color.NoColor = false
+	}
+}
+
 // OutputResult output result to sdtout, files...
 func (v *Venom) OutputResult(tests Tests, elapsed time.Duration) error {
 	if v.OutputDir == "" {