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{}