From 50e6c83dd87f5897203389f9579d2a8804c0ffb5 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 15 May 2025 19:53:12 +0200
Subject: [PATCH] Fix missing mock in hue v2 bridge tests (#144947)

---
 tests/components/hue/conftest.py         |  2 ++
 tests/components/hue/test_diagnostics.py |  7 ++++++-
 tests/components/hue/test_services.py    | 10 ++++++++++
 3 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/tests/components/hue/conftest.py b/tests/components/hue/conftest.py
index 7fc6c5ae33f..e6ade431ee6 100644
--- a/tests/components/hue/conftest.py
+++ b/tests/components/hue/conftest.py
@@ -254,6 +254,8 @@ async def setup_bridge(
         with patch("homeassistant.components.hue.HueBridge", return_value=mock_bridge):
             await hass.config_entries.async_setup(config_entry.entry_id)
 
+    assert config_entry.state == ConfigEntryState.LOADED
+
 
 async def setup_platform(
     hass: HomeAssistant,
diff --git a/tests/components/hue/test_diagnostics.py b/tests/components/hue/test_diagnostics.py
index 49681601ebf..a9171d2a12a 100644
--- a/tests/components/hue/test_diagnostics.py
+++ b/tests/components/hue/test_diagnostics.py
@@ -3,6 +3,7 @@
 from unittest.mock import Mock
 
 from homeassistant.core import HomeAssistant
+from homeassistant.util.json import JsonArrayType
 
 from .conftest import setup_platform
 
@@ -21,9 +22,13 @@ async def test_diagnostics_v1(
 
 
 async def test_diagnostics_v2(
-    hass: HomeAssistant, hass_client: ClientSessionGenerator, mock_bridge_v2: Mock
+    hass: HomeAssistant,
+    hass_client: ClientSessionGenerator,
+    mock_bridge_v2: Mock,
+    v2_resources_test_data: JsonArrayType,
 ) -> None:
     """Test diagnostics v2."""
+    await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
     mock_bridge_v2.api.get_diagnostics.return_value = {"hello": "world"}
     await setup_platform(hass, mock_bridge_v2, [])
     config_entry = hass.config_entries.async_entries("hue")[0]
diff --git a/tests/components/hue/test_services.py b/tests/components/hue/test_services.py
index 26a4cab8261..2fd8379a73a 100644
--- a/tests/components/hue/test_services.py
+++ b/tests/components/hue/test_services.py
@@ -9,6 +9,7 @@ from homeassistant.components.hue.const import (
     CONF_ALLOW_UNREACHABLE,
 )
 from homeassistant.core import HomeAssistant
+from homeassistant.util.json import JsonArrayType
 
 from .conftest import setup_bridge, setup_component
 
@@ -190,6 +191,7 @@ async def test_hue_multi_bridge_activate_scene_all_respond(
     mock_bridge_v2: Mock,
     mock_config_entry_v1: MockConfigEntry,
     mock_config_entry_v2: MockConfigEntry,
+    v2_resources_test_data: JsonArrayType,
 ) -> None:
     """Test that makes multiple bridges successfully activate a scene."""
     await setup_component(hass)
@@ -198,6 +200,8 @@ async def test_hue_multi_bridge_activate_scene_all_respond(
     mock_api_v1.mock_group_responses.append(GROUP_RESPONSE)
     mock_api_v1.mock_scene_responses.append(SCENE_RESPONSE)
 
+    await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
+
     await setup_bridge(hass, mock_bridge_v1, mock_config_entry_v1)
     await setup_bridge(hass, mock_bridge_v2, mock_config_entry_v2)
 
@@ -224,6 +228,7 @@ async def test_hue_multi_bridge_activate_scene_one_responds(
     mock_bridge_v2: Mock,
     mock_config_entry_v1: MockConfigEntry,
     mock_config_entry_v2: MockConfigEntry,
+    v2_resources_test_data: JsonArrayType,
 ) -> None:
     """Test that makes only one bridge successfully activate a scene."""
     await setup_component(hass)
@@ -232,6 +237,8 @@ async def test_hue_multi_bridge_activate_scene_one_responds(
     mock_api_v1.mock_group_responses.append(GROUP_RESPONSE)
     mock_api_v1.mock_scene_responses.append(SCENE_RESPONSE)
 
+    await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
+
     await setup_bridge(hass, mock_bridge_v1, mock_config_entry_v1)
     await setup_bridge(hass, mock_bridge_v2, mock_config_entry_v2)
 
@@ -257,6 +264,7 @@ async def test_hue_multi_bridge_activate_scene_zero_responds(
     mock_bridge_v2: Mock,
     mock_config_entry_v1: MockConfigEntry,
     mock_config_entry_v2: MockConfigEntry,
+    v2_resources_test_data: JsonArrayType,
 ) -> None:
     """Test that makes no bridge successfully activate a scene."""
     await setup_component(hass)
@@ -264,6 +272,8 @@ async def test_hue_multi_bridge_activate_scene_zero_responds(
     mock_api_v1.mock_group_responses.append(GROUP_RESPONSE)
     mock_api_v1.mock_scene_responses.append(SCENE_RESPONSE)
 
+    await mock_bridge_v2.api.load_test_data(v2_resources_test_data)
+
     await setup_bridge(hass, mock_bridge_v1, mock_config_entry_v1)
     await setup_bridge(hass, mock_bridge_v2, mock_config_entry_v2)