From b5d0297375c2397898dc844598ff69c2905accd0 Mon Sep 17 00:00:00 2001
From: Brian Bjarke Jensen <bbj@skov.dk>
Date: Sun, 14 Sep 2025 20:11:26 +0200
Subject: [PATCH] pyupgrade fixed imports

---
 tests/integration/conftest.py           | 8 ++++----
 tests/integration/redis_adapter_test.py | 8 ++++----
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/tests/integration/conftest.py b/tests/integration/conftest.py
index fb0658f..3f6daf9 100644
--- a/tests/integration/conftest.py
+++ b/tests/integration/conftest.py
@@ -1,7 +1,7 @@
 """Integration tests configuration."""
 
 import os
-from typing import Generator
+from collections.abc import Generator
 import pytest
 import redis
 import structlog
@@ -27,7 +27,7 @@ def configure_logging() -> None:
 
 
 @pytest.fixture(scope="session")
-def redis_container() -> Generator[str, None, None]:
+def redis_container() -> Generator[str]:
     """Set up a Redis container for testing and yield the Redis URI."""
     # Start container
     container = RedisContainer(
@@ -52,7 +52,7 @@ def redis_container() -> Generator[str, None, None]:
 @pytest.fixture(scope="session", autouse=True)
 def set_environment_variables(
     redis_container: str,
-) -> Generator[dict[str, str], None, None]:
+) -> Generator[dict[str, str]]:
     """Set environment variables needed for tests."""
     # Build environment variables dictionary
     env_vars = {"REDIS_URI": redis_container}
@@ -68,7 +68,7 @@ def set_environment_variables(
 
 
 @pytest.fixture(scope="session")
-def raw_redis_client(redis_container: str) -> Generator[redis.Redis, None, None]:
+def raw_redis_client(redis_container: str) -> Generator[redis.Redis]:
     """Provide a raw Redis client connected to the test Redis container."""
     # Connect client
     client = redis.Redis.from_url(
diff --git a/tests/integration/redis_adapter_test.py b/tests/integration/redis_adapter_test.py
index 327478e..46ac8cc 100644
--- a/tests/integration/redis_adapter_test.py
+++ b/tests/integration/redis_adapter_test.py
@@ -3,7 +3,7 @@
 # allowing tests to access RedisAdapter's internal methods as needed for integration testing.
 """Integration tests for the RedisAdapter."""
 
-from typing import Generator
+from collections.abc import Generator
 import pytest
 import redis
 from redis.commands.json.path import Path as RedisPath
@@ -11,7 +11,7 @@ from python_repositories.adapters.redis_adapter import RedisAdapter
 
 
 @pytest.fixture(scope="session")
-def data() -> Generator[dict[str, str], None, None]:
+def data() -> Generator[dict[str, str]]:
     """Provide a sample data dictionary for tests."""
     yield {"foo": "bar"}
 
@@ -20,7 +20,7 @@ def data() -> Generator[dict[str, str], None, None]:
 def data_in_redis(
     raw_redis_client: redis.Redis,
     data: dict[str, str],
-) -> Generator[tuple[str, dict[str, str]], None, None]:
+) -> Generator[tuple[str, dict[str, str]]]:
     """Fixture to set up a known value in Redis before each test."""
     key = "test_key"
     path = RedisPath.root_path()
@@ -33,7 +33,7 @@ def data_in_redis(
 
 
 @pytest.fixture(scope="module")
-def redis_adapter(redis_container: str) -> Generator[RedisAdapter, None, None]:
+def redis_adapter(redis_container: str) -> Generator[RedisAdapter]:
     """Fixture to provide a connected RedisAdapter instance."""
     adapter = RedisAdapter()
     adapter.connect()