From 80bc6fcd5ccd8fd3ebc8a84dcf4f891aea7ed490 Mon Sep 17 00:00:00 2001
From: abmantis <amfcalt@gmail.com>
Date: Tue, 24 Feb 2026 21:44:36 +0000
Subject: [PATCH] Add unavailable state restore test

---
 tests/components/infrared/test_init.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/tests/components/infrared/test_init.py b/tests/components/infrared/test_init.py
index 19138169740..d8653db986c 100644
--- a/tests/components/infrared/test_init.py
+++ b/tests/components/infrared/test_init.py
@@ -12,7 +12,7 @@ from homeassistant.components.infrared import (
     async_get_emitters,
     async_send_command,
 )
-from homeassistant.const import STATE_UNKNOWN
+from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN
 from homeassistant.core import HomeAssistant, State
 from homeassistant.exceptions import HomeAssistantError
 from homeassistant.setup import async_setup_component
@@ -125,15 +125,21 @@ async def test_async_send_command_component_not_loaded(hass: HomeAssistant) -> N
         await async_send_command(hass, "infrared.some_entity", command)
 
 
+@pytest.mark.parametrize(
+    ("restored_value", "expected_state"),
+    [
+        ("2026-01-01T12:00:00.000+00:00", "2026-01-01T12:00:00.000+00:00"),
+        (STATE_UNAVAILABLE, STATE_UNKNOWN),
+    ],
+)
 async def test_infrared_entity_state_restore(
     hass: HomeAssistant,
     mock_infrared_entity: MockInfraredEntity,
+    restored_value: str,
+    expected_state: str,
 ) -> None:
-    """Test infrared entity restores state from previous session."""
-    previous_timestamp = "2026-01-01T12:00:00.000+00:00"
-    mock_restore_cache(
-        hass, [State("infrared.test_ir_transmitter", previous_timestamp)]
-    )
+    """Test infrared entity state restore."""
+    mock_restore_cache(hass, [State("infrared.test_ir_transmitter", restored_value)])
 
     assert await async_setup_component(hass, DOMAIN, {})
     await hass.async_block_till_done()
@@ -143,4 +149,4 @@ async def test_infrared_entity_state_restore(
 
     state = hass.states.get("infrared.test_ir_transmitter")
     assert state is not None
-    assert state.state == previous_timestamp
+    assert state.state == expected_state