From d1c1eb84289ae48be9b276246f7ad9cdd559057e Mon Sep 17 00:00:00 2001
From: Jan-Philipp Benecke <jan-philipp@bnck.me>
Date: Tue, 9 Jan 2024 18:39:31 +0100
Subject: [PATCH] Add test for avoid triggering ping device tracker `home`
 after reload (#107107)

---
 tests/components/ping/test_device_tracker.py | 46 ++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/tests/components/ping/test_device_tracker.py b/tests/components/ping/test_device_tracker.py
index d91cb46da0c..de6b4918262 100644
--- a/tests/components/ping/test_device_tracker.py
+++ b/tests/components/ping/test_device_tracker.py
@@ -1,4 +1,5 @@
 """Test the binary sensor platform of ping."""
+from collections.abc import Generator
 from datetime import timedelta
 from unittest.mock import patch
 
@@ -17,6 +18,16 @@ from homeassistant.util.yaml import dump
 from tests.common import MockConfigEntry, async_fire_time_changed, patch_yaml_files
 
 
+@pytest.fixture
+def entity_registry_enabled_by_default() -> Generator[None, None, None]:
+    """Test fixture that ensures ping device_tracker entities are enabled in the registry."""
+    with patch(
+        "homeassistant.components.ping.device_tracker.PingDeviceTracker.entity_registry_enabled_default",
+        return_value=True,
+    ):
+        yield
+
+
 @pytest.mark.usefixtures("setup_integration")
 async def test_setup_and_update(
     hass: HomeAssistant,
@@ -125,3 +136,38 @@ async def test_import_delete_known_devices(
         await hass.async_block_till_done()
 
         assert len(remove_device_from_config.mock_calls) == 1
+
+
+@pytest.mark.usefixtures("entity_registry_enabled_by_default", "setup_integration")
+async def test_reload_not_triggering_home(
+    hass: HomeAssistant,
+    freezer: FrozenDateTimeFactory,
+    config_entry: MockConfigEntry,
+):
+    """Test if reload/restart does not trigger home when device is unavailable."""
+    assert hass.states.get("device_tracker.10_10_10_10").state == "home"
+
+    with patch(
+        "homeassistant.components.ping.helpers.async_ping",
+        return_value=Host("10.10.10.10", 5, []),
+    ):
+        # device should be "not_home" after consider_home interval
+        freezer.tick(timedelta(minutes=5, seconds=10))
+        async_fire_time_changed(hass)
+        await hass.async_block_till_done()
+
+        assert hass.states.get("device_tracker.10_10_10_10").state == "not_home"
+
+        # reload config entry
+        await hass.config_entries.async_reload(config_entry.entry_id)
+        await hass.async_block_till_done()
+
+    # device should still be "not_home" after a reload
+    assert hass.states.get("device_tracker.10_10_10_10").state == "not_home"
+
+    # device should be "home" after the next refresh
+    freezer.tick(timedelta(seconds=30))
+    async_fire_time_changed(hass)
+    await hass.async_block_till_done()
+
+    assert hass.states.get("device_tracker.10_10_10_10").state == "home"