Fix type of Addon.ip_address (#17)

home-assistant-libs · Oct 3, 2024 · f895eba · f895eba
1 parent 00c4938
commit f895eba
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git 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
@@ -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