test: add type hints to test_framework

CHANGES.md

@@ -0,0 +1,9 @@
+# 2.8.0
+
+* The type of a `Handle`'s `key` was expanded from `str` to `str|None`
+
+# 2.7.0
+
+* Added Unit.set_ports()
+* Type checks now allow comparing a `JujuVersion` to a `str`
+* Renamed `OpenPort` to `Port` (`OpenPort` remains as an alias)

ops/framework.py

@@ -101,7 +101,7 @@ class Handle:
     under the same parent and kind may have the same key.
     """
 
-    def __init__(self, parent: Optional[Union['Handle', 'Object']], kind: str, key: str):
+    def __init__(self, parent: Optional[Union['Handle', 'Object']], kind: str, key: Optional[str]):
         if isinstance(parent, Object):
             # if it's not an Object, it will be either a Handle (good) or None (no parent)
             parent = parent.handle
@@ -119,7 +119,7 @@ def __init__(self, parent: Optional[Union['Handle', 'Object']], kind: str, key:
             else:
                 self._path = f"{kind}"  # don't need f-string, but consistent with above
 
-    def nest(self, kind: str, key: str) -> 'Handle':
+    def nest(self, kind: str, key: Optional[str]) -> 'Handle':
         """Create a new handle as child of the current one."""
         return Handle(self, kind, key)
 
@@ -143,7 +143,7 @@ def kind(self) -> str:
         return self._kind
 
     @property
-    def key(self) -> str:
+    def key(self) -> Optional[str]:
         """Return the handle's key."""
         return self._key
 
@@ -1061,13 +1061,17 @@ def __init__(self, parent: Object, attr_name: str):
 
     if TYPE_CHECKING:
         @typing.overload
-        def __getattr__(self, key: Literal['on']) -> ObjectEvents:  # type: ignore
+        def __getattr__(self, key: Literal['on']) -> ObjectEvents:
+            pass
+
+        @typing.overload
+        def __getattr__(self, key: str) -> Any:
             pass
 
     def __getattr__(self, key: str) -> Any:
         # "on" is the only reserved key that can't be used in the data map.
         if key == "on":
-            return self._data.on  # type: ignore  # casting won't work for some reason
+            return self._data.on
         if key not in self._data:
             raise AttributeError(f"attribute '{key}' is not stored")
         return _wrap_stored(self._data, self._data[key])

pyproject.toml

@@ -39,6 +39,7 @@ include = ["ops/*.py", "ops/_private/*.py",
     "test/test_testing.py",
     "test/test_storage.py",
     "test/test_charm.py",
+    "test/test_framework.py",
 ]
 pythonVersion = "3.8" # check no python > 3.8 features are used
 pythonPlatform = "All"