Fix missing mock in hue v2 bridge tests (#144947)

This commit is contained in:
epenet 2025-05-15 19:53:12 +02:00 committed by GitHub
parent 3a58d97496
commit 50e6c83dd8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 18 additions and 1 deletions

View File

@ -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,

View File

@ -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]

View File

@ -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)