Add Load Average sensors for HTTP (#28)

custom_components/asuswrt_custom/bridge.py

@@ -466,6 +466,10 @@ async def async_get_available_sensors(self) -> dict[str, dict[str, Any]]:
                 KEY_SENSORS: sensors_cpu,
                 KEY_METHOD: self._get_cpu_usage,
             },
+            SENSORS_TYPE_LOAD_AVG: {
+                KEY_SENSORS: SENSORS_LOAD_AVG,
+                KEY_METHOD: self._get_load_avg,
+            },
             SENSORS_TYPE_MEMORY: {
                 KEY_SENSORS: SENSORS_MEMORY,
                 KEY_METHOD: self._get_memory_usage,
@@ -552,6 +556,15 @@ async def _get_memory_usage(self) -> dict[str, Any]:
 
         return _get_dict(SENSORS_MEMORY, list(memory.values()))
 
+    async def _get_load_avg(self) -> dict[str, Any]:
+        """Fetch cpu load avg information from the router."""
+        try:
+            load_avg = await self._api.async_get_loadavg()
+        except AsusWrtError as exc:
+            raise UpdateFailed(exc) from exc
+
+        return _get_dict(SENSORS_LOAD_AVG, list(load_avg.values()))
+
     async def _get_rates(self) -> dict[str, Any]:
         """Fetch rates information from the router."""
         try:

custom_components/asuswrt_custom/manifest.json

@@ -8,6 +8,6 @@
   "iot_class": "local_polling",
   "issue_tracker": "https://github.com/ollo69/ha_asuswrt_custom/issues",
   "loggers": ["aioasuswrt", "asyncssh", "pyasuswrt"],
-  "requirements": ["aioasuswrt==1.4.0", "pyasuswrt==0.1.16"],
+  "requirements": ["aioasuswrt==1.4.0", "pyasuswrt==0.1.17"],
   "version": "0.8.0"
 }

requirements.txt

@@ -8,4 +8,4 @@ flake8==6.0.0
 isort==5.12.0
 black==23.1.0
 aioasuswrt==1.4.0
-pyasuswrt==0.1.16
+pyasuswrt==0.1.17

requirements_test.txt

@@ -5,4 +5,4 @@ pytest==7.3.1
 pytest-homeassistant-custom-component==0.13.42
 # From our manifest.json for our custom component
 aioasuswrt==1.4.0
-pyasuswrt==0.1.16
+pyasuswrt==0.1.17

tests/test_sensor.py

@@ -73,7 +73,8 @@
 MOCK_CURRENT_TRANSFER_RATES_HTTP = {
     k: v for k, v in enumerate(MOCK_CURRENT_TRANSFER_RATES)
 }
-MOCK_LOAD_AVG = [1.1, 1.2, 1.3]
+MOCK_LOAD_AVG_HTTP = {"load_avg_1": 1.1, "load_avg_5": 1.2, "load_avg_15": 1.3}
+MOCK_LOAD_AVG = list(MOCK_LOAD_AVG_HTTP.values())
 MOCK_MEMORY_USAGE = {
     "mem_usage_perc": 52.48,
     "mem_total": 1048576,
@@ -209,6 +210,9 @@ def mock_controller_connect_http(mock_devices_http):
         service_mock.return_value.async_get_cpu_usage = AsyncMock(
             return_value=MOCK_CPU_USAGE
         )
+        service_mock.return_value.async_get_loadavg = AsyncMock(
+            return_value=MOCK_LOAD_AVG_HTTP
+        )
         service_mock.return_value.async_get_memory_usage = AsyncMock(
             return_value=MOCK_MEMORY_USAGE
         )
@@ -269,6 +273,9 @@ def mock_controller_connect_http_sens_fail():
         service_mock.return_value.async_get_cpu_usage = AsyncMock(
             side_effect=AsusWrtError
         )
+        service_mock.return_value.async_get_loadavg = AsyncMock(
+            side_effect=AsusWrtError
+        )
         service_mock.return_value.async_get_memory_usage = AsyncMock(
             side_effect=AsusWrtError
         )