mirror of
https://github.com/home-assistant/core.git
synced 2025-07-23 21:27:38 +00:00
Simplify UniFi uptime sensor test (#120794)
This commit is contained in:
Robert Svensson
GitHub
parent
be68255ca4
commit
e961ddd5fe
@ -459,115 +459,6 @@ async def test_bandwidth_sensors(
|
|||||||
assert hass.states.get("sensor.wired_client_tx")
|
assert hass.states.get("sensor.wired_client_tx")
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"config_entry_options",
|
|
||||||
[
|
|
||||||
{
|
|
||||||
CONF_ALLOW_BANDWIDTH_SENSORS: False,
|
|
||||||
CONF_ALLOW_UPTIME_SENSORS: True,
|
|
||||||
CONF_TRACK_CLIENTS: False,
|
|
||||||
CONF_TRACK_DEVICES: False,
|
|
||||||
}
|
|
||||||
],
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"client_payload",
|
|
||||||
[
|
|
||||||
[
|
|
||||||
{
|
|
||||||
"mac": "00:00:00:00:00:01",
|
|
||||||
"name": "client1",
|
|
||||||
"oui": "Producer",
|
|
||||||
"uptime": 0,
|
|
||||||
}
|
|
||||||
]
|
|
||||||
],
|
|
||||||
)
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
("initial_uptime", "event_uptime", "small_variation_uptime", "new_uptime"),
|
|
||||||
[
|
|
||||||
# Uptime listed in epoch time should never change
|
|
||||||
(1609462800, 1609462800, 1609462800, 1612141200),
|
|
||||||
# Uptime counted in seconds increases with every event
|
|
||||||
(60, 240, 480, 60),
|
|
||||||
],
|
|
||||||
)
|
|
||||||
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
|
||||||
async def test_uptime_sensors(
|
|
||||||
hass: HomeAssistant,
|
|
||||||
entity_registry: er.EntityRegistry,
|
|
||||||
freezer: FrozenDateTimeFactory,
|
|
||||||
mock_websocket_message,
|
|
||||||
config_entry_options: MappingProxyType[str, Any],
|
|
||||||
config_entry_factory: Callable[[], ConfigEntry],
|
|
||||||
client_payload: list[dict[str, Any]],
|
|
||||||
initial_uptime,
|
|
||||||
event_uptime,
|
|
||||||
small_variation_uptime,
|
|
||||||
new_uptime,
|
|
||||||
) -> None:
|
|
||||||
"""Verify that uptime sensors are working as expected."""
|
|
||||||
uptime_client = client_payload[0]
|
|
||||||
uptime_client["uptime"] = initial_uptime
|
|
||||||
freezer.move_to(datetime(2021, 1, 1, 1, 1, 0, tzinfo=dt_util.UTC))
|
|
||||||
config_entry = await config_entry_factory()
|
|
||||||
|
|
||||||
assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 1
|
|
||||||
assert hass.states.get("sensor.client1_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
assert (
|
|
||||||
entity_registry.async_get("sensor.client1_uptime").entity_category
|
|
||||||
is EntityCategory.DIAGNOSTIC
|
|
||||||
)
|
|
||||||
|
|
||||||
# Verify normal new event doesn't change uptime
|
|
||||||
# 4 minutes have passed
|
|
||||||
uptime_client["uptime"] = event_uptime
|
|
||||||
now = datetime(2021, 1, 1, 1, 4, 0, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.CLIENT, data=uptime_client)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.client1_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
# Verify small variation of uptime (<120 seconds) is ignored
|
|
||||||
# 15 seconds variation after 8 minutes
|
|
||||||
uptime_client["uptime"] = small_variation_uptime
|
|
||||||
now = datetime(2021, 1, 1, 1, 8, 15, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.CLIENT, data=uptime_client)
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.client1_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
# Verify new event change uptime
|
|
||||||
# 1 month has passed
|
|
||||||
uptime_client["uptime"] = new_uptime
|
|
||||||
now = datetime(2021, 2, 1, 1, 1, 0, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.CLIENT, data=uptime_client)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.client1_uptime").state == "2021-02-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
# Disable option
|
|
||||||
options = deepcopy(config_entry_options)
|
|
||||||
options[CONF_ALLOW_UPTIME_SENSORS] = False
|
|
||||||
hass.config_entries.async_update_entry(config_entry, options=options)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 0
|
|
||||||
assert hass.states.get("sensor.client1_uptime") is None
|
|
||||||
|
|
||||||
# Enable option
|
|
||||||
options = deepcopy(config_entry_options)
|
|
||||||
options[CONF_ALLOW_UPTIME_SENSORS] = True
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
hass.config_entries.async_update_entry(config_entry, options=options)
|
|
||||||
await hass.async_block_till_done()
|
|
||||||
|
|
||||||
assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 1
|
|
||||||
assert hass.states.get("sensor.client1_uptime")
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
"config_entry_options",
|
"config_entry_options",
|
||||||
[{CONF_ALLOW_BANDWIDTH_SENSORS: True, CONF_ALLOW_UPTIME_SENSORS: True}],
|
[{CONF_ALLOW_BANDWIDTH_SENSORS: True, CONF_ALLOW_UPTIME_SENSORS: True}],
|
||||||
@ -876,81 +767,6 @@ async def test_outlet_power_readings(
|
|||||||
assert sensor_data.state == expected_update_value
|
assert sensor_data.state == expected_update_value
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
|
||||||
"device_payload",
|
|
||||||
[
|
|
||||||
[
|
|
||||||
{
|
|
||||||
"board_rev": 3,
|
|
||||||
"device_id": "mock-id",
|
|
||||||
"has_fan": True,
|
|
||||||
"fan_level": 0,
|
|
||||||
"ip": "10.0.1.1",
|
|
||||||
"last_seen": 1562600145,
|
|
||||||
"mac": "00:00:00:00:01:01",
|
|
||||||
"model": "US16P150",
|
|
||||||
"name": "Device",
|
|
||||||
"next_interval": 20,
|
|
||||||
"overheating": True,
|
|
||||||
"state": 1,
|
|
||||||
"type": "usw",
|
|
||||||
"upgradable": True,
|
|
||||||
"uptime": 60,
|
|
||||||
"version": "4.0.42.10433",
|
|
||||||
}
|
|
||||||
]
|
|
||||||
],
|
|
||||||
)
|
|
||||||
async def test_device_uptime(
|
|
||||||
hass: HomeAssistant,
|
|
||||||
entity_registry: er.EntityRegistry,
|
|
||||||
mock_websocket_message,
|
|
||||||
config_entry_factory: Callable[[], ConfigEntry],
|
|
||||||
device_payload: list[dict[str, Any]],
|
|
||||||
) -> None:
|
|
||||||
"""Verify that uptime sensors are working as expected."""
|
|
||||||
now = datetime(2021, 1, 1, 1, 1, 0, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
await config_entry_factory()
|
|
||||||
assert len(hass.states.async_entity_ids(SENSOR_DOMAIN)) == 2
|
|
||||||
assert hass.states.get("sensor.device_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
assert (
|
|
||||||
entity_registry.async_get("sensor.device_uptime").entity_category
|
|
||||||
is EntityCategory.DIAGNOSTIC
|
|
||||||
)
|
|
||||||
|
|
||||||
# Verify normal new event doesn't change uptime
|
|
||||||
# 4 minutes have passed
|
|
||||||
device = device_payload[0]
|
|
||||||
device["uptime"] = 240
|
|
||||||
now = datetime(2021, 1, 1, 1, 4, 0, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.DEVICE, data=device)
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.device_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
# Verify small variation of uptime (<120 seconds) is ignored
|
|
||||||
# 15 seconds variation after 8 minutes
|
|
||||||
device = device_payload[0]
|
|
||||||
device["uptime"] = 480
|
|
||||||
now = datetime(2021, 1, 1, 1, 8, 15, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.DEVICE, data=device)
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.device_uptime").state == "2021-01-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
# Verify new event change uptime
|
|
||||||
# 1 month has passed
|
|
||||||
|
|
||||||
device["uptime"] = 60
|
|
||||||
now = datetime(2021, 2, 1, 1, 1, 0, tzinfo=dt_util.UTC)
|
|
||||||
with patch("homeassistant.util.dt.now", return_value=now):
|
|
||||||
mock_websocket_message(message=MessageKey.DEVICE, data=device)
|
|
||||||
|
|
||||||
assert hass.states.get("sensor.device_uptime").state == "2021-02-01T01:00:00+00:00"
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
"device_payload",
|
"device_payload",
|
||||||
[
|
[
|
||||||
@ -1425,3 +1241,186 @@ async def test_sensor_sources(
|
|||||||
assert state.attributes.get(ATTR_STATE_CLASS) == snapshot
|
assert state.attributes.get(ATTR_STATE_CLASS) == snapshot
|
||||||
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == snapshot
|
assert state.attributes.get(ATTR_UNIT_OF_MEASUREMENT) == snapshot
|
||||||
assert state.state == snapshot
|
assert state.state == snapshot
|
||||||
|
|
||||||
|
|
||||||
|
async def _test_uptime_entity(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
freezer: FrozenDateTimeFactory,
|
||||||
|
mock_websocket_message,
|
||||||
|
config_entry_factory: Callable[[], ConfigEntry],
|
||||||
|
payload: dict[str, Any],
|
||||||
|
entity_id: str,
|
||||||
|
message_key: MessageKey,
|
||||||
|
initial_uptime: int,
|
||||||
|
event_uptime: int,
|
||||||
|
small_variation_uptime: int,
|
||||||
|
new_uptime: int,
|
||||||
|
) -> None:
|
||||||
|
"""Verify that uptime entities are working as expected."""
|
||||||
|
payload["uptime"] = initial_uptime
|
||||||
|
freezer.move_to(datetime(2021, 1, 1, 1, 1, 0, tzinfo=dt_util.UTC))
|
||||||
|
config_entry = await config_entry_factory()
|
||||||
|
|
||||||
|
assert hass.states.get(entity_id).state == "2021-01-01T01:00:00+00:00"
|
||||||
|
|
||||||
|
# Verify normal new event doesn't change uptime
|
||||||
|
# 4 minutes have passed
|
||||||
|
|
||||||
|
payload["uptime"] = event_uptime
|
||||||
|
now = datetime(2021, 1, 1, 1, 4, 0, tzinfo=dt_util.UTC)
|
||||||
|
with patch("homeassistant.util.dt.now", return_value=now):
|
||||||
|
mock_websocket_message(message=message_key, data=payload)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert hass.states.get(entity_id).state == "2021-01-01T01:00:00+00:00"
|
||||||
|
|
||||||
|
# Verify small variation of uptime (<120 seconds) is ignored
|
||||||
|
# 15 seconds variation after 8 minutes
|
||||||
|
|
||||||
|
payload["uptime"] = small_variation_uptime
|
||||||
|
now = datetime(2021, 1, 1, 1, 8, 15, tzinfo=dt_util.UTC)
|
||||||
|
with patch("homeassistant.util.dt.now", return_value=now):
|
||||||
|
mock_websocket_message(message=message_key, data=payload)
|
||||||
|
|
||||||
|
assert hass.states.get(entity_id).state == "2021-01-01T01:00:00+00:00"
|
||||||
|
|
||||||
|
# Verify new event change uptime
|
||||||
|
# 1 month has passed
|
||||||
|
|
||||||
|
payload["uptime"] = new_uptime
|
||||||
|
now = datetime(2021, 2, 1, 1, 1, 0, tzinfo=dt_util.UTC)
|
||||||
|
with patch("homeassistant.util.dt.now", return_value=now):
|
||||||
|
mock_websocket_message(message=message_key, data=payload)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert hass.states.get(entity_id).state == "2021-02-01T01:00:00+00:00"
|
||||||
|
|
||||||
|
return config_entry
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("config_entry_options", [{CONF_ALLOW_UPTIME_SENSORS: True}])
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"client_payload",
|
||||||
|
[
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"mac": "00:00:00:00:00:01",
|
||||||
|
"name": "client1",
|
||||||
|
"oui": "Producer",
|
||||||
|
"uptime": 0,
|
||||||
|
}
|
||||||
|
]
|
||||||
|
],
|
||||||
|
)
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
("initial_uptime", "event_uptime", "small_variation_uptime", "new_uptime"),
|
||||||
|
[
|
||||||
|
# Uptime listed in epoch time should never change
|
||||||
|
(1609462800, 1609462800, 1609462800, 1612141200),
|
||||||
|
# Uptime counted in seconds increases with every event
|
||||||
|
(60, 240, 480, 60),
|
||||||
|
],
|
||||||
|
)
|
||||||
|
@pytest.mark.usefixtures("entity_registry_enabled_by_default")
|
||||||
|
async def test_client_uptime(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
entity_registry: er.EntityRegistry,
|
||||||
|
freezer: FrozenDateTimeFactory,
|
||||||
|
mock_websocket_message,
|
||||||
|
config_entry_options: MappingProxyType[str, Any],
|
||||||
|
config_entry_factory: Callable[[], ConfigEntry],
|
||||||
|
client_payload: list[dict[str, Any]],
|
||||||
|
initial_uptime,
|
||||||
|
event_uptime,
|
||||||
|
small_variation_uptime,
|
||||||
|
new_uptime,
|
||||||
|
) -> None:
|
||||||
|
"""Verify that client uptime sensors are working as expected."""
|
||||||
|
config_entry = await _test_uptime_entity(
|
||||||
|
hass,
|
||||||
|
freezer,
|
||||||
|
mock_websocket_message,
|
||||||
|
config_entry_factory,
|
||||||
|
payload=client_payload[0],
|
||||||
|
entity_id="sensor.client1_uptime",
|
||||||
|
message_key=MessageKey.CLIENT,
|
||||||
|
initial_uptime=initial_uptime,
|
||||||
|
event_uptime=event_uptime,
|
||||||
|
small_variation_uptime=small_variation_uptime,
|
||||||
|
new_uptime=new_uptime,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert (
|
||||||
|
entity_registry.async_get("sensor.client1_uptime").entity_category
|
||||||
|
is EntityCategory.DIAGNOSTIC
|
||||||
|
)
|
||||||
|
|
||||||
|
# Disable option
|
||||||
|
options = deepcopy(config_entry_options)
|
||||||
|
options[CONF_ALLOW_UPTIME_SENSORS] = False
|
||||||
|
hass.config_entries.async_update_entry(config_entry, options=options)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert hass.states.get("sensor.client1_uptime") is None
|
||||||
|
|
||||||
|
# Enable option
|
||||||
|
options = deepcopy(config_entry_options)
|
||||||
|
options[CONF_ALLOW_UPTIME_SENSORS] = True
|
||||||
|
hass.config_entries.async_update_entry(config_entry, options=options)
|
||||||
|
await hass.async_block_till_done()
|
||||||
|
|
||||||
|
assert hass.states.get("sensor.client1_uptime")
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"device_payload",
|
||||||
|
[
|
||||||
|
[
|
||||||
|
{
|
||||||
|
"board_rev": 3,
|
||||||
|
"device_id": "mock-id",
|
||||||
|
"has_fan": True,
|
||||||
|
"fan_level": 0,
|
||||||
|
"ip": "10.0.1.1",
|
||||||
|
"last_seen": 1562600145,
|
||||||
|
"mac": "00:00:00:00:01:01",
|
||||||
|
"model": "US16P150",
|
||||||
|
"name": "Device",
|
||||||
|
"next_interval": 20,
|
||||||
|
"overheating": True,
|
||||||
|
"state": 1,
|
||||||
|
"type": "usw",
|
||||||
|
"upgradable": True,
|
||||||
|
"uptime": 60,
|
||||||
|
"version": "4.0.42.10433",
|
||||||
|
}
|
||||||
|
]
|
||||||
|
],
|
||||||
|
)
|
||||||
|
async def test_device_uptime(
|
||||||
|
hass: HomeAssistant,
|
||||||
|
entity_registry: er.EntityRegistry,
|
||||||
|
freezer: FrozenDateTimeFactory,
|
||||||
|
mock_websocket_message,
|
||||||
|
config_entry_factory: Callable[[], ConfigEntry],
|
||||||
|
device_payload: list[dict[str, Any]],
|
||||||
|
) -> None:
|
||||||
|
"""Verify that device uptime sensors are working as expected."""
|
||||||
|
await _test_uptime_entity(
|
||||||
|
hass,
|
||||||
|
freezer,
|
||||||
|
mock_websocket_message,
|
||||||
|
config_entry_factory,
|
||||||
|
payload=device_payload[0],
|
||||||
|
entity_id="sensor.device_uptime",
|
||||||
|
message_key=MessageKey.DEVICE,
|
||||||
|
initial_uptime=60,
|
||||||
|
event_uptime=240,
|
||||||
|
small_variation_uptime=480,
|
||||||
|
new_uptime=60,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert (
|
||||||
|
entity_registry.async_get("sensor.device_uptime").entity_category
|
||||||
|
is EntityCategory.DIAGNOSTIC
|
||||||
|
)
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user