24 lines
473 B
Go
24 lines
473 B
Go
package shell_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"apigo.cc/go/shell"
|
|
)
|
|
|
|
func TestStripANSI(t *testing.T) {
|
|
input := shell.Red("hello")
|
|
output := shell.StripANSI(input)
|
|
if output != "hello" {
|
|
t.Errorf("Expected 'hello', got '%s'", output)
|
|
}
|
|
}
|
|
|
|
func TestColor(t *testing.T) {
|
|
red := shell.Style(shell.TextRed, "hello", " world")
|
|
if !strings.Contains(red, "hello") || !strings.Contains(red, "world") {
|
|
t.Errorf("Style() should contain string, got %s", red)
|
|
}
|
|
}
|