adding back ANSI strip

Author: digitalghost-dev

cli_test.go

@@ -3,9 +3,12 @@ package main
 import (
 	"bytes"
 	"os"
+	"regexp"
 	"testing"
 )
 
+var ansiRegex = regexp.MustCompile(`\x1b\[[0-9;]*m`)
+
 func captureOutput(f func()) string {
 	var buf bytes.Buffer
 	stdout := os.Stdout
@@ -21,6 +24,10 @@ func captureOutput(f func()) string {
 	return buf.String()
 }
 
+func stripANSI(input string) string {
+	return ansiRegex.ReplaceAllString(input, "")
+}
+
 func TestRunCLI(t *testing.T) {
 	tests := []struct {
 		name           string
@@ -118,6 +125,8 @@ func TestRunCLI(t *testing.T) {
 				}
 			})
 
+			output = stripANSI(output)
+
 			if !bytes.Contains([]byte(output), []byte(tt.expectedOutput)) {
 				t.Errorf("Expected output to contain %q, got %q", tt.expectedOutput, output)
 			}