add test for Remove

This commit is contained in:
Kent Gruber 2018-05-01 22:39:32 -04:00
parent 0024434944
commit aa69d7499a

View file

@ -80,3 +80,16 @@ func TestWiFiList(t *testing.T) {
t.Fatalf("expected '%v', got '%v'", exp, got)
}
}
func TestWiFiRemove(t *testing.T) {
exampleWiFi := buildExampleWiFi()
exampleAP := NewAccessPoint("my_wifi", "ff:ff:ff:ff:ff:ff", 2472, int8(0))
exampleWiFi.aps["ff:ff:ff:ff:ff:f1"] = exampleAP
exampleWiFi.aps["ff:ff:ff:ff:ff:f2"] = exampleAP
exampleWiFi.Remove("ff:ff:ff:ff:ff:f1")
exp := 1
got := len(exampleWiFi.List())
if got != exp {
t.Fatalf("expected '%v', got '%v'", exp, got)
}
}