Merge pull request #139 from unpoller/ubb-uci-pari-3

add ubb/uci to mock server and enumerate in devices
unpoller · Jan 2, 2025 · 88aa20c · 88aa20c
2 parents f2086f3 + 0f82d01
commit 88aa20c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/devices.go b/devices.go
@@ -27,6 +27,8 @@ func (u *Unifi) GetDevices(sites []*Site) (*Devices, error) {
 		devices.UDMs = append(devices.UDMs, loopDevices.UDMs...)
 		devices.UXGs = append(devices.UXGs, loopDevices.UXGs...)
 		devices.PDUs = append(devices.PDUs, loopDevices.PDUs...)
+		devices.UBBs = append(devices.UBBs, loopDevices.UBBs...)
+		devices.UCIs = append(devices.UCIs, loopDevices.UCIs...)
 	}
 
 	return devices, nil

diff --git a/mocks/mock_server.go b/mocks/mock_server.go
@@ -208,6 +208,14 @@ func (m *MockHTTPTestServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 			devices = append(devices, d)
 		}
 
+		for _, d := range device.UBBs {
+			devices = append(devices, d)
+		}
+
+		for _, d := range device.UCIs {
+			devices = append(devices, d)
+		}
+
 		respondResultOrErr(w, devices, err, true)
 
 		return