Merge pull request #10 from matryer/nocolorflag

added -nocolor flag support

Author: matryer

README.md

@@ -28,3 +28,11 @@ func Test(t *testing.T) {
 
 }
 ```
+
+## Color
+
+To turn off the colors, run `go test` with the `-nocolor` flag.
+
+```
+go test -nocolor
+```

is.go

@@ -39,6 +39,7 @@ package is
 import (
 	"bufio"
 	"bytes"
+	"flag"
 	"fmt"
 	"io"
 	"os"
@@ -70,20 +71,28 @@ type I struct {
 	colorful bool
 }
 
+var isColorful bool
+
+func init() {
+	noColor := flag.Bool("nocolor", false, "turns off colors")
+	flag.Parse()
+	isColorful = !*noColor
+}
+
 // New makes a new testing helper using the specified
 // T through which failures will be reported.
 // In strict mode, failures call T.FailNow causing the test
 // to be aborted. See NewRelaxed for alternative behavior.
 func New(t T) *I {
-	return &I{t, t.FailNow, os.Stdout, true}
+	return &I{t, t.FailNow, os.Stdout, isColorful}
 }
 
 // NewRelaxed makes a new testing helper using the specified
 // T through which failures will be reported.
 // In relaxed mode, failures call T.Fail allowing
 // multiple failures per test.
 func NewRelaxed(t T) *I {
-	return &I{t, t.Fail, os.Stdout, true}
+	return &I{t, t.Fail, os.Stdout, isColorful}
 }
 
 func (is *I) log(args ...interface{}) {