mirror of
https://github.com/bettercap/bettercap
synced 2025-07-16 10:03:39 -07:00
45dc257aad
Trivial, but bumps coverage for core from 79.3% to 94.8%. (Local coverage reporting seems to differ from coveralls)
95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
package core
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestIsDumbTerminal(t *testing.T) {
|
|
term := os.Getenv("TERM")
|
|
os.Setenv("TERM", "dumb")
|
|
if !isDumbTerminal() {
|
|
t.Fatal("Expected false when TERM==dumb")
|
|
}
|
|
os.Setenv("TERM", "")
|
|
if !isDumbTerminal() {
|
|
t.Fatal("Expected false when TERM empty")
|
|
}
|
|
os.Setenv("TERM", term)
|
|
}
|
|
|
|
func TestW(t *testing.T) {
|
|
exp := "<3\033[0m"
|
|
got := W("<", "3")
|
|
if got != exp {
|
|
t.Fatalf("expected '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestBold(t *testing.T) {
|
|
exp := "\033[1mgohpers\033[0m"
|
|
got := Bold("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestDim(t *testing.T) {
|
|
exp := "\033[2mgohpers\033[0m"
|
|
got := Dim("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestRed(t *testing.T) {
|
|
exp := "\033[31mgohpers\033[0m"
|
|
got := Red("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestGreen(t *testing.T) {
|
|
exp := "\033[32mgohpers\033[0m"
|
|
got := Green("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestBlue(t *testing.T) {
|
|
exp := "\033[34mgohpers\033[0m"
|
|
got := Blue("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestYellow(t *testing.T) {
|
|
exp := "\033[33mgohpers\033[0m"
|
|
got := Yellow("gohpers")
|
|
if got != exp {
|
|
t.Fatalf("expected path '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestInitSwag(t *testing.T) {
|
|
// Run after other tests to avoid breaking globals
|
|
// Test InitSwag unsets globals when set
|
|
BOLD = "\033[1m"
|
|
InitSwag(true)
|
|
if BOLD != "" {
|
|
t.Fatal("expected BOLD to be empty string")
|
|
}
|
|
term := os.Getenv("TERM")
|
|
os.Setenv("TERM", "dumb")
|
|
BOLD = "\033[1m"
|
|
InitSwag(false)
|
|
if BOLD != "" {
|
|
t.Fatal("expected BOLD to be empty string")
|
|
}
|
|
os.Setenv("TERM", term)
|
|
// Would be good to test BOLD *isn't* unset when we have a TTY
|
|
// but less trivial to stub os.File.Fd() without complicating architecture
|
|
}
|