From 6a376009364935af7524751c8bff10097a0ce3e2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Micha=C5=82=20Huryn?= Date: Tue, 12 Jul 2022 10:01:53 +0200 Subject: [PATCH] Remove incorrect device class from blebox button (#75042) * Removed redundant attr device class from button. * Removed irrelevant test for checking buttons device_class. --- homeassistant/components/blebox/button.py | 3 +-- tests/components/blebox/test_button.py | 3 --- 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/homeassistant/components/blebox/button.py b/homeassistant/components/blebox/button.py index 01de5fa56b7..e9ceaac2dc7 100644 --- a/homeassistant/components/blebox/button.py +++ b/homeassistant/components/blebox/button.py @@ -1,7 +1,7 @@ """BleBox button entities implementation.""" from __future__ import annotations -from homeassistant.components.button import ButtonDeviceClass, ButtonEntity +from homeassistant.components.button import ButtonEntity from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -26,7 +26,6 @@ class BleBoxButtonEntity(BleBoxEntity, ButtonEntity): def __init__(self, feature): """Initialize a BleBox button feature.""" super().__init__(feature) - self._attr_device_class = ButtonDeviceClass.UPDATE self._attr_icon = self.get_icon() def get_icon(self): diff --git a/tests/components/blebox/test_button.py b/tests/components/blebox/test_button.py index 22e1b8cb734..8d89f31d260 100644 --- a/tests/components/blebox/test_button.py +++ b/tests/components/blebox/test_button.py @@ -5,7 +5,6 @@ from unittest.mock import PropertyMock import blebox_uniapi import pytest -from homeassistant.components.button import ButtonDeviceClass from homeassistant.const import ATTR_ICON from .conftest import async_setup_entity, mock_feature @@ -50,8 +49,6 @@ async def test_tvliftbox_init(tvliftbox, hass, config, caplog): assert entry.unique_id == "BleBox-tvLiftBox-4a3fdaad90aa-open_or_stop" - assert state.attributes["device_class"] == ButtonDeviceClass.UPDATE - assert state.name == "tvLiftBox-open_or_stop"