From 58b12bfd1356f131f395cf3bd2c21c4766eeb5f4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mislav=20Marohni=C4=87?= <mislav@github.com>
Date: Tue, 25 Oct 2022 16:56:28 +0200
Subject: [PATCH] Fake terminal-like object in tests

---
 pkg/iostreams/iostreams.go | 59 +++++++++++++++++++++++++++++---------
 1 file changed, 46 insertions(+), 13 deletions(-)

diff --git a/pkg/iostreams/iostreams.go b/pkg/iostreams/iostreams.go
index b888c5b5c..3c3f1edf7 100644
--- a/pkg/iostreams/iostreams.go
+++ b/pkg/iostreams/iostreams.go
@@ -37,8 +37,17 @@ type fileReader interface {
 	Fd() uintptr
 }
 
+type term interface {
+	IsTerminalOutput() bool
+	IsColorEnabled() bool
+	Is256ColorSupported() bool
+	IsTrueColorSupported() bool
+	Theme() string
+	Size() (int, int, error)
+}
+
 type IOStreams struct {
-	term ghTerm.Term
+	term term
 
 	In     fileReader
 	Out    fileWriter
@@ -149,10 +158,12 @@ func (s *IOStreams) IsStdoutTTY() bool {
 	if s.stdoutTTYOverride {
 		return s.stdoutIsTTY
 	}
-	if stdout, ok := s.Out.(*os.File); ok {
-		return isTerminal(stdout)
+	// support GH_FORCE_TTY
+	if s.term.IsTerminalOutput() {
+		return true
 	}
-	return false
+	stdout, ok := s.Out.(*os.File)
+	return ok && isCygwinTerminal(stdout.Fd())
 }
 
 func (s *IOStreams) SetStderrTTY(isTTY bool) {
@@ -376,10 +387,6 @@ func (s *IOStreams) TempFile(dir, pattern string) (*os.File, error) {
 func System() *IOStreams {
 	terminal := ghTerm.FromEnv()
 
-	// we avoid using terminal.IsTerminalOutput() in order to additionally support cygwin
-	stdoutIsTTY := isTerminal(os.Stdout)
-	stderrIsTTY := isTerminal(os.Stderr)
-
 	var stdout fileWriter = os.Stdout
 	// On Windows with no virtual terminal processing support, translate ANSI escape
 	// sequences to console syscalls
@@ -396,10 +403,12 @@ func System() *IOStreams {
 		Out:          stdout,
 		ErrOut:       colorable.NewColorable(os.Stderr),
 		pagerCommand: os.Getenv("PAGER"),
-		term:         terminal,
+		term:         &terminal,
 	}
 
-	if stdoutIsTTY && stderrIsTTY {
+	stdoutIsTTY := io.IsStdoutTTY()
+
+	if stdoutIsTTY && io.IsStderrTTY() {
 		io.progressIndicatorEnabled = true
 	}
 
@@ -407,12 +416,35 @@ func System() *IOStreams {
 		io.alternateScreenBufferEnabled = true
 	}
 
-	// prevent duplicate isTerminal queries now that we know the answer
-	io.SetStdoutTTY(stdoutIsTTY)
-	io.SetStderrTTY(stderrIsTTY)
 	return io
 }
 
+type fakeTerm struct{}
+
+func (t fakeTerm) IsTerminalOutput() bool {
+	return false
+}
+
+func (t fakeTerm) IsColorEnabled() bool {
+	return false
+}
+
+func (t fakeTerm) Is256ColorSupported() bool {
+	return false
+}
+
+func (t fakeTerm) IsTrueColorSupported() bool {
+	return false
+}
+
+func (t fakeTerm) Theme() string {
+	return ""
+}
+
+func (t fakeTerm) Size() (int, int, error) {
+	return 80, -1, nil
+}
+
 func Test() (*IOStreams, *bytes.Buffer, *bytes.Buffer, *bytes.Buffer) {
 	in := &bytes.Buffer{}
 	out := &bytes.Buffer{}
@@ -424,6 +456,7 @@ func Test() (*IOStreams, *bytes.Buffer, *bytes.Buffer, *bytes.Buffer) {
 		},
 		Out:    &fdWriter{fd: 1, Writer: out},
 		ErrOut: errOut,
+		term:   &fakeTerm{},
 	}
 	io.SetStdinTTY(false)
 	io.SetStdoutTTY(false)