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

Fix some tests

Browse Source
This commit is contained in:
G Johansson 2025-07-08 17:34:08 +00:00
parent 20a4e91098
commit 9a193c2585
2 changed files with 11 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests/helpers/test_device_registry.py
View File

@ -269,7 +269,7 @@ async def test_multiple_config_subentries(
assert entry.id == entry_id assert entry.id == entry_id
assert entry.config_entries == {config_entry_1.entry_id} assert entry.config_entries == {config_entry_1.entry_id}
assert entry.config_entries_subentries == { assert entry.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1"} config_entry_1.entry_id: {"mock-subentry-id-1-1"}
} }
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
@ -283,7 +283,7 @@ async def test_multiple_config_subentries(
assert entry.id == entry_id assert entry.id == entry_id
assert entry.config_entries == {config_entry_1.entry_id} assert entry.config_entries == {config_entry_1.entry_id}
assert entry.config_entries_subentries == { assert entry.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1", "mock-subentry-id-1-2"} config_entry_1.entry_id: {"mock-subentry-id-1-1", "mock-subentry-id-1-2"}
} }
entry = device_registry.async_get_or_create( entry = device_registry.async_get_or_create(
@ -297,7 +297,7 @@ async def test_multiple_config_subentries(
assert entry.id == entry_id assert entry.id == entry_id
assert entry.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id} assert entry.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id}
assert entry.config_entries_subentries == { assert entry.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1", "mock-subentry-id-1-2"}, config_entry_1.entry_id: {"mock-subentry-id-1-1", "mock-subentry-id-1-2"},
config_entry_2.entry_id: {"mock-subentry-id-2-1"}, config_entry_2.entry_id: {"mock-subentry-id-2-1"},
} }
@ -1850,7 +1850,7 @@ async def test_removing_config_subentries(
assert entry.id == entry4.id assert entry.id == entry4.id
assert entry4.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id} assert entry4.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id}
assert entry4.config_entries_subentries == { assert entry4.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1", "mock-subentry-id-1-2"}, config_entry_1.entry_id: {"mock-subentry-id-1-1", "mock-subentry-id-1-2"},
config_entry_2.entry_id: {"mock-subentry-id-2-1"}, config_entry_2.entry_id: {"mock-subentry-id-2-1"},
} }
@ -1887,7 +1887,7 @@ async def test_removing_config_subentries(
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(update_events) == 8 assert len(update_events) == 7
assert update_events[0].data == { assert update_events[0].data == {
"action": "create", "action": "create",
"device_id": entry.id, "device_id": entry.id,
@ -1904,7 +1904,7 @@ async def test_removing_config_subentries(
"device_id": entry.id, "device_id": entry.id,
"changes": { "changes": {
"config_entries_subentries": { "config_entries_subentries": {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1"} config_entry_1.entry_id: {"mock-subentry-id-1-1"}
}, },
}, },
} }
@ -1915,7 +1915,6 @@ async def test_removing_config_subentries(
"config_entries": {config_entry_1.entry_id}, "config_entries": {config_entry_1.entry_id},
"config_entries_subentries": { "config_entries_subentries": {
config_entry_1.entry_id: { config_entry_1.entry_id: {
None,
"mock-subentry-id-1-1", "mock-subentry-id-1-1",
"mock-subentry-id-1-2", "mock-subentry-id-1-2",
} }
@ -1929,7 +1928,6 @@ async def test_removing_config_subentries(
"changes": { "changes": {
"config_entries_subentries": { "config_entries_subentries": {
config_entry_1.entry_id: { config_entry_1.entry_id: {
None,
"mock-subentry-id-1-1", "mock-subentry-id-1-1",
"mock-subentry-id-1-2", "mock-subentry-id-1-2",
}, },
@ -1940,21 +1938,6 @@ async def test_removing_config_subentries(
}, },
} }
assert update_events[5].data == { assert update_events[5].data == {
"action": "update",
"device_id": entry.id,
"changes": {
"config_entries_subentries": {
config_entry_1.entry_id: {
"mock-subentry-id-1-1",
"mock-subentry-id-1-2",
},
config_entry_2.entry_id: {
"mock-subentry-id-2-1",
},
},
},
}
assert update_events[6].data == {
"action": "update", "action": "update",
"device_id": entry.id, "device_id": entry.id,
"changes": { "changes": {
@ -1970,7 +1953,7 @@ async def test_removing_config_subentries(
"primary_config_entry": config_entry_1.entry_id, "primary_config_entry": config_entry_1.entry_id,
}, },
} }
assert update_events[7].data == { assert update_events[6].data == {
"action": "remove", "action": "remove",
"device_id": entry.id, "device_id": entry.id,
} }
@ -2052,7 +2035,7 @@ async def test_deleted_device_removing_config_subentries(
assert entry.id == entry4.id assert entry.id == entry4.id
assert entry4.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id} assert entry4.config_entries == {config_entry_1.entry_id, config_entry_2.entry_id}
assert entry4.config_entries_subentries == { assert entry4.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1", "mock-subentry-id-1-2"}, config_entry_1.entry_id: {"mock-subentry-id-1-1", "mock-subentry-id-1-2"},
config_entry_2.entry_id: {"mock-subentry-id-2-1"}, config_entry_2.entry_id: {"mock-subentry-id-2-1"},
} }
@ -2080,7 +2063,7 @@ async def test_deleted_device_removing_config_subentries(
"device_id": entry.id, "device_id": entry.id,
"changes": { "changes": {
"config_entries_subentries": { "config_entries_subentries": {
config_entry_1.entry_id: {None, "mock-subentry-id-1-1"} config_entry_1.entry_id: {"mock-subentry-id-1-1"}
}, },
}, },
} }
@ -2091,7 +2074,6 @@ async def test_deleted_device_removing_config_subentries(
"config_entries": {config_entry_1.entry_id}, "config_entries": {config_entry_1.entry_id},
"config_entries_subentries": { "config_entries_subentries": {
config_entry_1.entry_id: { config_entry_1.entry_id: {
None,
"mock-subentry-id-1-1", "mock-subentry-id-1-1",
"mock-subentry-id-1-2", "mock-subentry-id-1-2",
} }

5
tests/helpers/test_entity_registry.py
View File

@ -1719,7 +1719,7 @@ async def test_remove_config_subentry_from_device_removes_entities(
) )
assert device_entry.config_entries == {config_entry_1.entry_id} assert device_entry.config_entries == {config_entry_1.entry_id}
assert device_entry.config_entries_subentries == { assert device_entry.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1", "mock-subentry-id-2"}, config_entry_1.entry_id: {"mock-subentry-id-1", "mock-subentry-id-2"},
} }
# Create one entity entry for each config entry or subentry # Create one entity entry for each config entry or subentry
@ -1749,7 +1749,6 @@ async def test_remove_config_subentry_from_device_removes_entities(
config_subentry_id=None, config_subentry_id=None,
device_id=device_entry.id, device_id=device_entry.id,
) )
assert entity_registry.async_is_registered(entry_1.entity_id) assert entity_registry.async_is_registered(entry_1.entity_id)
assert entity_registry.async_is_registered(entry_2.entity_id) assert entity_registry.async_is_registered(entry_2.entity_id)
assert entity_registry.async_is_registered(entry_3.entity_id) assert entity_registry.async_is_registered(entry_3.entity_id)
@ -1841,7 +1840,7 @@ async def test_remove_config_subentry_from_device_removes_entities_2(
) )
assert device_entry.config_entries == {config_entry_1.entry_id} assert device_entry.config_entries == {config_entry_1.entry_id}
assert device_entry.config_entries_subentries == { assert device_entry.config_entries_subentries == {
config_entry_1.entry_id: {None, "mock-subentry-id-1", "mock-subentry-id-2"}, config_entry_1.entry_id: {"mock-subentry-id-1", "mock-subentry-id-2"},
} }
# Create an entity without config entry or subentry # Create an entity without config entry or subentry