From d2f862b7b9166f921da5f5515bbc72d9b66144ab Mon Sep 17 00:00:00 2001
From: Noah Husby <32528627+noahhusby@users.noreply.github.com>
Date: Tue, 26 Nov 2024 04:55:03 -0500
Subject: [PATCH] Add disconnect/reconnect tests to Cambridge Audio (#131100)

---
 tests/components/cambridge_audio/__init__.py  | 12 +++++++++
 tests/components/cambridge_audio/test_init.py | 26 +++++++++++++++++--
 .../cambridge_audio/test_media_player.py      |  9 +------
 3 files changed, 37 insertions(+), 10 deletions(-)

diff --git a/tests/components/cambridge_audio/__init__.py b/tests/components/cambridge_audio/__init__.py
index f6b5f48d39d..4e11a728f41 100644
--- a/tests/components/cambridge_audio/__init__.py
+++ b/tests/components/cambridge_audio/__init__.py
@@ -1,5 +1,9 @@
 """Tests for the Cambridge Audio integration."""
 
+from unittest.mock import AsyncMock
+
+from aiostreammagic.models import CallbackType
+
 from homeassistant.core import HomeAssistant
 
 from tests.common import MockConfigEntry
@@ -11,3 +15,11 @@ async def setup_integration(hass: HomeAssistant, config_entry: MockConfigEntry)
 
     await hass.config_entries.async_setup(config_entry.entry_id)
     await hass.async_block_till_done()
+
+
+async def mock_state_update(
+    client: AsyncMock, callback_type: CallbackType = CallbackType.STATE
+) -> None:
+    """Trigger a callback in the media player."""
+    for callback in client.register_state_update_callbacks.call_args_list:
+        await callback[0][0](client, callback_type)
diff --git a/tests/components/cambridge_audio/test_init.py b/tests/components/cambridge_audio/test_init.py
index 4a8c1b668e2..a058f7c8b6c 100644
--- a/tests/components/cambridge_audio/test_init.py
+++ b/tests/components/cambridge_audio/test_init.py
@@ -1,8 +1,10 @@
 """Tests for the Cambridge Audio integration."""
 
-from unittest.mock import AsyncMock
+from unittest.mock import AsyncMock, Mock
 
 from aiostreammagic import StreamMagicError
+from aiostreammagic.models import CallbackType
+import pytest
 from syrupy import SnapshotAssertion
 
 from homeassistant.components.cambridge_audio.const import DOMAIN
@@ -10,7 +12,7 @@ from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import device_registry as dr
 
-from . import setup_integration
+from . import mock_state_update, setup_integration
 
 from tests.common import MockConfigEntry
 
@@ -43,3 +45,23 @@ async def test_device_info(
     )
     assert device_entry is not None
     assert device_entry == snapshot
+
+
+async def test_disconnect_reconnect_log(
+    hass: HomeAssistant,
+    snapshot: SnapshotAssertion,
+    mock_stream_magic_client: AsyncMock,
+    mock_config_entry: MockConfigEntry,
+    device_registry: dr.DeviceRegistry,
+    caplog: pytest.LogCaptureFixture,
+) -> None:
+    """Test device registry integration."""
+    await setup_integration(hass, mock_config_entry)
+
+    mock_stream_magic_client.is_connected = Mock(return_value=False)
+    await mock_state_update(mock_stream_magic_client, CallbackType.CONNECTION)
+    assert "Disconnected from device at 192.168.20.218" in caplog.text
+
+    mock_stream_magic_client.is_connected = Mock(return_value=True)
+    await mock_state_update(mock_stream_magic_client, CallbackType.CONNECTION)
+    assert "Reconnected to device at 192.168.20.218" in caplog.text
diff --git a/tests/components/cambridge_audio/test_media_player.py b/tests/components/cambridge_audio/test_media_player.py
index b857e61c235..bb2ccd1aec4 100644
--- a/tests/components/cambridge_audio/test_media_player.py
+++ b/tests/components/cambridge_audio/test_media_player.py
@@ -7,7 +7,6 @@ from aiostreammagic import (
     ShuffleMode,
     TransportControl,
 )
-from aiostreammagic.models import CallbackType
 import pytest
 
 from homeassistant.components.media_player import (
@@ -49,18 +48,12 @@ from homeassistant.const import (
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import HomeAssistantError, ServiceValidationError
 
-from . import setup_integration
+from . import mock_state_update, setup_integration
 from .const import ENTITY_ID
 
 from tests.common import MockConfigEntry
 
 
-async def mock_state_update(client: AsyncMock) -> None:
-    """Trigger a callback in the media player."""
-    for callback in client.register_state_update_callbacks.call_args_list:
-        await callback[0][0](client, CallbackType.STATE)
-
-
 async def test_entity_supported_features(
     hass: HomeAssistant,
     mock_stream_magic_client: AsyncMock,