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

Improve backup store in tests (#135798)

Browse Source
This commit is contained in:
Erik Montnemery 2025-01-16 23:14:19 +01:00 committed by GitHub
parent ef34a33a7b
commit 1fee0a5aa2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 105 additions and 89 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/components/backup/snapshots/test_websocket.ambr
View File

@ -227,7 +227,7 @@
'type': 'result', 'type': 'result',
}) })
# --- # ---
# name: test_config_info[None] # name: test_config_info[storage_data0]
dict({ dict({
'id': 1, 'id': 1,
'result': dict({ 'result': dict({

14
tests/components/backup/test_manager.py
View File

@ -848,11 +848,6 @@ async def test_async_initiate_backup_non_agent_upload_error(
exception: Exception, exception: Exception,
) -> None: ) -> None:
"""Test an unknown or writer upload error during backup generation.""" """Test an unknown or writer upload error during backup generation."""
hass_storage[DOMAIN] = {
"data": {},
"key": DOMAIN,
"version": 1,
}
agent_ids = [LOCAL_AGENT_ID, "test.remote"] agent_ids = [LOCAL_AGENT_ID, "test.remote"]
local_agent = local_backup_platform.CoreLocalBackupAgent(hass) local_agent = local_backup_platform.CoreLocalBackupAgent(hass)
remote_agent = BackupAgentTest("remote", backups=[]) remote_agent = BackupAgentTest("remote", backups=[])
@ -944,7 +939,7 @@ async def test_async_initiate_backup_non_agent_upload_error(
result = await ws_client.receive_json() result = await ws_client.receive_json()
assert result["event"] == {"manager_state": BackupManagerState.IDLE} assert result["event"] == {"manager_state": BackupManagerState.IDLE}
assert not hass_storage[DOMAIN]["data"] assert DOMAIN not in hass_storage
@pytest.mark.usefixtures("mock_backup_generation") @pytest.mark.usefixtures("mock_backup_generation")
@ -1724,11 +1719,6 @@ async def test_receive_backup_non_agent_upload_error(
exception: Exception, exception: Exception,
) -> None: ) -> None:
"""Test non agent upload error during backup receive.""" """Test non agent upload error during backup receive."""
hass_storage[DOMAIN] = {
"data": {},
"key": DOMAIN,
"version": 1,
}
local_agent = local_backup_platform.CoreLocalBackupAgent(hass) local_agent = local_backup_platform.CoreLocalBackupAgent(hass)
remote_agent = BackupAgentTest("remote", backups=[]) remote_agent = BackupAgentTest("remote", backups=[])
@ -1814,7 +1804,7 @@ async def test_receive_backup_non_agent_upload_error(
result = await ws_client.receive_json() result = await ws_client.receive_json()
assert result["event"] == {"manager_state": BackupManagerState.IDLE} assert result["event"] == {"manager_state": BackupManagerState.IDLE}
assert not hass_storage[DOMAIN]["data"] assert DOMAIN not in hass_storage
assert resp.status == 500 assert resp.status == 500
assert open_mock.call_count == 1 assert open_mock.call_count == 1
assert move_mock.call_count == 0 assert move_mock.call_count == 0

38
tests/components/backup/test_websocket.py
View File

@ -906,8 +906,10 @@ async def test_agents_info(
@pytest.mark.parametrize( @pytest.mark.parametrize(
"storage_data", "storage_data",
[ [
None, {},
{ {
"backup": {
"data": {
"backups": {}, "backups": {},
"config": { "config": {
"create_backup": { "create_backup": {
@ -925,7 +927,13 @@ async def test_agents_info(
"schedule": {"state": "daily"}, "schedule": {"state": "daily"},
}, },
}, },
"key": DOMAIN,
"version": 1,
},
},
{ {
"backup": {
"data": {
"backups": {}, "backups": {},
"config": { "config": {
"create_backup": { "create_backup": {
@ -943,7 +951,13 @@ async def test_agents_info(
"schedule": {"state": "never"}, "schedule": {"state": "never"},
}, },
}, },
"key": DOMAIN,
"version": 1,
},
},
{ {
"backup": {
"data": {
"backups": {}, "backups": {},
"config": { "config": {
"create_backup": { "create_backup": {
@ -961,7 +975,13 @@ async def test_agents_info(
"schedule": {"state": "never"}, "schedule": {"state": "never"},
}, },
}, },
"key": DOMAIN,
"version": 1,
},
},
{ {
"backup": {
"data": {
"backups": {}, "backups": {},
"config": { "config": {
"create_backup": { "create_backup": {
@ -979,7 +999,13 @@ async def test_agents_info(
"schedule": {"state": "mon"}, "schedule": {"state": "mon"},
}, },
}, },
"key": DOMAIN,
"version": 1,
},
},
{ {
"backup": {
"data": {
"backups": {}, "backups": {},
"config": { "config": {
"create_backup": { "create_backup": {
@ -997,6 +1023,10 @@ async def test_agents_info(
"schedule": {"state": "sat"}, "schedule": {"state": "sat"},
}, },
}, },
"key": DOMAIN,
"version": 1,
},
},
], ],
) )
async def test_config_info( async def test_config_info(
@ -1007,11 +1037,7 @@ async def test_config_info(
storage_data: dict[str, Any] | None, storage_data: dict[str, Any] | None,
) -> None: ) -> None:
"""Test getting backup config info.""" """Test getting backup config info."""
hass_storage[DOMAIN] = { hass_storage.update(storage_data)
"data": storage_data,
"key": DOMAIN,
"version": 1,
}
await setup_backup_integration(hass) await setup_backup_integration(hass)
await hass.async_block_till_done() await hass.async_block_till_done()