home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-15 10:31:39 +02:00
Code Issues Packages Projects Releases Wiki Activity

cover

Browse Source
This commit is contained in:
J. Nick Koston
2025-06-23 13:24:35 +02:00
parent 13b612848a
commit 4345a6e03b
2 changed files with 17 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/components/esphome/test_entity.py
View File

@@ -12,6 +12,7 @@ from aioesphomeapi import (
DeviceInfo,
SensorInfo,
SensorState,
SubDeviceInfo,
build_unique_id,
)
import pytest
@@ -712,8 +713,8 @@ async def test_entity_assignment_to_sub_device(
# Define sub devices
sub_devices = [
{"device_id": 11111111, "name": "Motion Sensor", "area_id": 0},
{"device_id": 22222222, "name": "Door Sensor", "area_id": 0},
SubDeviceInfo(device_id=11111111, name="Motion Sensor", area_id=0),
SubDeviceInfo(device_id=22222222, name="Door Sensor", area_id=0),
]
device_info = {

26
tests/components/esphome/test_manager.py
View File

@@ -7,6 +7,7 @@ from unittest.mock import AsyncMock, Mock, call
from aioesphomeapi import (
APIClient,
APIConnectionError,
AreaInfo,
DeviceInfo,
EncryptionPlaintextAPIError,
HomeassistantServiceCall,
@@ -14,6 +15,7 @@ from aioesphomeapi import (
InvalidEncryptionKeyAPIError,
LogLevel,
RequiresEncryptionAPIError,
SubDeviceInfo,
UserService,
UserServiceArg,
UserServiceArgType,
@@ -1512,22 +1514,22 @@ async def test_sub_device_creation(
# Define areas
areas = [
{"area_id": 1, "name": "Living Room"},
{"area_id": 2, "name": "Bedroom"},
{"area_id": 3, "name": "Kitchen"},
AreaInfo(area_id=1, name="Living Room"),
AreaInfo(area_id=2, name="Bedroom"),
AreaInfo(area_id=3, name="Kitchen"),
]
# Define sub devices
sub_devices = [
{"device_id": 11111111, "name": "Motion Sensor", "area_id": 1},
{"device_id": 22222222, "name": "Light Switch", "area_id": 1},
{"device_id": 33333333, "name": "Temperature Sensor", "area_id": 2},
SubDeviceInfo(device_id=11111111, name="Motion Sensor", area_id=1),
SubDeviceInfo(device_id=22222222, name="Light Switch", area_id=1),
SubDeviceInfo(device_id=33333333, name="Temperature Sensor", area_id=2),
]
device_info = {
"areas": areas,
"devices": sub_devices,
"area": {"area_id": 0, "name": "Main Hub"},
"area": AreaInfo(area_id=0, name="Main Hub"),
}
device = await mock_esphome_device(
@@ -1578,9 +1580,9 @@ async def test_sub_device_cleanup(
# Initial sub devices
sub_devices_initial = [
{"device_id": 11111111, "name": "Device 1", "area_id": 0},
{"device_id": 22222222, "name": "Device 2", "area_id": 0},
{"device_id": 33333333, "name": "Device 3", "area_id": 0},
SubDeviceInfo(device_id=11111111, name="Device 1", area_id=0),
SubDeviceInfo(device_id=22222222, name="Device 2", area_id=0),
SubDeviceInfo(device_id=33333333, name="Device 3", area_id=0),
]
device_info = {
@@ -1614,8 +1616,8 @@ async def test_sub_device_cleanup(
# Now update with fewer sub devices (device 2 removed)
sub_devices_updated = [
{"device_id": 11111111, "name": "Device 1", "area_id": 0},
{"device_id": 33333333, "name": "Device 3", "area_id": 0},
SubDeviceInfo(device_id=11111111, name="Device 1", area_id=0),
SubDeviceInfo(device_id=33333333, name="Device 3", area_id=0),
]
# Update device info