mirror of
https://github.com/bettercap/bettercap
synced 2025-08-14 10:46:57 -07:00
Merge pull request #261 from picatz/test-network-meta
add test for meta related functions in network package
This commit is contained in:
commit
674188283f
1 changed files with 102 additions and 0 deletions
102
network/meta_test.go
Normal file
102
network/meta_test.go
Normal file
|
@ -0,0 +1,102 @@
|
|||
package network
|
||||
|
||||
import (
|
||||
"strings"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func buildExampleMeta() *Meta {
|
||||
return NewMeta()
|
||||
}
|
||||
|
||||
func TestNewMeta(t *testing.T) {
|
||||
exp := len(Meta{}.m)
|
||||
got := len(NewMeta().m)
|
||||
if got != exp {
|
||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaMarshalJSON(t *testing.T) {
|
||||
_, err := buildExampleMeta().MarshalJSON()
|
||||
if err != nil {
|
||||
t.Error("unable to marshal JSON from meta struct")
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaSet(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
example.Set("picat", "<3")
|
||||
if example.m["picat"] != "<3" {
|
||||
t.Error("unable to set meta data in struct")
|
||||
}
|
||||
}
|
||||
|
||||
// TODO document what this does, not too clear,
|
||||
// at least for me today lolololol
|
||||
func TestMetaGetIntsWith(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
example.m["picat"] = "3,"
|
||||
|
||||
exp := []int{4, 3}
|
||||
got := example.GetIntsWith("picat", 4, false)
|
||||
|
||||
if len(exp) != len(got) {
|
||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaSetInts(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
example.SetInts("picat", []int{0, 1})
|
||||
|
||||
exp := strings.Join([]string{"0", "1"}, ",")
|
||||
got := example.m["picat"]
|
||||
|
||||
if exp != got {
|
||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaGetOr(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
dflt := "picat"
|
||||
exp := dflt
|
||||
got := example.GetOr("evilsocket", dflt)
|
||||
if exp != got {
|
||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaEach(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
example.m["picat"] = true
|
||||
example.m["evilsocket"] = true
|
||||
|
||||
count := 0
|
||||
exampleCB := func(name string, value interface{}) {
|
||||
count++
|
||||
}
|
||||
example.Each(exampleCB)
|
||||
|
||||
exp := 2
|
||||
got := count
|
||||
if exp != got {
|
||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMetaEmpty(t *testing.T) {
|
||||
example := buildExampleMeta()
|
||||
|
||||
if !example.Empty() {
|
||||
t.Error("unable to check if filled struct is empty")
|
||||
}
|
||||
|
||||
example.m["picat"] = true //fill struct so not empty
|
||||
|
||||
if example.Empty() {
|
||||
t.Error("unable to check if filled struct is empty")
|
||||
}
|
||||
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue