jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use MockConfigEntry in hue tests (#112149)

Browse Source 
needed for https://github.com/home-assistant/core/pull/112141
This commit is contained in:
J. Nick Koston 2024-03-03 16:21:33 -10:00 committed by GitHub
parent 331989de4c
commit bef8376f83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 33 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
tests/components/hue/test_bridge.py
View File

@ -1,6 +1,6 @@
"""Test Hue bridge.""" """Test Hue bridge."""
import asyncio import asyncio
from unittest.mock import AsyncMock, Mock, patch from unittest.mock import patch
from aiohttp import client_exceptions from aiohttp import client_exceptions
from aiohue.errors import Unauthorized from aiohue.errors import Unauthorized
@ -12,16 +12,21 @@ from homeassistant.components.hue import bridge
from homeassistant.components.hue.const import ( from homeassistant.components.hue.const import (
CONF_ALLOW_HUE_GROUPS, CONF_ALLOW_HUE_GROUPS,
CONF_ALLOW_UNREACHABLE, CONF_ALLOW_UNREACHABLE,
DOMAIN,
) )
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.exceptions import ConfigEntryNotReady
from tests.common import MockConfigEntry
async def test_bridge_setup_v1(hass: HomeAssistant, mock_api_v1) -> None: async def test_bridge_setup_v1(hass: HomeAssistant, mock_api_v1) -> None:
"""Test a successful setup for V1 bridge.""" """Test a successful setup for V1 bridge."""
config_entry = Mock() config_entry = MockConfigEntry(
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} domain=DOMAIN,
config_entry.options = {CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False} data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1},
options={CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False},
)
with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1), patch.object( with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1), patch.object(
hass.config_entries, "async_forward_entry_setup" hass.config_entries, "async_forward_entry_setup"
@ -39,8 +44,10 @@ async def test_bridge_setup_v1(hass: HomeAssistant, mock_api_v1) -> None:
async def test_bridge_setup_v2(hass: HomeAssistant, mock_api_v2) -> None: async def test_bridge_setup_v2(hass: HomeAssistant, mock_api_v2) -> None:
"""Test a successful setup for V2 bridge.""" """Test a successful setup for V2 bridge."""
config_entry = Mock() config_entry = MockConfigEntry(
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 2} domain=DOMAIN,
data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 2},
)
with patch.object(bridge, "HueBridgeV2", return_value=mock_api_v2), patch.object( with patch.object(bridge, "HueBridgeV2", return_value=mock_api_v2), patch.object(
hass.config_entries, "async_forward_entry_setup" hass.config_entries, "async_forward_entry_setup"
@ -65,9 +72,11 @@ async def test_bridge_setup_v2(hass: HomeAssistant, mock_api_v2) -> None:
async def test_bridge_setup_invalid_api_key(hass: HomeAssistant) -> None: async def test_bridge_setup_invalid_api_key(hass: HomeAssistant) -> None:
"""Test we start config flow if username is no longer whitelisted.""" """Test we start config flow if username is no longer whitelisted."""
entry = Mock() entry = MockConfigEntry(
entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} domain=DOMAIN,
entry.options = {CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False} data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1},
options={CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False},
)
hue_bridge = bridge.HueBridge(hass, entry) hue_bridge = bridge.HueBridge(hass, entry)
with patch.object( with patch.object(
@ -81,9 +90,11 @@ async def test_bridge_setup_invalid_api_key(hass: HomeAssistant) -> None:
async def test_bridge_setup_timeout(hass: HomeAssistant) -> None: async def test_bridge_setup_timeout(hass: HomeAssistant) -> None:
"""Test we retry to connect if we cannot connect.""" """Test we retry to connect if we cannot connect."""
entry = Mock() entry = MockConfigEntry(
entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} domain=DOMAIN,
entry.options = {CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False} data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1},
options={CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False},
)
hue_bridge = bridge.HueBridge(hass, entry) hue_bridge = bridge.HueBridge(hass, entry)
with patch.object( with patch.object(
@ -96,9 +107,11 @@ async def test_bridge_setup_timeout(hass: HomeAssistant) -> None:
async def test_reset_unloads_entry_if_setup(hass: HomeAssistant, mock_api_v1) -> None: async def test_reset_unloads_entry_if_setup(hass: HomeAssistant, mock_api_v1) -> None:
"""Test calling reset while the entry has been setup.""" """Test calling reset while the entry has been setup."""
config_entry = Mock() config_entry = MockConfigEntry(
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} domain=DOMAIN,
config_entry.options = {CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False} data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1},
options={CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False},
)
with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1), patch.object( with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1), patch.object(
hass.config_entries, "async_forward_entry_setup" hass.config_entries, "async_forward_entry_setup"
@ -122,9 +135,11 @@ async def test_reset_unloads_entry_if_setup(hass: HomeAssistant, mock_api_v1) ->
async def test_handle_unauthorized(hass: HomeAssistant, mock_api_v1) -> None: async def test_handle_unauthorized(hass: HomeAssistant, mock_api_v1) -> None:
"""Test handling an unauthorized error on update.""" """Test handling an unauthorized error on update."""
config_entry = Mock(async_setup=AsyncMock()) config_entry = MockConfigEntry(
config_entry.data = {"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1} domain=DOMAIN,
config_entry.options = {CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False} data={"host": "1.2.3.4", "api_key": "mock-api-key", "api_version": 1},
options={CONF_ALLOW_HUE_GROUPS: False, CONF_ALLOW_UNREACHABLE: False},
)
with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1): with patch.object(bridge, "HueBridgeV1", return_value=mock_api_v1):
hue_bridge = bridge.HueBridge(hass, config_entry) hue_bridge = bridge.HueBridge(hass, config_entry)