paker: tests for ColoredUi not showing color with env vars

packer/ui_test.go

@@ -2,6 +2,7 @@ package packer
 
 import (
 	"bytes"
+	"os"
 	"strings"
 	"testing"
 )
@@ -36,6 +37,34 @@ func TestColoredUi(t *testing.T) {
 	}
 }
 
+func TestColoredUi_noColorEnv(t *testing.T) {
+	bufferUi := testUi()
+	ui := &ColoredUi{UiColorYellow, UiColorRed, bufferUi}
+
+	// Set the env var to get rid of the color
+	oldenv := os.Getenv("PACKER_NO_COLOR")
+	os.Setenv("PACKER_NO_COLOR", "1")
+	defer os.Setenv("PACKER_NO_COLOR", oldenv)
+
+	ui.Say("foo")
+	result := readWriter(bufferUi)
+	if result != "foo\n" {
+		t.Fatalf("invalid output: %s", result)
+	}
+
+	ui.Message("foo")
+	result = readWriter(bufferUi)
+	if result != "foo\n" {
+		t.Fatalf("invalid output: %s", result)
+	}
+
+	ui.Error("foo")
+	result = readWriter(bufferUi)
+	if result != "foo\n" {
+		t.Fatalf("invalid output: %s", result)
+	}
+}
+
 func TestTargettedUi(t *testing.T) {
 	bufferUi := testUi()
 	targettedUi := &TargettedUi{