Function pre_set_event_context

tests/units/test_app.py:3457–3478

Optionally pre-set an EventContext in the isolated context. Args: isolated_context: The empty context to optionally populate. request: The pytest fixture request with the param value. Returns: The pre-set EventContext, or None if unset.

(
    isolated_context: contextvars.Context,
    request: pytest.FixtureRequest,
)

    params=[False, True], ids=["unset_event_context", "preset_event_context"]
)
def pre_set_event_context(
    isolated_context: contextvars.Context,
    request: pytest.FixtureRequest,
) -> EventContext | None:
    """Optionally pre-set an EventContext in the isolated context.

    Args:
        isolated_context: The empty context to optionally populate.
        request: The pytest fixture request with the param value.

    Returns:
        The pre-set EventContext, or None if unset.
    """
    if request.param:
        ctx = EventContext(
            token="pre-existing",
            state_manager=StateManagerMemory(),
            enqueue_impl=AsyncMock(),
        )
        isolated_context.run(EventContext.set, ctx)
        return ctx
    return None


def test_set_contexts(