evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use entity registry fixture in octoprint tests (#110887)

Browse source
This commit is contained in:
Jan-Philipp Benecke 2024-02-18 20:55:12 +01:00 • committed by GitHub
parent b358a03379
commit 85587f995b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/components/octoprint/test_sensor.py
View file

@ -9,7 +9,11 @@ from homeassistant.helpers import entity_registry as er
from . import init_integration
async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> None:
async def test_sensors(
hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
entity_registry: er.EntityRegistry,
) -> None:
"""Test the underlying sensors."""
printer = {
"state": {
@ -26,8 +30,6 @@ async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> N
freezer.move_to(datetime(2020, 2, 20, 9, 10, 13, 543, tzinfo=UTC))
await init_integration(hass, "sensor", printer=printer, job=job)
entity_registry = er.async_get(hass)
state = hass.states.get("sensor.octoprint_job_percentage")
assert state is not None
assert state.state == "50"
@ -79,7 +81,9 @@ async def test_sensors(hass: HomeAssistant, freezer: FrozenDateTimeFactory) -> N
async def test_sensors_no_target_temp(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
entity_registry: er.EntityRegistry,
) -> None:
"""Test the underlying sensors."""
printer = {
@ -92,8 +96,6 @@ async def test_sensors_no_target_temp(
freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
await init_integration(hass, "sensor", printer=printer)
entity_registry = er.async_get(hass)
state = hass.states.get("sensor.octoprint_actual_tool1_temp")
assert state is not None
assert state.state == "18.83"
@ -110,7 +112,9 @@ async def test_sensors_no_target_temp(
async def test_sensors_paused(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
entity_registry: er.EntityRegistry,
) -> None:
"""Test the underlying sensors."""
printer = {
@ -128,8 +132,6 @@ async def test_sensors_paused(
freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
await init_integration(hass, "sensor", printer=printer, job=job)
entity_registry = er.async_get(hass)
state = hass.states.get("sensor.octoprint_start_time")
assert state is not None
assert state.state == "unknown"
@ -146,7 +148,9 @@ async def test_sensors_paused(
async def test_sensors_printer_disconnected(
hass: HomeAssistant, freezer: FrozenDateTimeFactory
hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
entity_registry: er.EntityRegistry,
) -> None:
"""Test the underlying sensors."""
job = {
@ -157,8 +161,6 @@ async def test_sensors_printer_disconnected(
freezer.move_to(datetime(2020, 2, 20, 9, 10, 0))
await init_integration(hass, "sensor", printer=None, job=job)
entity_registry = er.async_get(hass)
state = hass.states.get("sensor.octoprint_job_percentage")
assert state is not None
assert state.state == "50"