Add filename input to backup APIs (#43)

home-assistant-libs · Jan 28, 2025 · eaa2558 · eaa2558
1 parent c61384b
commit eaa2558
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/aiohasupervisor/models/backups.py b/aiohasupervisor/models/backups.py
@@ -4,6 +4,7 @@
 from dataclasses import dataclass
 from datetime import datetime
 from enum import StrEnum
+from pathlib import PurePath
 
 from .base import DEFAULT, Request, ResponseData
 
@@ -145,6 +146,7 @@ class FullBackupOptions(Request):
     homeassistant_exclude_database: bool | None = None
     background: bool | None = None
     extra: dict | None = None
+    filename: PurePath | None = None
 
 
 @dataclass(frozen=True, slots=True)

diff --git a/tests/test_backups.py b/tests/test_backups.py
@@ -3,6 +3,7 @@
 import asyncio
 from collections.abc import AsyncIterator
 from datetime import UTC, datetime
+from pathlib import PurePath
 from typing import Any
 
 from aioresponses import CallbackResult, aioresponses
@@ -188,6 +189,12 @@ def backup_callback(url: str, **kwargs: dict[str, Any]) -> CallbackResult:  # no
             "9ecf0028",
         ),
         (FullBackupOptions(name="Test", background=False, extra=None), "9ecf0028"),
+        (
+            FullBackupOptions(
+                name="test", background=False, filename=PurePath("backup.tar")
+            ),
+            "9ecf0028",
+        ),
         (None, "9ecf0028"),
     ],
 )
@@ -251,6 +258,15 @@ async def test_backups_full_backup(
             ),
             "9ecf0028",
         ),
+        (
+            PartialBackupOptions(
+                name="Test",
+                background=False,
+                addons={"core_ssh"},
+                filename=PurePath("backup.tar"),
+            ),
+            "9ecf0028",
+        ),
         (
             PartialBackupOptions(name="Test", background=None, addons={"core_ssh"}),
             "9ecf0028",
@@ -520,6 +536,10 @@ async def test_download_backup(
             PartialBackupOptions(homeassistant=True, location=None),
             {"homeassistant": True, "location": None},
         ),
+        (
+            PartialBackupOptions(homeassistant=True, filename=PurePath("backup.tar")),
+            {"homeassistant": True, "filename": "backup.tar"},
+        ),
     ],
 )
 async def test_partial_backup_model(
@@ -547,6 +567,10 @@ async def test_partial_backup_model(
         ),
         (FullBackupOptions(location="test"), {"location": "test"}),
         (FullBackupOptions(location=None), {"location": None}),
+        (
+            FullBackupOptions(filename=PurePath("backup.tar")),
+            {"filename": "backup.tar"},
+        ),
     ],
 )
 async def test_full_backup_model(