From aa69d7499a3dddb64d6e4d7eb3d6e5acdec2e2b4 Mon Sep 17 00:00:00 2001 From: Kent Gruber Date: Tue, 1 May 2018 22:39:32 -0400 Subject: [PATCH] add test for Remove --- network/wifi_test.go | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/network/wifi_test.go b/network/wifi_test.go index 8feb4aaa..160504a1 100644 --- a/network/wifi_test.go +++ b/network/wifi_test.go @@ -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) + } +}