diff --git a/aiohasupervisor/models/addons.py b/aiohasupervisor/models/addons.py index 12d02be..3bca7b5 100644 --- a/aiohasupervisor/models/addons.py +++ b/aiohasupervisor/models/addons.py @@ -3,6 +3,7 @@ from abc import ABC from dataclasses import dataclass, field from enum import StrEnum +from ipaddress import IPv4Address from typing import Any from mashumaro import field_options @@ -242,7 +243,7 @@ class InstalledAddonComplete( audio_input: str | None audio_output: str | None auto_update: bool - ip_address: bool + ip_address: IPv4Address watchdog: bool devices: list[str] diff --git a/tests/test_addons.py b/tests/test_addons.py index bf026e3..0f9d288 100644 --- a/tests/test_addons.py +++ b/tests/test_addons.py @@ -1,5 +1,7 @@ """Test addons supervisor client.""" +from ipaddress import IPv4Address + from aioresponses import aioresponses from yarl import URL @@ -55,7 +57,7 @@ async def test_addons_info( assert addon.changelog is True assert addon.watchdog is False assert addon.auto_update is False - assert addon.ip_address == "172.30.33.0" + assert addon.ip_address == IPv4Address("172.30.33.0") assert Capability.NET_RAW in addon.privileged assert "not_real" in addon.privileged assert addon.supervisor_api is True