mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-09 02:49:43 +00:00
Support share mounts (#4318)
This commit is contained in:
@@ -427,6 +427,67 @@ async def test_backup_media_with_mounts(
|
||||
assert not mount_dir.exists()
|
||||
|
||||
|
||||
async def test_backup_share_with_mounts(
|
||||
coresys: CoreSys,
|
||||
all_dbus_services: dict[str, DBusServiceMock],
|
||||
tmp_supervisor_data,
|
||||
path_extern,
|
||||
):
|
||||
"""Test backing up share folder with mounts."""
|
||||
systemd_service: SystemdService = all_dbus_services["systemd"]
|
||||
systemd_service.response_get_unit = [
|
||||
DBusError("org.freedesktop.systemd1.NoSuchUnit", "error"),
|
||||
"/org/freedesktop/systemd1/unit/tmp_2dyellow_2emount",
|
||||
DBusError("org.freedesktop.systemd1.NoSuchUnit", "error"),
|
||||
"/org/freedesktop/systemd1/unit/tmp_2dyellow_2emount",
|
||||
"/org/freedesktop/systemd1/unit/tmp_2dyellow_2emount",
|
||||
"/org/freedesktop/systemd1/unit/tmp_2dyellow_2emount",
|
||||
]
|
||||
|
||||
# Make some normal test files
|
||||
(test_file_1 := coresys.config.path_share / "test.txt").touch()
|
||||
(test_dir := coresys.config.path_share / "test").mkdir()
|
||||
(test_file_2 := coresys.config.path_share / "test" / "inner.txt").touch()
|
||||
|
||||
# Add a media mount
|
||||
await coresys.mounts.load()
|
||||
await coresys.mounts.create_mount(
|
||||
Mount.from_dict(
|
||||
coresys,
|
||||
{
|
||||
"name": "share_test",
|
||||
"usage": "share",
|
||||
"type": "cifs",
|
||||
"server": "test.local",
|
||||
"share": "test",
|
||||
},
|
||||
)
|
||||
)
|
||||
assert (mount_dir := coresys.config.path_share / "share_test").is_dir()
|
||||
|
||||
# Make a partial backup
|
||||
coresys.core.state = CoreState.RUNNING
|
||||
coresys.hardware.disk.get_disk_free_space = lambda x: 5000
|
||||
backup: Backup = await coresys.backups.do_backup_partial("test", folders=["share"])
|
||||
|
||||
# Remove the mount and wipe the media folder
|
||||
await coresys.mounts.remove_mount("share_test")
|
||||
rmtree(coresys.config.path_share)
|
||||
coresys.config.path_share.mkdir()
|
||||
|
||||
# Restore the backup and check that only the test files we made returned
|
||||
async def mock_async_true(*args, **kwargs):
|
||||
return True
|
||||
|
||||
with patch.object(HomeAssistantCore, "is_running", new=mock_async_true):
|
||||
await coresys.backups.do_restore_partial(backup, folders=["share"])
|
||||
|
||||
assert test_file_1.exists()
|
||||
assert test_dir.is_dir()
|
||||
assert test_file_2.exists()
|
||||
assert not mount_dir.exists()
|
||||
|
||||
|
||||
async def test_full_backup_to_mount(coresys: CoreSys, tmp_supervisor_data, path_extern):
|
||||
"""Test full backup to and restoring from a mount."""
|
||||
(marker := coresys.config.path_homeassistant / "test.txt").touch()
|
||||
|
||||
Reference in New Issue
Block a user