Fix type cast in snmp (#114795)
parent
6040272c04
commit
1f37774352
|
@ -270,7 +270,7 @@ class SnmpData:
|
|||
"SNMP OID %s received type=%s and data %s",
|
||||
self._baseoid,
|
||||
type(value),
|
||||
bytes(value),
|
||||
value,
|
||||
)
|
||||
if isinstance(value, NoSuchObject):
|
||||
_LOGGER.error(
|
||||
|
|
|
@ -0,0 +1,79 @@
|
|||
"""SNMP sensor tests."""
|
||||
|
||||
from unittest.mock import patch
|
||||
|
||||
from pysnmp.hlapi import Integer32
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers import entity_registry as er
|
||||
from homeassistant.setup import async_setup_component
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def hlapi_mock():
|
||||
"""Mock out 3rd party API."""
|
||||
mock_data = Integer32(-13)
|
||||
with patch(
|
||||
"homeassistant.components.snmp.sensor.getCmd",
|
||||
return_value=(None, None, None, [[mock_data]]),
|
||||
):
|
||||
yield
|
||||
|
||||
|
||||
async def test_basic_config(hass: HomeAssistant) -> None:
|
||||
"""Test basic entity configuration."""
|
||||
|
||||
config = {
|
||||
SENSOR_DOMAIN: {
|
||||
"platform": "snmp",
|
||||
"host": "192.168.1.32",
|
||||
"baseoid": "1.3.6.1.4.1.2021.10.1.3.1",
|
||||
},
|
||||
}
|
||||
|
||||
assert await async_setup_component(hass, SENSOR_DOMAIN, config)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get("sensor.snmp")
|
||||
assert state.state == "-13"
|
||||
assert state.attributes == {"friendly_name": "SNMP"}
|
||||
|
||||
|
||||
async def test_entity_config(hass: HomeAssistant) -> None:
|
||||
"""Test entity configuration."""
|
||||
|
||||
config = {
|
||||
SENSOR_DOMAIN: {
|
||||
# SNMP configuration
|
||||
"platform": "snmp",
|
||||
"host": "192.168.1.32",
|
||||
"baseoid": "1.3.6.1.4.1.2021.10.1.3.1",
|
||||
# Entity configuration
|
||||
"icon": "{{'mdi:one_two_three'}}",
|
||||
"picture": "{{'blabla.png'}}",
|
||||
"device_class": "temperature",
|
||||
"name": "{{'SNMP' + ' ' + 'Sensor'}}",
|
||||
"state_class": "measurement",
|
||||
"unique_id": "very_unique",
|
||||
"unit_of_measurement": "°C",
|
||||
},
|
||||
}
|
||||
|
||||
assert await async_setup_component(hass, SENSOR_DOMAIN, config)
|
||||
await hass.async_block_till_done()
|
||||
|
||||
entity_registry = er.async_get(hass)
|
||||
assert entity_registry.async_get("sensor.snmp_sensor").unique_id == "very_unique"
|
||||
|
||||
state = hass.states.get("sensor.snmp_sensor")
|
||||
assert state.state == "-13"
|
||||
assert state.attributes == {
|
||||
"device_class": "temperature",
|
||||
"entity_picture": "blabla.png",
|
||||
"friendly_name": "SNMP Sensor",
|
||||
"icon": "mdi:one_two_three",
|
||||
"state_class": "measurement",
|
||||
"unit_of_measurement": "°C",
|
||||
}
|
Loading…
Reference in New Issue