diff --git a/homeassistant/components/mikrotik/update.py b/homeassistant/components/mikrotik/update.py index 9aa717ac3a7e..099bbeed5a23 100644 --- a/homeassistant/components/mikrotik/update.py +++ b/homeassistant/components/mikrotik/update.py @@ -4,6 +4,7 @@ from dataclasses import dataclass from typing import Any, cast, override from homeassistant.components.update import ( + UpdateDeviceClass, UpdateEntity, UpdateEntityDescription, UpdateEntityFeature, @@ -41,6 +42,7 @@ UPDATES = { MikrotikUpdateEntityDescription( key="fw-update", translation_key="firmware_update", + device_class=UpdateDeviceClass.FIRMWARE, entity_category=EntityCategory.CONFIG, supported_features=UpdateEntityFeature.INSTALL | UpdateEntityFeature.BACKUP, path=UPDATE, @@ -51,6 +53,7 @@ UPDATES = { MikrotikUpdateEntityDescription( key="routerboard-update", translation_key="routerboard_update", + device_class=UpdateDeviceClass.FIRMWARE, entity_category=EntityCategory.CONFIG, supported_features=UpdateEntityFeature.INSTALL, path=ROUTERBOARD, diff --git a/tests/components/mikrotik/snapshots/test_update.ambr b/tests/components/mikrotik/snapshots/test_update.ambr index 33809e99cdeb..5f668901ed1a 100644 --- a/tests/components/mikrotik/snapshots/test_update.ambr +++ b/tests/components/mikrotik/snapshots/test_update.ambr @@ -24,7 +24,7 @@ 'object_id_base': 'RouterBOARD', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterBOARD', 'platform': 'mikrotik', @@ -40,6 +40,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterBOARD', @@ -86,7 +87,7 @@ 'object_id_base': 'RouterOS', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterOS', 'platform': 'mikrotik', @@ -102,6 +103,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterOS', @@ -148,7 +150,7 @@ 'object_id_base': 'RouterBOARD', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterBOARD', 'platform': 'mikrotik', @@ -164,6 +166,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterBOARD', @@ -210,7 +213,7 @@ 'object_id_base': 'RouterOS', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterOS', 'platform': 'mikrotik', @@ -226,6 +229,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterOS', @@ -272,7 +276,7 @@ 'object_id_base': 'RouterBOARD', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterBOARD', 'platform': 'mikrotik', @@ -288,6 +292,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterBOARD', @@ -334,7 +339,7 @@ 'object_id_base': 'RouterOS', 'options': dict({ }), - 'original_device_class': None, + 'original_device_class': , 'original_icon': None, 'original_name': 'RouterOS', 'platform': 'mikrotik', @@ -350,6 +355,7 @@ StateSnapshot({ 'attributes': ReadOnlyDict({ : False, + : 'firmware', : 0, : '/api/brands/integration/mikrotik/icon.png', : 'Mikrotik RouterOS',