diff --git a/aiohasupervisor/models/supervisor.py b/aiohasupervisor/models/supervisor.py index d13c63c..ae8abff 100644 --- a/aiohasupervisor/models/supervisor.py +++ b/aiohasupervisor/models/supervisor.py @@ -25,6 +25,7 @@ class SupervisorInfo(ResponseData): debug_block: bool diagnostics: bool | None auto_update: bool + country: str | None @dataclass(frozen=True, slots=True) diff --git a/tests/fixtures/supervisor_info.json b/tests/fixtures/supervisor_info.json index 45772f0..7bc04d6 100644 --- a/tests/fixtures/supervisor_info.json +++ b/tests/fixtures/supervisor_info.json @@ -15,6 +15,7 @@ "debug_block": false, "diagnostics": false, "auto_update": true, + "country": null, "wait_boot": 5, "addons": [ { diff --git a/tests/test_supervisor.py b/tests/test_supervisor.py index fffcdce..cf2ee53 100644 --- a/tests/test_supervisor.py +++ b/tests/test_supervisor.py @@ -42,6 +42,7 @@ async def test_supervisor_info( assert info.healthy is True assert info.logging == "info" assert info.ip_address == IPv4Address("172.30.32.2") + assert info.country is None async def test_supervisor_stats(