home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-15 18:41:44 +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, DeviceInfo,
SensorInfo, SensorInfo,
SensorState, SensorState,
SubDeviceInfo,
build_unique_id, build_unique_id,
) )
import pytest import pytest
@@ -712,8 +713,8 @@ async def test_entity_assignment_to_sub_device(
# Define sub devices # Define sub devices
sub_devices = [ sub_devices = [
{"device_id": 11111111, "name": "Motion Sensor", "area_id": 0}, SubDeviceInfo(device_id=11111111, name="Motion Sensor", area_id=0),
{"device_id": 22222222, "name": "Door Sensor", "area_id": 0}, SubDeviceInfo(device_id=22222222, name="Door Sensor", area_id=0),
] ]
device_info = { 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 ( from aioesphomeapi import (
APIClient, APIClient,
APIConnectionError, APIConnectionError,
AreaInfo,
DeviceInfo, DeviceInfo,
EncryptionPlaintextAPIError, EncryptionPlaintextAPIError,
HomeassistantServiceCall, HomeassistantServiceCall,
@@ -14,6 +15,7 @@ from aioesphomeapi import (
InvalidEncryptionKeyAPIError, InvalidEncryptionKeyAPIError,
LogLevel, LogLevel,
RequiresEncryptionAPIError, RequiresEncryptionAPIError,
SubDeviceInfo,
UserService, UserService,
UserServiceArg, UserServiceArg,
UserServiceArgType, UserServiceArgType,
@@ -1512,22 +1514,22 @@ async def test_sub_device_creation(
# Define areas # Define areas
areas = [ areas = [
{"area_id": 1, "name": "Living Room"}, AreaInfo(area_id=1, name="Living Room"),
{"area_id": 2, "name": "Bedroom"}, AreaInfo(area_id=2, name="Bedroom"),
{"area_id": 3, "name": "Kitchen"}, AreaInfo(area_id=3, name="Kitchen"),
] ]
# Define sub devices # Define sub devices
sub_devices = [ sub_devices = [
{"device_id": 11111111, "name": "Motion Sensor", "area_id": 1}, SubDeviceInfo(device_id=11111111, name="Motion Sensor", area_id=1),
{"device_id": 22222222, "name": "Light Switch", "area_id": 1}, SubDeviceInfo(device_id=22222222, name="Light Switch", area_id=1),
{"device_id": 33333333, "name": "Temperature Sensor", "area_id": 2}, SubDeviceInfo(device_id=33333333, name="Temperature Sensor", area_id=2),
] ]
device_info = { device_info = {
"areas": areas, "areas": areas,
"devices": sub_devices, "devices": sub_devices,
"area": {"area_id": 0, "name": "Main Hub"}, "area": AreaInfo(area_id=0, name="Main Hub"),
} }
device = await mock_esphome_device( device = await mock_esphome_device(
@@ -1578,9 +1580,9 @@ async def test_sub_device_cleanup(
# Initial sub devices # Initial sub devices
sub_devices_initial = [ sub_devices_initial = [
{"device_id": 11111111, "name": "Device 1", "area_id": 0}, SubDeviceInfo(device_id=11111111, name="Device 1", area_id=0),
{"device_id": 22222222, "name": "Device 2", "area_id": 0}, SubDeviceInfo(device_id=22222222, name="Device 2", area_id=0),
{"device_id": 33333333, "name": "Device 3", "area_id": 0}, SubDeviceInfo(device_id=33333333, name="Device 3", area_id=0),
] ]
device_info = { device_info = {
@@ -1614,8 +1616,8 @@ async def test_sub_device_cleanup(
# Now update with fewer sub devices (device 2 removed) # Now update with fewer sub devices (device 2 removed)
sub_devices_updated = [ sub_devices_updated = [
{"device_id": 11111111, "name": "Device 1", "area_id": 0}, SubDeviceInfo(device_id=11111111, name="Device 1", area_id=0),
{"device_id": 33333333, "name": "Device 3", "area_id": 0}, SubDeviceInfo(device_id=33333333, name="Device 3", area_id=0),
] ]
# Update device info # Update device info