From 1d6ecbd1d583f1236e7692e801fa5a38bdd53f1c Mon Sep 17 00:00:00 2001
From: starkillerOG <starkiller.og@gmail.com>
Date: Tue, 31 Dec 2024 22:57:43 +0100
Subject: [PATCH] Change Reolink test switch entity ID (#134339)

---
 tests/components/reolink/test_switch.py | 35 +++++++++++++------------
 1 file changed, 18 insertions(+), 17 deletions(-)

diff --git a/tests/components/reolink/test_switch.py b/tests/components/reolink/test_switch.py
index b2e82040ad4..c6bf7b51213 100644
--- a/tests/components/reolink/test_switch.py
+++ b/tests/components/reolink/test_switch.py
@@ -248,16 +248,17 @@ async def test_switch(
 ) -> None:
     """Test switch entity."""
     reolink_connect.camera_name.return_value = TEST_CAM_NAME
+    reolink_connect.audio_record.return_value = True
 
     with patch("homeassistant.components.reolink.PLATFORMS", [Platform.SWITCH]):
         assert await hass.config_entries.async_setup(config_entry.entry_id)
     await hass.async_block_till_done()
     assert config_entry.state is ConfigEntryState.LOADED
 
-    entity_id = f"{Platform.SWITCH}.{TEST_CAM_NAME}_record"
+    entity_id = f"{Platform.SWITCH}.{TEST_CAM_NAME}_record_audio"
     assert hass.states.get(entity_id).state == STATE_ON
 
-    reolink_connect.recording_enabled.return_value = False
+    reolink_connect.audio_record.return_value = False
     freezer.tick(DEVICE_UPDATE_INTERVAL)
     async_fire_time_changed(hass)
     await hass.async_block_till_done()
@@ -271,9 +272,9 @@ async def test_switch(
         {ATTR_ENTITY_ID: entity_id},
         blocking=True,
     )
-    reolink_connect.set_recording.assert_called_with(0, True)
+    reolink_connect.set_audio.assert_called_with(0, True)
 
-    reolink_connect.set_recording.side_effect = ReolinkError("Test error")
+    reolink_connect.set_audio.side_effect = ReolinkError("Test error")
     with pytest.raises(HomeAssistantError):
         await hass.services.async_call(
             SWITCH_DOMAIN,
@@ -283,16 +284,16 @@ async def test_switch(
         )
 
     # test switch turn off
-    reolink_connect.set_recording.reset_mock(side_effect=True)
+    reolink_connect.set_audio.reset_mock(side_effect=True)
     await hass.services.async_call(
         SWITCH_DOMAIN,
         SERVICE_TURN_OFF,
         {ATTR_ENTITY_ID: entity_id},
         blocking=True,
     )
-    reolink_connect.set_recording.assert_called_with(0, False)
+    reolink_connect.set_audio.assert_called_with(0, False)
 
-    reolink_connect.set_recording.side_effect = ReolinkError("Test error")
+    reolink_connect.set_audio.side_effect = ReolinkError("Test error")
     with pytest.raises(HomeAssistantError):
         await hass.services.async_call(
             SWITCH_DOMAIN,
@@ -301,7 +302,7 @@ async def test_switch(
             blocking=True,
         )
 
-    reolink_connect.set_recording.reset_mock(side_effect=True)
+    reolink_connect.set_audio.reset_mock(side_effect=True)
 
     reolink_connect.camera_online.return_value = False
     freezer.tick(DEVICE_UPDATE_INTERVAL)
@@ -321,17 +322,17 @@ async def test_host_switch(
 ) -> None:
     """Test host switch entity."""
     reolink_connect.camera_name.return_value = TEST_CAM_NAME
-    reolink_connect.recording_enabled.return_value = True
+    reolink_connect.email_enabled.return_value = True
 
     with patch("homeassistant.components.reolink.PLATFORMS", [Platform.SWITCH]):
         assert await hass.config_entries.async_setup(config_entry.entry_id)
     await hass.async_block_till_done()
     assert config_entry.state is ConfigEntryState.LOADED
 
-    entity_id = f"{Platform.SWITCH}.{TEST_NVR_NAME}_record"
+    entity_id = f"{Platform.SWITCH}.{TEST_NVR_NAME}_email_on_event"
     assert hass.states.get(entity_id).state == STATE_ON
 
-    reolink_connect.recording_enabled.return_value = False
+    reolink_connect.email_enabled.return_value = False
     freezer.tick(DEVICE_UPDATE_INTERVAL)
     async_fire_time_changed(hass)
     await hass.async_block_till_done()
@@ -345,9 +346,9 @@ async def test_host_switch(
         {ATTR_ENTITY_ID: entity_id},
         blocking=True,
     )
-    reolink_connect.set_recording.assert_called_with(None, True)
+    reolink_connect.set_email.assert_called_with(None, True)
 
-    reolink_connect.set_recording.side_effect = ReolinkError("Test error")
+    reolink_connect.set_email.side_effect = ReolinkError("Test error")
     with pytest.raises(HomeAssistantError):
         await hass.services.async_call(
             SWITCH_DOMAIN,
@@ -357,16 +358,16 @@ async def test_host_switch(
         )
 
     # test switch turn off
-    reolink_connect.set_recording.reset_mock(side_effect=True)
+    reolink_connect.set_email.reset_mock(side_effect=True)
     await hass.services.async_call(
         SWITCH_DOMAIN,
         SERVICE_TURN_OFF,
         {ATTR_ENTITY_ID: entity_id},
         blocking=True,
     )
-    reolink_connect.set_recording.assert_called_with(None, False)
+    reolink_connect.set_email.assert_called_with(None, False)
 
-    reolink_connect.set_recording.side_effect = ReolinkError("Test error")
+    reolink_connect.set_email.side_effect = ReolinkError("Test error")
     with pytest.raises(HomeAssistantError):
         await hass.services.async_call(
             SWITCH_DOMAIN,
@@ -375,7 +376,7 @@ async def test_host_switch(
             blocking=True,
         )
 
-    reolink_connect.set_recording.reset_mock(side_effect=True)
+    reolink_connect.set_email.reset_mock(side_effect=True)
 
 
 async def test_chime_switch(
-- 
GitLab