fixturify 0.1.11__tar.gz → 0.1.12__tar.gz

{fixturify-0.1.11 → fixturify-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fixturify
-Version: 0.1.11
+Version: 0.1.12
 Summary: A collection of convenient testing utilities for Python
 Project-URL: Homepage, https://github.com/eleven-sea/pytools
 Project-URL: Repository, https://github.com/eleven-sea/pytools

{fixturify-0.1.11 → fixturify-0.1.12}/docker-compose.yml

@@ -18,11 +18,11 @@ services:
     environment:
       - POSTGRES_USER=postgres
       - POSTGRES_PASSWORD=postgres
-      - POSTGRES_DB=pytools
+      - POSTGRES_DB=fixturify
     volumes:
       - postgres_data:/var/lib/postgresql/data
     healthcheck:
-      test: ["CMD-SHELL", "pg_isready -U postgres -d pytools"]
+      test: ["CMD-SHELL", "pg_isready -U postgres -d fixturify"]
       interval: 5s
       timeout: 5s
       retries: 5
@@ -33,7 +33,7 @@ services:
       - "3306:3306"
     environment:
       - MYSQL_ALLOW_EMPTY_PASSWORD=yes
-      - MYSQL_DATABASE=pytools
+      - MYSQL_DATABASE=fixturify
     volumes:
       - mysql_data:/var/lib/mysql
     healthcheck:

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/__init__.py

@@ -1,6 +1,6 @@
 """PyTools - A collection of reusable Python utility modules."""
 
-__version__ = "0.1.11"
+__version__ = "0.1.12"
 
 from fixturify.sql_d import sql, Phase, SqlTestConfig
 from fixturify.read_d import read

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/http_d/_recorder.py

@@ -811,10 +811,36 @@ def create_response_from_http_client(
     Returns:
         HttpResponse model
     """
+    import gzip
+    import zlib
+
     headers = dict(response.getheaders()) if hasattr(response, "getheaders") else {}
     content_type = headers.get("content-type", headers.get("Content-Type", ""))
+    content_encoding = headers.get(
+        "content-encoding", headers.get("Content-Encoding", "")
+    ).lower()
 
     body_bytes = response.read()
+
+    # Decompress if needed (http.client doesn't auto-decompress)
+    if body_bytes and content_encoding in ("gzip", "deflate"):
+        try:
+            if content_encoding == "gzip":
+                body_bytes = gzip.decompress(body_bytes)
+            elif content_encoding == "deflate":
+                # deflate can be raw or zlib-wrapped
+                try:
+                    body_bytes = zlib.decompress(body_bytes)
+                except zlib.error:
+                    body_bytes = zlib.decompress(body_bytes, -zlib.MAX_WBITS)
+            # Remove content-encoding header since we decompressed
+            headers = {
+                k: v for k, v in headers.items()
+                if k.lower() != "content-encoding"
+            }
+        except (gzip.BadGzipFile, OSError, zlib.error):
+            pass  # Not actually compressed, use as-is
+
     body, body_encoding, is_binary = _serialize_body(body_bytes, content_type)
 
     if is_binary:
@@ -1312,18 +1338,43 @@ def create_response_from_httpcore(
     Returns:
         HttpResponse model
     """
+    import gzip
+    import zlib
+
     # Convert headers from list of tuples to dict
     headers_dict: Dict[str, str] = {}
+    content_encoding = ""
     if response.headers:
         for k, v in response.headers:
             key = k.decode() if isinstance(k, bytes) else k
             val = v.decode() if isinstance(v, bytes) else v
             headers_dict[key] = val
+            if key.lower() == "content-encoding":
+                content_encoding = val.lower()
 
     content_type = headers_dict.get("content-type", headers_dict.get("Content-Type", ""))
 
     # Get body content
     body_bytes = response.content
+
+    # Decompress if needed (httpcore doesn't auto-decompress)
+    if body_bytes and content_encoding in ("gzip", "deflate"):
+        try:
+            if content_encoding == "gzip":
+                body_bytes = gzip.decompress(body_bytes)
+            elif content_encoding == "deflate":
+                try:
+                    body_bytes = zlib.decompress(body_bytes)
+                except zlib.error:
+                    body_bytes = zlib.decompress(body_bytes, -zlib.MAX_WBITS)
+            # Remove content-encoding header since we decompressed
+            headers_dict = {
+                k: v for k, v in headers_dict.items()
+                if k.lower() != "content-encoding"
+            }
+        except (gzip.BadGzipFile, OSError, zlib.error):
+            pass  # Not actually compressed, use as-is
+
     body, body_encoding, is_binary = _serialize_body(body_bytes, content_type)
 
     if is_binary:

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/http_d/_stubs/_httpcore.py

@@ -9,7 +9,7 @@ import functools
 from typing import TYPE_CHECKING, Any
 
 from .._models import HttpRequest, HttpResponse
-from .._recorder import create_request_from_httpcore, create_response_from_httpcore
+from .._recorder import create_request_from_httpcore
 
 if TYPE_CHECKING:
     from .._mock_context import HttpMockContext
@@ -72,13 +72,14 @@ def _build_response(response_model: HttpResponse) -> Any:
     """Build httpcore Response from HttpResponse model."""
     from httpcore import Response
     import gzip
-    
+    import zlib
+
     # Get content encoding before filtering headers
     content_encoding = response_model.headers.get(
-        "content-encoding", 
+        "content-encoding",
         response_model.headers.get("Content-Encoding", "")
     ).lower()
-    
+
     # Build headers as list of tuples
     headers = [
         (k.encode("ascii"), v.encode("ascii"))
@@ -86,15 +87,21 @@ def _build_response(response_model: HttpResponse) -> Any:
         # Skip headers that cause issues
         if k.lower() not in ("transfer-encoding", "content-encoding")
     ]
-    
+
     body = response_model.get_body_bytes()
-    
-    # Decompress gzip body if needed (since we removed content-encoding header)
-    if content_encoding == "gzip" and body:
+
+    # Decompress body if needed (since we removed content-encoding header)
+    if body and content_encoding in ("gzip", "deflate"):
         try:
-            body = gzip.decompress(body)
-        except (gzip.BadGzipFile, OSError):
-            pass  # Not actually gzip, use as-is
+            if content_encoding == "gzip":
+                body = gzip.decompress(body)
+            elif content_encoding == "deflate":
+                try:
+                    body = zlib.decompress(body)
+                except zlib.error:
+                    body = zlib.decompress(body, -zlib.MAX_WBITS)
+        except (gzip.BadGzipFile, OSError, zlib.error):
+            pass  # Not actually compressed, use as-is
     
     return Response(
         status=response_model.status,
@@ -105,16 +112,17 @@ def _build_response(response_model: HttpResponse) -> Any:
 
 def _serialize_response(httpcore_response: Any) -> HttpResponse:
     """Create HttpResponse from httpcore Response object.
-    
-    Handles gzip decompression before serialization to ensure
+
+    Handles gzip/deflate decompression before serialization to ensure
     body is stored as readable text, not binary.
     """
     import gzip
+    import zlib
     from .._models import HttpResponse
-    
+
     # Get body - should be available after read()/aread()
     body_bytes = httpcore_response.content
-    
+
     # Convert headers
     headers_dict = {}
     content_encoding = ""
@@ -124,21 +132,24 @@ def _serialize_response(httpcore_response: Any) -> HttpResponse:
         headers_dict[key] = val
         if key.lower() == "content-encoding":
             content_encoding = val.lower()
-    
-    # If gzip, decompress body before serialization
-    if content_encoding == "gzip" and body_bytes:
+
+    # Decompress body before serialization if needed
+    if body_bytes and content_encoding in ("gzip", "deflate"):
         try:
-            body_bytes = gzip.decompress(body_bytes)
+            if content_encoding == "gzip":
+                body_bytes = gzip.decompress(body_bytes)
+            elif content_encoding == "deflate":
+                try:
+                    body_bytes = zlib.decompress(body_bytes)
+                except zlib.error:
+                    body_bytes = zlib.decompress(body_bytes, -zlib.MAX_WBITS)
             # Remove content-encoding header since we decompressed
             headers_dict = {
                 k: v for k, v in headers_dict.items()
                 if k.lower() != "content-encoding"
             }
-        except (gzip.BadGzipFile, OSError):
-            pass  # Not actually gzip, use as-is
-    
-    # Determine content type
-    content_type = headers_dict.get("content-type", headers_dict.get("Content-Type", ""))
+        except (gzip.BadGzipFile, OSError, zlib.error):
+            pass  # Not actually compressed, use as-is
     
     # Try to decode as text/JSON
     body_str = None

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_connection_cache.py

@@ -43,6 +43,9 @@ class ConnectionCache:
     Maintains separate caches for sync and async connections.
     Connections are created on first access and reused for subsequent calls.
     All connections are closed when the process exits via atexit handler.
+
+    Async connections are tracked with their event loop - if the loop changes
+    or closes, the cached connection is invalidated automatically.
     """
 
     _instance: Optional["ConnectionCache"] = None
@@ -58,7 +61,8 @@ class ConnectionCache:
             return
         self._initialized = True
         self._sync_connections: Dict[CacheKey, Any] = {}
-        self._async_connections: Dict[CacheKey, Any] = {}
+        # Store (connection, event_loop) tuple for async connections
+        self._async_connections: Dict[CacheKey, Tuple[Any, Any]] = {}
         self._sync_close_funcs: Dict[CacheKey, Any] = {}
         self._async_close_funcs: Dict[CacheKey, Any] = {}
 
@@ -90,25 +94,36 @@ class ConnectionCache:
     async def get_async(
         self,
         config: "SqlTestConfig",
-        connect_coro: Any,
+        connect_factory: Any,
     ) -> Any:
         """
         Get or create an async connection.
 
+        Tracks the event loop - if the cached connection was created in a
+        different or closed loop, it's invalidated and a new one is created.
+
         Args:
             config: Database configuration
-            connect_coro: Coroutine that creates connection
+            connect_factory: Callable that returns a coroutine when called
 
         Returns:
             Database connection object
         """
         key = CacheKey.from_config(config)
+        current_loop = asyncio.get_running_loop()
 
-        if key not in self._async_connections:
-            connection = await connect_coro
-            self._async_connections[key] = connection
+        if key in self._async_connections:
+            connection, cached_loop = self._async_connections[key]
+            # Check if loop is the same and still running
+            if cached_loop is current_loop and not cached_loop.is_closed():
+                return connection
+            # Loop changed or closed - remove stale connection
+            del self._async_connections[key]
 
-        return self._async_connections[key]
+        # Create new connection and track with current loop
+        connection = await connect_factory()
+        self._async_connections[key] = (connection, current_loop)
+        return connection
 
     def register_sync_closer(
         self,
@@ -150,7 +165,7 @@ class ConnectionCache:
         This is called by atexit handler which runs in sync context.
         We need to handle async cleanup carefully.
         """
-        for key, connection in list(self._async_connections.items()):
+        for key, (connection, _loop) in list(self._async_connections.items()):
             try:
                 if hasattr(connection, "close"):
                     # Some async connections have sync close()
@@ -174,7 +189,7 @@ class ConnectionCache:
 
     async def close_all_async(self) -> None:
         """Close all cached async connections from async context."""
-        for key, connection in list(self._async_connections.items()):
+        for key, (connection, _loop) in list(self._async_connections.items()):
             try:
                 if key in self._async_close_funcs:
                     await self._async_close_funcs[key](connection)

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_strategies/_aiomysql.py

@@ -28,7 +28,7 @@ class AiomysqlStrategy(AsyncSqlExecutionStrategy):
         params = self.build_connection_params(config)
 
         connection = await self.get_cached_connection_async(
-            config, driver.connect(**params)
+            config, lambda: driver.connect(**params)
         )
         async with connection.cursor() as cursor:
             # aiomysql doesn't support multi-statement by default

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_strategies/_aiosqlite.py

@@ -26,7 +26,7 @@ class AiosqliteStrategy(AsyncSqlExecutionStrategy):
         database_path = params.get("database", ":memory:")
 
         connection = await self.get_cached_connection_async(
-            config, driver.connect(database_path)
+            config, lambda: driver.connect(database_path)
         )
         await connection.executescript(sql_content)
         await connection.commit()

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_strategies/_asyncpg.py

@@ -9,7 +9,11 @@ if TYPE_CHECKING:
 
 
 class AsyncpgStrategy(AsyncSqlExecutionStrategy):
-    """Execution strategy for asyncpg (PostgreSQL async)."""
+    """Execution strategy for asyncpg (PostgreSQL async).
+
+    Uses connection pools instead of single connections to support
+    concurrent operations across multiple tests.
+    """
 
     driver_name = "asyncpg"
     is_async = True
@@ -23,11 +27,13 @@ class AsyncpgStrategy(AsyncSqlExecutionStrategy):
     }
 
     async def execute_async(self, sql_content: str, config: "SqlTestConfig") -> None:
-        """Execute SQL using asyncpg."""
+        """Execute SQL using asyncpg with connection pool."""
         driver = self.get_driver_module()
         params = self.build_connection_params(config)
 
-        connection = await self.get_cached_connection_async(
-            config, driver.connect(**params)
+        # Use pool instead of single connection for concurrency support
+        pool = await self.get_cached_connection_async(
+            config, lambda: driver.create_pool(**params, min_size=1, max_size=5)
         )
-        await connection.execute(sql_content)
+        async with pool.acquire() as connection:
+            await connection.execute(sql_content)

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_strategies/_base.py

@@ -142,14 +142,20 @@ class AsyncSqlExecutionStrategy(SqlExecutionStrategy):
     async def get_cached_connection_async(
         self,
         config: "SqlTestConfig",
-        connect_coro: Any,
+        connect_factory: Any,
     ) -> Any:
         """
         Get a cached async connection or create a new one.
 
+        Note: Async connections are tied to the event loop. For caching to work,
+        pytest must use session-scoped event loop:
+
+            [tool.pytest.ini_options]
+            asyncio_default_fixture_loop_scope = "session"
+
         Args:
             config: Database configuration
-            connect_coro: Coroutine that creates connection
+            connect_factory: Callable that returns a coroutine when called
 
         Returns:
             Database connection (cached or newly created)
@@ -157,4 +163,4 @@ class AsyncSqlExecutionStrategy(SqlExecutionStrategy):
         from .._connection_cache import get_connection_cache
 
         cache = get_connection_cache()
-        return await cache.get_async(config, connect_coro)
+        return await cache.get_async(config, connect_factory)

{fixturify-0.1.11 → fixturify-0.1.12}/fixturify/sql_d/_strategies/_psycopg.py

@@ -28,7 +28,7 @@ class PsycopgStrategy(AsyncSqlExecutionStrategy):
         params = self.build_connection_params(config)
 
         connection = await self.get_cached_connection_async(
-            config, driver.AsyncConnection.connect(**params)
+            config, lambda: driver.AsyncConnection.connect(**params)
         )
         await connection.execute(sql_content)
         await connection.commit()

{fixturify-0.1.11 → fixturify-0.1.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fixturify"
-version = "0.1.11"
+version = "0.1.12"
 description = "A collection of convenient testing utilities for Python"
 requires-python = ">=3.10"
 license = "MIT"
@@ -64,3 +64,5 @@ dev-dependencies = [
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+asyncio_default_test_loop_scope = "function"

{fixturify-0.1.11 → fixturify-0.1.12}/tests/test_json_assert/nested/deep/test_grandparent_path.py

@@ -130,11 +130,9 @@ class TestJsonAssertGrandparentPathMismatch:
 
     def test_nonexistent_file_creates_snapshot_grandparent(self, tmp_path):
         """Test that nonexistent file creates snapshot with ../../ path."""
-        import os
 
         # This test verifies the snapshot creation behavior
         # JsonAssert creates the file if it doesn't exist (snapshot testing)
-        data = {"name": "John"}
 
         # Use a temp file path that doesn't exist
         temp_file = tmp_path / "snapshot.json"

{fixturify-0.1.11 → fixturify-0.1.12}/tests/test_sql/test_connection_cache.py

@@ -1,7 +1,7 @@
 """Tests for the connection cache module."""
 
 import pytest
-from unittest.mock import Mock, MagicMock, AsyncMock
+from unittest.mock import Mock, AsyncMock
 
 from fixturify.sql_d._connection_cache import (
     CacheKey,
@@ -193,10 +193,10 @@ class TestConnectionCache:
 
         mock_conn = AsyncMock()
 
-        async def connect_coro():
+        async def connect_factory():
             return mock_conn
 
-        result = await cache.get_async(config, connect_coro())
+        result = await cache.get_async(config, connect_factory)
 
         assert result is mock_conn
 
@@ -215,19 +215,18 @@ class TestConnectionCache:
         mock_conn = AsyncMock()
         call_count = 0
 
-        async def connect_coro():
+        async def connect_factory():
             nonlocal call_count
             call_count += 1
             return mock_conn
 
-        # First call
-        result1 = await cache.get_async(config, connect_coro())
-        # Second call - should not create new coroutine
-        result2 = await cache.get_async(config, connect_coro())
+        # First call - creates connection
+        result1 = await cache.get_async(config, connect_factory)
+        # Second call - reuses connection
+        result2 = await cache.get_async(config, connect_factory)
 
         assert result1 is result2
-        # connect_coro was called twice but connection created only once
-        # (second coro is not awaited because cache hit)
+        # Factory should only be called once (cache hit on second call)
         assert call_count == 1
 
     def test_close_all_sync(self) -> None:

{fixturify-0.1.11 → fixturify-0.1.12}/uv.lock

@@ -554,7 +554,7 @@ wheels = [
 
 [[package]]
 name = "fixturify"
-version = "0.1.11"
+version = "0.1.12"
 source = { editable = "." }
 dependencies = [
     { name = "deepdiff" },