From 1b120a258a76d48a98842b114dbd7fdf04114f9f Mon Sep 17 00:00:00 2001
From: Cody Lee <cody.lee@datadoghq.com>
Date: Thu, 2 Jan 2025 11:33:28 -0600
Subject: [PATCH] tests pass locally;

---
 go.mod                                        |  2 +-
 go.sum                                        |  4 +--
 .../integration_test_expectations.yaml        | 20 +++++++++++++
 .../integration_test_expectations.yaml        | 29 +++++++++++++++++++
 4 files changed, 52 insertions(+), 3 deletions(-)

diff --git a/go.mod b/go.mod
index fee506c8..c8b4e76c 100644
--- a/go.mod
+++ b/go.mod
@@ -13,7 +13,7 @@ require (
 	github.com/prometheus/common v0.61.0
 	github.com/spf13/pflag v1.0.6-0.20201009195203-85dd5c8bc61c
 	github.com/stretchr/testify v1.10.0
-	github.com/unpoller/unifi/v5 v5.0.6
+	github.com/unpoller/unifi/v5 v5.0.7
 	golang.org/x/crypto v0.31.0
 	golang.org/x/net v0.33.0
 	golang.org/x/term v0.27.0
diff --git a/go.sum b/go.sum
index 8f6e3923..73e39633 100644
--- a/go.sum
+++ b/go.sum
@@ -75,8 +75,8 @@ github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO
 github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=
 github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
 github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
-github.com/unpoller/unifi/v5 v5.0.6 h1:nokn8k5HOnxcOhLt5d0xzotJ0tvhbKoExAesrWenLhQ=
-github.com/unpoller/unifi/v5 v5.0.6/go.mod h1:G45KRuSH9PFrIUFmDTzWEEM/E/e7GuyXp36AVOfhm7I=
+github.com/unpoller/unifi/v5 v5.0.7 h1:Dj5HY2Nhdic4Ygvh2YYW6QKIZjXCSo9IBzVDKaj86Zg=
+github.com/unpoller/unifi/v5 v5.0.7/go.mod h1:G45KRuSH9PFrIUFmDTzWEEM/E/e7GuyXp36AVOfhm7I=
 github.com/yuin/goldmark v1.3.5/go.mod h1:mwnBkeHKe2W/ZEtQ+71ViKU8L12m81fl3OWwC1Zlc8k=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
diff --git a/pkg/datadogunifi/integration_test_expectations.yaml b/pkg/datadogunifi/integration_test_expectations.yaml
index 6c7894b5..9ada00b0 100644
--- a/pkg/datadogunifi/integration_test_expectations.yaml
+++ b/pkg/datadogunifi/integration_test_expectations.yaml
@@ -346,6 +346,26 @@ gauges:
   - unifi.usw.tx_bytes
   - unifi.usw.upgradeable
   - unifi.usw.uptime
+  - unifi.ubb.loadavg_1
+  - unifi.ubb.mem_used
+  - unifi.ubb.cpu
+  - unifi.ubb.system_uptime
+  - unifi.ubb.probe
+  - unifi.ubb.mem_total
+  - unifi.ubb.mem_buffer
+  - unifi.ubb.bytes
+  - unifi.ubb.loadavg_5
+  - unifi.ubb.state
+  - unifi.ubb.tx_bytes
+  - unifi.ubb.loadavg_15
+  - unifi.ubb.user_num_sta
+  - unifi.ubb.memory
+  - unifi.ubb.rx_bytes
+  - unifi.ubb.last_seen
+  - unifi.ubb.sys
+  - unifi.ubb.uptime
+  - unifi.ubb.mem
+  - unifi.ubb.network
 counts:
   - unifi.collector.num_devices
   - unifi.collector.num_errors
diff --git a/pkg/influxunifi/integration_test_expectations.yaml b/pkg/influxunifi/integration_test_expectations.yaml
index c00cac38..e1e9afda 100644
--- a/pkg/influxunifi/integration_test_expectations.yaml
+++ b/pkg/influxunifi/integration_test_expectations.yaml
@@ -390,6 +390,35 @@ points:
       wifi_tx_latency_mov_max: float
       wifi_tx_latency_mov_min: float
       wifi_tx_latency_mov_total: float
+  ubb:
+    tags:
+      - mac
+      - model
+      - name
+      - serial
+      - site_name
+      - source
+      - type
+      - version
+    fields:
+      bytes: float
+      ip: string
+      last_seen: float
+      rx_bytes: float
+      stat_bytes: float
+      stat_rx_bytes: float
+      stat_rx_crypts: float
+      stat_rx_dropped: float
+      stat_rx_errors: float
+      stat_rx_frags: float
+      stat_rx_packets: float
+      stat_tx_bytes: float
+      stat_tx_dropped: float
+      stat_tx_errors: float
+      stat_tx_packets: float
+      stat_tx_retries: float
+      tx_bytes: float
+      uptime: float
   unifi_alarm:
     tags:
       - action