chore: slightly better type hints

Author: dimaqq

juju/model/_idle.py

@@ -7,7 +7,7 @@
 import logging
 import time
 from dataclasses import dataclass
-from typing import AsyncIterable
+from typing import AbstractSet, AsyncIterable
 
 from ..client._definitions import (
     ApplicationStatus,
@@ -35,7 +35,7 @@ class CheckStatus:
 async def loop(
     foo: AsyncIterable[CheckStatus | None],
     *,
-    apps: frozenset[str],
+    apps: AbstractSet[str],
     wait_for_exact_units: int | None = None,
     wait_for_units: int,
     idle_period: float,
@@ -101,7 +101,7 @@ async def loop(
 def check(
     full_status: FullStatus,
     *,
-    apps: frozenset[str],
+    apps: AbstractSet[str],
     raise_on_error: bool,
     raise_on_blocked: bool,
     status: str | None,

tests/unit/test_idle_loop.py

@@ -19,7 +19,7 @@ async def checks():
     assert await alist(
         loop(
             checks(),
-            apps=frozenset(["a"]),
+            apps={"a"},
             wait_for_units=0,
             idle_period=0,
         )
@@ -38,7 +38,7 @@ async def checks():
         assert await alist(
             loop(
                 checks(),
-                apps=frozenset(["u"]),
+                apps={"u"},
                 wait_for_units=2,
                 idle_period=0,
             )
@@ -71,7 +71,7 @@ async def checks():
     assert await alist(
         loop(
             checks(),
-            apps=frozenset(["u"]),
+            apps={"u"},
             wait_for_units=1,
             wait_for_exact_units=2,
             idle_period=0,
@@ -92,7 +92,7 @@ async def checks():
     assert await alist(
         loop(
             checks(),
-            apps=frozenset(["hexanator"]),
+            apps={"hexanator"},
             wait_for_units=1,
             idle_period=15,
         )
@@ -109,7 +109,7 @@ async def checks():
     assert await alist(
         loop(
             checks(),
-            apps=frozenset(["hexanator"]),
+            apps={"hexanator"},
             wait_for_units=1,
             idle_period=15,
         )