add type annotation in test

Author: atstaeff

eventsourcingdb/errors/client_error.py

@@ -2,5 +2,5 @@
 
 
 class ClientError(CustomError):
-    def __init__(self, cause: str):
+    def __init__(self, cause: str) -> None:
         super().__init__(f'Client error occurred: {cause}')

eventsourcingdb/errors/internal_error.py

@@ -2,5 +2,5 @@
 
 
 class InternalError(CustomError):
-    def __init__(self, cause: str):
+    def __init__(self, cause: str) -> None:
         super().__init__(f'Internal error occurred: {cause}')

eventsourcingdb/errors/server_error.py

@@ -2,5 +2,5 @@
 
 
 class ServerError(CustomError):
-    def __init__(self, cause: str):
+    def __init__(self, cause: str) -> None:
         super().__init__(f'Server error occurred: {cause}')

eventsourcingdb/errors/validation_error.py

@@ -2,5 +2,5 @@
 
 
 class ValidationError(CustomError):
-    def __init__(self, cause: str):
+    def __init__(self, cause: str) -> None:
         super().__init__(f'Validation error occurred: {cause}')

eventsourcingdb/event/event.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 from dataclasses import dataclass
-from typing import TypeVar
+from typing import TypeVar, Any
 
 from ..errors.internal_error import InternalError
 from ..errors.validation_error import ValidationError
@@ -24,7 +24,7 @@ class Event():
     trace_state: str | None = None
 
     @staticmethod
-    def parse(unknown_object: dict) -> "Event":
+    def parse(unknown_object: dict) -> 'Event':
         source = unknown_object.get('source')
         if not isinstance(source, str):
             raise ValidationError(
@@ -103,7 +103,7 @@ def parse(unknown_object: dict) -> "Event":
             trace_state=trace_state
         )
 
-    def to_json(self):
+    def to_json(self) -> dict[str, Any]:
         json = {
             'specversion': self.spec_version,
             'id': self.event_id,

eventsourcingdb/event/event_candidate.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-
+from typing import Any
 
 @dataclass
 class EventCandidate:
@@ -10,7 +10,7 @@ class EventCandidate:
     trace_parent: str | None = None
     trace_state: str | None = None
 
-    def to_json(self):
+    def to_json(self) -> dict[str, Any]:
         json = {
             'data': self.data,
             'source': self.source,

eventsourcingdb/http_client/http_client.py

@@ -15,12 +15,12 @@ def __init__(
         self,
         base_url: str,
         api_token: str,
-    ):
+    ) -> None:
         self.__base_url = base_url
         self.__api_token = api_token
         self.__session: ClientSession | None = None
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> 'HttpClient':
         await self.initialize()
         return self
 
@@ -35,7 +35,7 @@ async def __aexit__(
     async def initialize(self) -> None:
         self.__session = aiohttp.ClientSession()
 
-    async def close(self):
+    async def close(self) -> None:
         if self.__session is not None:
             await self.__session.close()
             self.__session = None

eventsourcingdb/http_client/response.py

@@ -8,21 +8,20 @@
 
 
 class Response:
-    def __init__(self, response: aiohttp.ClientResponse):
+    def __init__(self, response: aiohttp.ClientResponse) -> None:
         self.__response: aiohttp.ClientResponse = response
 
-    async def __aenter__(self):
-        # Properly await any async initialization if needed
+    async def __aenter__(self) -> 'Response':
         return self
 
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
+    async def __aexit__(self, exc_type, exc_val, exc_tb) -> None:
         if not self.__response.closed:
             self.__response.close()
 
-    def __enter__(self):
+    def __enter__(self) -> 'Response':
         return self
 
-    def __exit__(self, exc_type, exc_val, exc_tb):
+    def __exit__(self, exc_type, exc_val, exc_tb) -> None:
         if not self.__response.closed:
             self.__response.close()
 

eventsourcingdb/observe_events/observe_from_latest_event.py

@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from typing import Any
 
 from .if_event_is_missing_during_observe import IfEventIsMissingDuringObserve
 
@@ -9,7 +10,7 @@ class ObserveFromLatestEvent:
     type: str
     if_event_is_missing: IfEventIsMissingDuringObserve
 
-    def to_json(self):
+    def to_json(self) -> dict[str, Any]:
         return {
             'subject': self.subject,
             'type': self.type,

eventsourcingdb/read_event_types/event_type.py

@@ -13,7 +13,7 @@ class EventType:
     schema: dict[str, Any] | None = None
 
     @staticmethod
-    def parse(unknown_object: dict) -> "EventType":
+    def parse(unknown_object: dict) -> 'EventType':
         event_type = unknown_object.get('eventType')
         if not isinstance(event_type, str):
             raise ValidationError(
@@ -38,7 +38,7 @@ def parse(unknown_object: dict) -> "EventType":
             schema=schema,
         )
 
-    def __hash__(self):
+    def __hash__(self) -> str:
         # Convert dictionary schema to a hashable form (tuple of items)
         if isinstance(self.schema, dict):
             # Sort items to ensure consistent hashing