diff --git a/aiohasupervisor/models/supervisor.py b/aiohasupervisor/models/supervisor.py index ae8abff..488e999 100644 --- a/aiohasupervisor/models/supervisor.py +++ b/aiohasupervisor/models/supervisor.py @@ -53,3 +53,4 @@ class SupervisorOptions(Options): content_trust: bool | None = None force_security: bool | None = None auto_update: bool | None = None + country: str | None = None diff --git a/tests/test_supervisor.py b/tests/test_supervisor.py index cf2ee53..1af17b0 100644 --- a/tests/test_supervisor.py +++ b/tests/test_supervisor.py @@ -107,13 +107,23 @@ async def test_supervisor_options( responses.post(f"{SUPERVISOR_URL}/supervisor/options", status=200) assert ( await supervisor_client.supervisor.set_options( - SupervisorOptions(debug=True, debug_block=True) + SupervisorOptions(debug=True, debug_block=True, country="NL") ) is None ) assert responses.requests.keys() == { ("POST", URL(f"{SUPERVISOR_URL}/supervisor/options")) } + assert ( + request := responses.requests[ + ("POST", URL(f"{SUPERVISOR_URL}/supervisor/options")) + ] + ) + assert request[0].kwargs["json"] == { + "debug": True, + "debug_block": True, + "country": "NL", + } async def test_supervisor_repair(