mirror of
https://github.com/bettercap/bettercap
synced 2025-07-11 15:46:59 -07:00
82 lines
1.6 KiB
Go
82 lines
1.6 KiB
Go
package core
|
|
|
|
import (
|
|
"bytes"
|
|
"regexp"
|
|
"testing"
|
|
)
|
|
|
|
func TestViewLen(t *testing.T) {
|
|
exp := 2
|
|
got := viewLen("<3")
|
|
if got != exp {
|
|
t.Fatalf("expected '%d', got '%d'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestMaxLen(t *testing.T) {
|
|
exp := 7
|
|
got := maxLen([]string{"go", "python", "ruby", "crystal"})
|
|
if got != exp {
|
|
t.Fatalf("expected '%d', got '%d'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestAlignLeft(t *testing.T) {
|
|
exp := Alignment(0)
|
|
got := AlignLeft
|
|
if got != exp {
|
|
t.Fatalf("expected '%d', got '%d'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestAlignCenter(t *testing.T) {
|
|
exp := Alignment(1)
|
|
got := AlignCenter
|
|
if got != exp {
|
|
t.Fatalf("expected '%d', got '%d'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestAlignRight(t *testing.T) {
|
|
exp := Alignment(2)
|
|
got := AlignRight
|
|
if got != exp {
|
|
t.Fatalf("expected '%d', got '%d'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestGetPads(t *testing.T) {
|
|
lPadExp := -9
|
|
rPadExp := -8
|
|
lPadGot, rPadGot := getPads("Pikachu, thunderbolt!", 3, AlignCenter)
|
|
if rPadGot != rPadExp {
|
|
t.Fatalf("expected '%d', got '%d'", rPadExp, rPadGot)
|
|
}
|
|
if lPadGot != lPadExp {
|
|
t.Fatalf("expected '%d', got '%d'", lPadExp, lPadGot)
|
|
}
|
|
}
|
|
|
|
func TestPadded(t *testing.T) {
|
|
exp := "<3"
|
|
got := padded("<3", 1, AlignLeft)
|
|
if got != exp {
|
|
t.Fatalf("expected '%s', got '%s'", exp, got)
|
|
}
|
|
}
|
|
|
|
func TestAsTable(t *testing.T) {
|
|
var b bytes.Buffer
|
|
|
|
AsTable(&b, []string{"top"}, [][]string{[]string{"bottom"}})
|
|
|
|
// look for "+-------+" table style for whichever size
|
|
match, err := regexp.MatchString(`\++-+\+`, b.String())
|
|
if err != nil {
|
|
t.Fatalf("unable to perform regex on table output")
|
|
}
|
|
if !match {
|
|
t.Fatalf("expected table in format not found")
|
|
}
|
|
}
|