github/bettercap
1
0
Fork 0
mirror of https://github.com/bettercap/bettercap synced 2025-07-30 11:40:33 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

add test for EachHost

Browse source
This commit is contained in:
Kent Gruber 2018-05-01 15:02:30 -04:00
parent 050eaaf6bc
commit 4b82e917c7
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
network/lan_test.go
View file

@ -143,3 +143,19 @@ func TestHas(t *testing.T) {
t.Error("unable find a known IP address in LAN struct")
}
}
func TestEachHost(t *testing.T) {
exampleBuffer := []string{}
exampleLAN := buildExampleLAN()
exampleEndpoint := buildExampleEndpoint()
exampleLAN.hosts[exampleEndpoint.HwAddress] = exampleEndpoint
exampleCB := func(mac string, e *Endpoint) {
exampleBuffer = append(exampleBuffer, exampleEndpoint.HwAddress)
}
exampleLAN.EachHost(exampleCB)
exp := 1
got := len(exampleBuffer)
if got != exp {
t.Fatalf("expected '%d', got '%d'", exp, got)
}
}