Add API for /os/config/swap endpoints (#50)

* Add API for /os/config/swap endpoints

Add methods for endpoints implemented in home-assistant/supervisor#5770

* Subclass SwapInfo from ResponseData

Author: sairon

aiohasupervisor/models/os.py

@@ -124,3 +124,19 @@ class YellowOptions(Options):
     disk_led: bool | None = None
     heartbeat_led: bool | None = None
     power_led: bool | None = None
+
+
+@dataclass(frozen=True, slots=True)
+class SwapInfo(ResponseData):
+    """SwapInfo model."""
+
+    swap_size: str | None
+    swappiness: int | None
+
+
+@dataclass(frozen=True, slots=True)
+class SwapOptions(Options):
+    """SwapOptions model."""
+
+    swap_size: str | None = None
+    swappiness: int | None = None

aiohasupervisor/os.py

@@ -10,6 +10,8 @@
     OSInfo,
     OSUpdate,
     SetBootSlotOptions,
+    SwapInfo,
+    SwapOptions,
     YellowInfo,
     YellowOptions,
 )
@@ -29,6 +31,15 @@ async def update(self, options: OSUpdate | None = None) -> None:
             "os/update", json=options.to_dict() if options else None, timeout=None
         )
 
+    async def swap_info(self) -> SwapInfo:
+        """Get swap settings."""
+        result = await self._client.get("os/config/swap")
+        return SwapInfo.from_dict(result.data)
+
+    async def set_swap_options(self, options: SwapOptions) -> None:
+        """Set swap settings."""
+        await self._client.post("os/config/swap", json=options.to_dict())
+
     async def config_sync(self) -> None:
         """Trigger config reload on OS."""
         await self._client.post("os/config/sync")

tests/fixtures/os_config_swap.json

@@ -0,0 +1,7 @@
+{
+  "result": "ok",
+  "data": {
+    "swappiness": 1,
+    "swap_size": "1G"
+  }
+}

tests/test_os.py

@@ -15,6 +15,7 @@
     SetBootSlotOptions,
     YellowOptions,
 )
+from aiohasupervisor.models.os import SwapOptions
 
 from . import load_fixture
 from .const import SUPERVISOR_URL
@@ -51,6 +52,36 @@ async def test_os_update(
     assert responses.requests.keys() == {("POST", URL(f"{SUPERVISOR_URL}/os/update"))}
 
 
+async def test_os_swap_info(
+    responses: aioresponses, supervisor_client: SupervisorClient
+) -> None:
+    """Test OS config swap API."""
+    responses.get(
+        f"{SUPERVISOR_URL}/os/config/swap",
+        status=200,
+        body=load_fixture("os_config_swap.json"),
+    )
+    info = await supervisor_client.os.swap_info()
+    assert info.swap_size == "1G"
+    assert info.swappiness == 1
+
+
+async def test_os_set_swap_options(
+    responses: aioresponses, supervisor_client: SupervisorClient
+) -> None:
+    """Test OS set swap options API."""
+    responses.post(f"{SUPERVISOR_URL}/os/config/swap", status=200)
+    assert (
+        await supervisor_client.os.set_swap_options(
+            SwapOptions(swap_size="1G", swappiness=20)
+        )
+        is None
+    )
+    assert responses.requests.keys() == {
+        ("POST", URL(f"{SUPERVISOR_URL}/os/config/swap"))
+    }
+
+
 async def test_os_config_sync(
     responses: aioresponses, supervisor_client: SupervisorClient
 ) -> None: