add feat verify_api_token

Author: atstaeff

eventsourcingdb/client.py

@@ -83,9 +83,28 @@ async def ping(self) -> None:
         raise ServerError(f"Received unexpected response: {response_body}")
 
     async def verify_api_token(self) -> None:
-        # POST Request. Ist wie ping, aber mit POST. Authorization header token
-        # mitschicken. Token gültig dann event, Token ungültig 401
-        raise NotImplementedError("verify_api_token is not implemented yet.")
+        request_body = json.dumps({})
+        
+        response: Response = await self.http_client.post(
+            path='/api/v1/verify-api-token',
+            request_body=request_body,
+        )
+        async with response:
+            if response.status_code != HTTPStatus.OK:
+                raise ServerError(
+                    f'Failed to verify API token: {response}'
+                )
+
+            response_data = await response.body.read()
+            response_data = bytes.decode(response_data, encoding='utf-8')
+            response_json = json.loads(response_data)
+
+            if not isinstance(response_json, dict) or 'type' not in response_json:
+                raise ServerError('Failed to parse response: {response}')
+
+            expected_event_type = 'io.eventsourcingdb.api.api-token-verified'
+            if response_json.get('type') != expected_event_type:
+                raise ServerError(f'Failed to verify API token: {response}')
 
     async def write_events(
         self,
@@ -111,7 +130,7 @@ async def write_events(
         if response.status_code != HTTPStatus.OK:
             raise ServerError(
                 f'Unexpected response status:  '
-        )
+            )
 
         response_data = await response.body.read()
         response_data = bytes.decode(response_data, encoding='utf-8')
@@ -144,7 +163,7 @@ async def read_events(
             if response.status_code != HTTPStatus.OK:
                 raise ServerError(
                     f'Unexpected response status: {response}'
-            )
+                )
             async for raw_message in response.body:
                 message = parse_raw_message(raw_message)
 
@@ -209,7 +228,7 @@ async def run_eventql_query(self, query: str) -> AsyncGenerator[Any, None]:
 
                 if message.get('type') == 'row':
                     payload = message['payload']
-                
+
                     yield payload
                     continue
 
@@ -228,7 +247,7 @@ async def observe_events(
             'options': options.to_json()
         })
 
-        response : Response = await self.http_client.post(
+        response: Response = await self.http_client.post(
             path='/api/v1/observe-events',
             request_body=request_body,
         )
@@ -296,7 +315,7 @@ async def read_subjects(
             'baseSubject': base_subject
         })
 
-        response : Response = await self.http_client.post(
+        response: Response = await self.http_client.post(
             path='/api/v1/read-subjects',
             request_body=request_body,
         )

eventsourcingdb/http_client/http_client.py

@@ -43,7 +43,7 @@ async def close(self):
 
     async def post(self, path: str, request_body: str) -> Response:
         if self.__session is None:
-            raise CustomError()
+            await self.initialize()
 
         url_path = url.join_segments(self.__base_url, path)
         headers = get_post_headers(self.__api_token)
@@ -58,14 +58,13 @@ async def post(self, path: str, request_body: str) -> Response:
 
         return response
 
-
     async def get(
         self,
         path: str,
         with_authorization: bool = True,
     ) -> Response:
         if self.__session is None:
-            raise CustomError()
+            raise CustomError("HTTP client session not initialized. Call initialize() before making requests.")
 
         async def __request_executor() -> Response:
             url_path = url.join_segments(self.__base_url, path)

tests/shared/database.py

@@ -18,7 +18,7 @@ def __init__(
         self,
         create_key,
         with_authorization_client: Client,
-        with_invalid_url_client: Client ,
+        with_invalid_url_client: Client,
     ):
         assert create_key == Database.__create_key, \
             'Database objects must be created using Database.create.'
@@ -114,6 +114,14 @@ def get_client(self, client_type: str = CLIENT_TYPE_WITH_AUTH) -> Client:
 
         raise ValueError(f"Unknown client type: {client_type}")
 
+    def get_base_url(self) -> str:
+        """Get the base URL of the EventSourceDB container."""
+        return self.__container.get_base_url()
+
+    def get_api_token(self) -> str:
+        """Get the API token for the EventSourceDB container."""
+        return self.__container.get_api_token()
+
     async def stop(self) -> None:
         # Use walrus operator for concise access and check
         if (container := getattr(self.__class__, '_Database__container', None)):

tests/test_observe_events.py

@@ -406,12 +406,12 @@ async def test_cancelling_observe_events_async(
         async def process_events():
             nonlocal events_processed
             async for _ in client.observe_events(
-                    '/',
-                    ObserveEventsOptions(
-                        recursive=True,
-                        from_latest_event=None
-                    )
-                ):
+                '/',
+                ObserveEventsOptions(
+                    recursive=True,
+                    from_latest_event=None
+                )
+            ):
                 events_processed += 1
                 await asyncio.sleep(0.25)
 

tests/test_register_event_schema.py

@@ -86,7 +86,7 @@ async def test_throws_error_if_schema_is_invalid(
     ):
         client = database.get_client()
 
-        with pytest.raises(ServerError,  match='value must be "object"'):
+        with pytest.raises(ServerError, match='value must be "object"'):
             await client.register_event_schema(
                 "com.gornisht.ekht",
                 {

tests/test_run_eventql_query.py

@@ -29,11 +29,11 @@ async def test_reads_no_rows_if_query_does_not_return_any_rows(
         database: Database
     ):
         client = database.get_client()
-        
+
         did_read_rows = False
         async for _ in client.run_eventql_query('FROM e IN events PROJECT INTO e'):
             did_read_rows = True
-        
+
         assert did_read_rows is False
 
     @staticmethod
@@ -42,7 +42,7 @@ async def test_reads_all_rows_the_query_returns(
         database: Database
     ):
         client = database.get_client()
-        
+
         first_event = EventCandidate(
             source='https://www.eventsourcingdb.io',
             subject='/test',
@@ -51,7 +51,7 @@ async def test_reads_all_rows_the_query_returns(
                 'value': 23,
             },
         )
-        
+
         second_event = EventCandidate(
             source='https://www.eventsourcingdb.io',
             subject='/test',
@@ -60,21 +60,20 @@ async def test_reads_all_rows_the_query_returns(
                 'value': 42,
             },
         )
-        
+
         await client.write_events([first_event, second_event])
-        
+
         rows_read = []
         async for row in client.run_eventql_query('FROM e IN events PROJECT INTO e'):
             rows_read.append(row)
-        
+
         assert len(rows_read) == 2
-        
+
         first_row = rows_read[0]
         # Use dictionary access instead of attribute access
         assert first_row['id'] == '0'
         assert first_row['data']['value'] == 23
-        
+
         second_row = rows_read[1]
         assert second_row['id'] == '1'
         assert second_row['data']['value'] == 42
-

tests/test_verify_api_token.py

@@ -0,0 +1,33 @@
+import pytest
+
+from eventsourcingdb.client import Client
+from eventsourcingdb.errors.custom_error import CustomError
+from eventsourcingdb.errors.server_error import ServerError
+
+from .shared.database import Database
+
+
+class TestVerifyApiToken:
+    @staticmethod
+    @pytest.mark.asyncio
+    async def test_does_not_throw_if_token_is_valid(
+        database: Database
+    ):
+        client = database.get_client()
+        await client.verify_api_token()
+
+
+    @staticmethod
+    @pytest.mark.asyncio
+    async def test_throws_error_if_token_is_invalid(
+        database: Database
+    ):
+        base_url = database.get_base_url()
+
+        valid_token = database.get_api_token()
+        invalid_token = f"{valid_token}-invalid"
+
+        client = Client(base_url=base_url, api_token=invalid_token)
+
+        with pytest.raises(ServerError, match="Failed to verify API token"):
+            await client.verify_api_token()