Use abstract base classes in tests

Author: tisnik

tests/integration/conftest.py

@@ -2,7 +2,7 @@
 
 import os
 from pathlib import Path
-from typing import Generator
+from collections.abc import Generator
 
 import pytest
 from fastapi import Request, Response

tests/integration/endpoints/test_health_integration.py

@@ -1,6 +1,7 @@
 """Integration tests for the /health endpoint."""
 
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from fastapi import Response

tests/integration/endpoints/test_info_integration.py

@@ -1,6 +1,7 @@
 """Integration tests for the /info endpoint."""
 
-from typing import Generator, Any
+from typing import Any
+from collections.abc import Generator
 import pytest
 from pytest_mock import MockerFixture, AsyncMockType
 

tests/integration/endpoints/test_model_list.py

@@ -1,6 +1,7 @@
 """Integration tests for the /models endpoint (using Responses API)."""
 
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from fastapi import Request

tests/integration/endpoints/test_query_integration.py

@@ -4,7 +4,8 @@
 # pylint: disable=too-many-arguments  # Integration tests need many fixtures
 # pylint: disable=too-many-positional-arguments  # Integration tests need many fixtures
 
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from fastapi import HTTPException, Request, status

tests/integration/endpoints/test_root_endpoint.py

@@ -1,6 +1,7 @@
 """Integration tests for the /root endpoint."""
 
-from typing import Generator, Any
+from typing import Any
+from collections.abc import Generator
 import pytest
 from pytest_mock import MockerFixture
 

tests/integration/endpoints/test_streaming_query_integration.py

@@ -1,7 +1,8 @@
 """Integration tests for the /streaming_query endpoint (using Responses API)."""
 
 from collections.abc import AsyncIterator
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from fastapi import HTTPException, Request, status

tests/integration/endpoints/test_tools_integration.py

@@ -1,6 +1,7 @@
 """Integration tests for the /tools endpoint."""
 
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from fastapi import HTTPException, Request, status

tests/integration/test_rh_identity_integration.py

@@ -5,7 +5,7 @@
 import base64
 import json
 import os
-from typing import Generator
+from collections.abc import Generator
 
 import pytest
 from fastapi.testclient import TestClient

tests/unit/a2a_storage/test_storage_factory.py

@@ -3,7 +3,8 @@
 # pylint: disable=protected-access
 
 from pathlib import Path
-from typing import Any, Generator
+from typing import Any
+from collections.abc import Generator
 
 import pytest
 from pytest_mock import MockerFixture