mirror of
https://github.com/bettercap/bettercap
synced 2025-08-19 13:09:49 -07:00
Merge pull request #262 from picatz/test-network-aliases
add test for aliases related functions in network package
This commit is contained in:
commit
eaa5dab166
2 changed files with 72 additions and 2 deletions
70
network/aliases_test.go
Normal file
70
network/aliases_test.go
Normal file
|
@ -0,0 +1,70 @@
|
||||||
|
package network
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func buildExampleAliases() *Aliases {
|
||||||
|
return &Aliases{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesLoadAliases(t *testing.T) {
|
||||||
|
err, _ := LoadAliases()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesSaveUnlocked(t *testing.T) {
|
||||||
|
exampleAliases := buildExampleAliases()
|
||||||
|
err := exampleAliases.saveUnlocked()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesSave(t *testing.T) {
|
||||||
|
exampleAliases := buildExampleAliases()
|
||||||
|
err := exampleAliases.Save()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesGet(t *testing.T) {
|
||||||
|
exampleAliases := buildExampleAliases()
|
||||||
|
|
||||||
|
exp := ""
|
||||||
|
got := exampleAliases.Get("pi:ca:tw:as:he:re")
|
||||||
|
|
||||||
|
if got != exp {
|
||||||
|
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesSet(t *testing.T) {
|
||||||
|
exampleAliases := buildExampleAliases()
|
||||||
|
exampleAliases.data = make(map[string]string)
|
||||||
|
|
||||||
|
if exampleAliases.Set("pi:ca:tw:as:he:re", "picat") != nil {
|
||||||
|
t.Error("unable to set alias")
|
||||||
|
}
|
||||||
|
|
||||||
|
if exampleAliases.Get("pi:ca:tw:as:he:re") != "picat" {
|
||||||
|
t.Error("unable to get set alias")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAliasesFind(t *testing.T) {
|
||||||
|
exampleAliases := buildExampleAliases()
|
||||||
|
exampleAliases.data = make(map[string]string)
|
||||||
|
err := exampleAliases.Set("pi:ca:tw:as:he:re", "picat")
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
}
|
||||||
|
mac, found := exampleAliases.Find("picat")
|
||||||
|
if !found {
|
||||||
|
t.Error("unable to find mac address for alias")
|
||||||
|
}
|
||||||
|
if mac != "pi:ca:tw:as:he:re" {
|
||||||
|
t.Error("unable to find correct mac address for alias")
|
||||||
|
}
|
||||||
|
}
|
|
@ -112,9 +112,9 @@ func TestAliases(t *testing.T) {
|
||||||
exampleAlias := "picat"
|
exampleAlias := "picat"
|
||||||
exampleLAN := buildExampleLAN()
|
exampleLAN := buildExampleLAN()
|
||||||
exampleEndpoint := buildExampleEndpoint()
|
exampleEndpoint := buildExampleEndpoint()
|
||||||
exampleLAN.hosts[exampleEndpoint.HwAddress] = exampleEndpoint
|
exampleLAN.hosts["pi:ca:tw:as:he:re"] = exampleEndpoint
|
||||||
exp := exampleAlias
|
exp := exampleAlias
|
||||||
got := exampleLAN.Aliases().Get(exampleEndpoint.HwAddress)
|
got := exampleLAN.Aliases().Get("pi:ca:tw:as:he:re")
|
||||||
if got != exp {
|
if got != exp {
|
||||||
t.Fatalf("expected '%v', got '%v'", exp, got)
|
t.Fatalf("expected '%v', got '%v'", exp, got)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue