vercel-queue-bundle 0.7.1__py3-none-any.whl

vercel/queue/devserver.py

@@ -0,0 +1,24 @@
+"""Standalone queue development server."""
+
+from __future__ import annotations
+
+from ._internal.devserver import (
+    EmbeddedQueueDevServer,
+    QueueClientAsgiDevServer,
+    embedded_queue_dev_server,
+    main,
+    queue_client_asgi_dev_server,
+)
+
+# Only add public symbols to __all__; internal helpers must stay unexported.
+__all__ = (
+    "EmbeddedQueueDevServer",
+    "QueueClientAsgiDevServer",
+    "embedded_queue_dev_server",
+    "main",
+    "queue_client_asgi_dev_server",
+)
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

vercel/queue/embedded.py

@@ -0,0 +1,50 @@
+"""Embedded local queue server for in-process Queue API development.
+
+This module provides a dependency-light Queue API implementation for embedded
+local runtime code. It preserves the same client and push-delivery path as the
+hosted service: callers send through ``QueueClient``, messages are stored
+in a local server, visible messages are leased, and a dispatcher invokes
+subscribers through ``accept_and_handle`` with synthetic Vercel Queue push
+headers. Successful handlers are acknowledged by the normal client follow-up
+path, and failures retry through lease expiry or explicit visibility changes.
+
+Use this for embedded, single-process local mode and tests.
+
+Run an embedded queue server in a test and use a client that sends to
+it directly:
+
+    from vercel.internal._vendor import anyio
+    import vercel.queue as vq
+    import vercel.queue.embedded as vq_embedded
+
+    @vq.subscribe(topic="jobs*", consumer_group="just-testing")
+    async def handle_job(message: vq.Message[dict[str, object]]) -> None:
+        print(message.payload)
+
+    async def main() -> None:
+        async with vq_embedded.embedded_queue_service() as service:
+            client = service.get_async_client()
+            await client.send("email", {"subject": "hi"})
+
+    anyio.run(main)
+
+
+Enable the pytest fixture and use it in tests:
+
+    pytest_plugins = ["vercel.queue.testing.pytest"]
+
+    async def test_queue(embedded_queue_server):
+        client = embedded_queue_server.get_async_client()
+        message_id = await client.send("emails", {"subject": "hi"}, retention=60)
+"""
+
+from ._internal.embedded import (
+    create_embedded_queue_app,
+    embedded_queue_service,
+)
+
+# Only add public symbols to __all__; internal helpers must stay unexported.
+__all__ = (
+    "create_embedded_queue_app",
+    "embedded_queue_service",
+)

vercel/queue/py.typed

@@ -0,0 +1 @@
+

vercel/queue/sync.py

@@ -0,0 +1,8 @@
+"""Synchronous Vercel Queue client APIs."""
+# ruff: noqa: F403
+
+from ._internal import api_common as _api_common, api_sync as _api_sync
+from ._internal.api_common import *
+from ._internal.api_sync import *
+
+__all__ = _api_common.__all__ + _api_sync.__all__

vercel/queue/testing/__init__.py

@@ -0,0 +1,14 @@
+"""Testing helpers for Vercel Queue."""
+
+from .state import (
+    clear_subscriptions,
+    reset_default_async_queue_clients,
+    reset_default_queue_clients,
+)
+
+# Only add public symbols to __all__; internal helpers must stay unexported.
+__all__ = (
+    "clear_subscriptions",
+    "reset_default_async_queue_clients",
+    "reset_default_queue_clients",
+)

vercel/queue/testing/pytest.py

@@ -0,0 +1,42 @@
+"""Pytest fixtures for the embedded queue server."""
+
+from __future__ import annotations
+
+from collections.abc import Iterator
+
+import pytest
+
+from vercel.queue.devserver import EmbeddedQueueDevServer, embedded_queue_dev_server
+
+
+@pytest.fixture(scope="session")
+def _embedded_queue_server_session() -> Iterator[EmbeddedQueueDevServer]:
+    """Run one embedded queue server process for the test session."""
+    with embedded_queue_dev_server() as server:
+        yield server
+
+
+@pytest.fixture
+def embedded_queue_server(
+    _embedded_queue_server_session: EmbeddedQueueDevServer,
+) -> Iterator[EmbeddedQueueDevServer]:
+    """Return an isolated embedded queue server state for a test."""
+    _embedded_queue_server_session.reset()
+    try:
+        yield _embedded_queue_server_session
+    finally:
+        _embedded_queue_server_session.reset()
+
+
+@pytest.fixture
+def eqs(embedded_queue_server: EmbeddedQueueDevServer) -> EmbeddedQueueDevServer:
+    """Alias for the embedded queue server fixture."""
+    return embedded_queue_server
+
+
+# Only add public symbols to __all__; internal helpers must stay unexported.
+__all__ = (
+    "EmbeddedQueueDevServer",
+    "embedded_queue_server",
+    "eqs",
+)

vercel/queue/testing/state.py

@@ -0,0 +1,32 @@
+"""Queue test state helpers."""
+
+from __future__ import annotations
+
+from .._internal import subscribers as queue_subscribers
+from .._internal.http import (
+    reset_async_http_client_pool_for_tests,
+    reset_http_client_pools_for_tests,
+)
+
+
+def reset_default_queue_clients() -> None:
+    """Reset cached default Queue clients for tests."""
+    reset_http_client_pools_for_tests()
+
+
+async def reset_default_async_queue_clients() -> None:
+    """Reset cached default asynchronous Queue clients for tests."""
+    await reset_async_http_client_pool_for_tests()
+
+
+def clear_subscriptions() -> None:
+    """Clear registered subscribers and embedded dispatchers for tests."""
+    queue_subscribers.clear_subscriptions_for_tests()
+
+
+# Only add public symbols to __all__; internal helpers must stay unexported.
+__all__ = (
+    "clear_subscriptions",
+    "reset_default_async_queue_clients",
+    "reset_default_queue_clients",
+)

vercel/queue/version.py

@@ -0,0 +1,3 @@
+"""Package version metadata."""
+
+__version__ = "0.7.1"