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

Fix Shelly mock initialization for sleepy RPC device in tests (#139003)

Browse Source
This commit is contained in:
Shay Levy 2025-02-21 20:52:10 +02:00 committed by GitHub
parent 672df7355c
commit fb5af9acd0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/components/shelly/conftest.py
View File

@ -1,5 +1,6 @@
"""Test configuration for Shelly.""" """Test configuration for Shelly."""
from copy import deepcopy
from unittest.mock import AsyncMock, Mock, PropertyMock, patch from unittest.mock import AsyncMock, Mock, PropertyMock, patch
from aioshelly.ble.const import ( from aioshelly.ble.const import (
@ -592,8 +593,11 @@ def _mock_sleepy_not_initialized_rpc_device():
def initialize_sleepy_rpc_device(device): def initialize_sleepy_rpc_device(device):
"""Initialize a sleepy RPC (Gen2+, Websocket) device.""" """Initialize a sleepy RPC (Gen2+, Websocket) device."""
type(device).requires_auth = PropertyMock() status = deepcopy(MOCK_STATUS_RPC)
type(device).status = PropertyMock(return_value=MOCK_STATUS_RPC) status["sys"]["wakeup_period"] = 1000
type(device).requires_auth = PropertyMock(return_value=False)
type(device).status = PropertyMock(return_value=status)
type(device).event = PropertyMock(return_value={}) type(device).event = PropertyMock(return_value={})
type(device).config = PropertyMock(return_value=MOCK_CONFIG) type(device).config = PropertyMock(return_value=MOCK_CONFIG)
type(device).shelly = PropertyMock(return_value=MOCK_SHELLY_RPC) type(device).shelly = PropertyMock(return_value=MOCK_SHELLY_RPC)
@ -601,14 +605,13 @@ def initialize_sleepy_rpc_device(device):
type(device).firmware_version = PropertyMock( type(device).firmware_version = PropertyMock(
return_value="20240425-141520/1.3.0-ga3fdd3d" return_value="20240425-141520/1.3.0-ga3fdd3d"
) )
type(device).version = PropertyMock("1.3.0") type(device).version = PropertyMock(return_value="1.3.0")
type(device).model = PropertyMock("SPSW-201PE16EU") type(device).model = PropertyMock(return_value="SPSW-201PE16EU")
type(device).xmod_info = PropertyMock(return_value={}) type(device).xmod_info = PropertyMock(return_value={})
type(device).hostname = PropertyMock(return_value="hostname") type(device).hostname = PropertyMock(return_value="hostname")
type(device).name = PropertyMock(return_value="Test Name") type(device).name = PropertyMock(return_value="Test Name")
type(device).firmware_supported = PropertyMock(return_value=True) type(device).firmware_supported = PropertyMock(return_value=True)
device.status["sys"]["wakeup_period"] = 1000
device.connected = True device.connected = True
device.initialized = True device.initialized = True