From 6911941fa28907600a2f3bd24c5acc7713f4a50b Mon Sep 17 00:00:00 2001 From: Kent Gruber Date: Tue, 1 May 2018 22:40:24 -0400 Subject: [PATCH] add test for GetClient --- network/wifi_test.go | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/network/wifi_test.go b/network/wifi_test.go index 260f2b89..5d17141d 100644 --- a/network/wifi_test.go +++ b/network/wifi_test.go @@ -116,3 +116,15 @@ func TestWiFiGet(t *testing.T) { t.Error("unable to get access point from wifi struct with mac address") } } + +func TestWiFiGetClient(t *testing.T) { + exampleWiFi := buildExampleWiFi() + exampleAP := NewAccessPoint("my_wifi", "ff:ff:ff:ff:ff:ff", 2472, int8(0)) + exampleClient := NewStation("my_wifi", "ff:ff:ff:ff:ff:xx", 2472, int8(0)) + exampleAP.clients["ff:ff:ff:ff:ff:xx"] = exampleClient + exampleWiFi.aps["ff:ff:ff:ff:ff:ff"] = exampleAP + _, found := exampleWiFi.GetClient("ff:ff:ff:ff:ff:xx") + if !found { + t.Error("unable to get client from wifi struct with mac address") + } +}