Fix mypy errors: Use Optional types with assertions

The attributes are initialized to None in child __init__, then set to actual values by initialize() method. Using Optional types with assertions at usage points correctly reflects this lifecycle and satisfies mypy

Co-authored-by: mitsha-microsoft <58204159+mitsha-microsoft@users.noreply.github.com>

Author: Copilot

sdk/loadtesting/azure-developer-loadtesting/azure/developer/loadtesting/_operations/_patch.py

@@ -27,17 +27,18 @@
 class LoadTestingPollingMethod(PollingMethod):
     """Base class for custom sync polling methods."""
 
-    _status: str
+    _status: Optional[str]
     _termination_statuses: List[str]
     _polling_interval: int
-    _resource: JSON
-    _command: Any
-    _initial_response: JSON
+    _resource: Optional[JSON]
+    _command: Optional[Any]
+    _initial_response: Optional[JSON]
 
     def _update_status(self) -> None:
         raise NotImplementedError("This method needs to be implemented")
 
     def _update_resource(self) -> None:
+        assert self._command is not None
         self._resource = self._command()
 
     def initialize(self, client, initial_response, deserialization_callback) -> None:
@@ -46,12 +47,15 @@ def initialize(self, client, initial_response, deserialization_callback) -> None
         self._resource = initial_response
 
     def status(self) -> str:
+        assert self._status is not None
         return self._status
 
     def finished(self) -> bool:
+        assert self._status is not None
         return self._status in self._termination_statuses
 
     def resource(self) -> JSON:
+        assert self._resource is not None
         return self._resource
 
     def run(self) -> None:
@@ -84,6 +88,7 @@ def __init__(self, interval=5) -> None:
         ]
 
     def _update_status(self) -> None:
+        assert self._resource is not None
         self._status = self._resource["validationStatus"]
 
 
@@ -99,6 +104,7 @@ def __init__(self, interval=5) -> None:
         self._termination_statuses = ["DONE", "FAILED", "CANCELLED"]
 
     def _update_status(self) -> None:
+        assert self._resource is not None
         self._status = self._resource["status"]
 
 
@@ -114,6 +120,7 @@ def __init__(self, interval=5) -> None:
         self._termination_statuses = ["DONE", "FAILED", "CANCELLED"]
 
     def _update_status(self):
+        assert self._resource is not None
         self._status = self._resource["status"]
 
 

sdk/loadtesting/azure-developer-loadtesting/azure/developer/loadtesting/aio/_operations/_patch.py

@@ -28,17 +28,18 @@
 class AsyncLoadTestingPollingMethod(AsyncPollingMethod):
     """Base class for custom async polling methods."""
 
-    _status: str
+    _status: Optional[str]
     _termination_statuses: List[str]
     _polling_interval: int
-    _resource: JSON
-    _command: Any
-    _initial_response: JSON
+    _resource: Optional[JSON]
+    _command: Optional[Any]
+    _initial_response: Optional[JSON]
 
     def _update_status(self) -> None:
         raise NotImplementedError("This method needs to be implemented")
 
     async def _update_resource(self) -> None:
+        assert self._command is not None
         self._resource = await self._command()
 
     def initialize(self, client, initial_response, deserialization_callback) -> None:
@@ -47,12 +48,15 @@ def initialize(self, client, initial_response, deserialization_callback) -> None
         self._resource = initial_response
 
     def status(self) -> str:
+        assert self._status is not None
         return self._status
 
     def finished(self) -> bool:
+        assert self._status is not None
         return self._status in self._termination_statuses
 
     def resource(self) -> JSON:
+        assert self._resource is not None
         return self._resource
 
     async def run(self) -> None:
@@ -83,6 +87,7 @@ def __init__(self, interval=5) -> None:
         ]
 
     def _update_status(self) -> None:
+        assert self._resource is not None
         self._status = self._resource["validationStatus"]
 
 
@@ -96,6 +101,7 @@ def __init__(self, interval=5) -> None:
         self._termination_statuses = ["DONE", "FAILED", "CANCELLED"]
 
     def _update_status(self) -> None:
+        assert self._resource is not None
         self._status = self._resource["status"]
 
 
@@ -109,6 +115,7 @@ def __init__(self, interval=5) -> None:
         self._termination_statuses = ["DONE", "FAILED", "CANCELLED"]
 
     def _update_status(self) -> None:
+        assert self._resource is not None
         self._status = self._resource["status"]