From 9765f792abca51f034ec58fdefb3fec03b0e4c0c Mon Sep 17 00:00:00 2001
From: Shorssaud <shaun.orssaud@epitech.eu>
Date: Thu, 16 Feb 2023 16:34:27 +0100
Subject: [PATCH] ADD color text

---
 tests/functional.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/tests/functional.go b/tests/functional.go
index 5fc5118..ec3ac4f 100644
--- a/tests/functional.go
+++ b/tests/functional.go
@@ -3,11 +3,37 @@ package codex
 import (
 	"bytes"
 	"context"
+	"fmt"
 
 	"github.com/guseggert/clustertest/cluster"
 	"github.com/guseggert/clustertest/cluster/basic"
 )
 
+var (
+	Info  = Teal
+	Warn  = Yellow
+	Fatal = Red
+)
+
+var (
+	Black   = Color("\033[1;30m%s\033[0m")
+	Red     = Color("\033[1;31m%s\033[0m")
+	Green   = Color("\033[1;32m%s\033[0m")
+	Yellow  = Color("\033[1;33m%s\033[0m")
+	Purple  = Color("\033[1;34m%s\033[0m")
+	Magenta = Color("\033[1;35m%s\033[0m")
+	Teal    = Color("\033[1;36m%s\033[0m")
+	White   = Color("\033[1;37m%s\033[0m")
+)
+
+func Color(colorString string) func(...interface{}) string {
+	sprint := func(args ...interface{}) string {
+		return fmt.Sprintf(colorString,
+			fmt.Sprint(args...))
+	}
+	return sprint
+}
+
 func getIp(ctx context.Context, node *basic.Node) (string, error) {
 	stdout := &bytes.Buffer{}
 	stderr := &bytes.Buffer{}