Function handler_fixture

python/e2e/test_copilot_request_handler_e2e.py:231–255

(ctx: E2ETestContext)

@pytest_asyncio.fixture(loop_scope="module")
async def handler_fixture(ctx: E2ETestContext):
    upstream = await _start_fake_upstream()
    counters = _Counters()
    handler = _TestHandler(upstream, counters)
    github_token = (
        "fake-token-for-e2e-tests" if os.environ.get("GITHUB_ACTIONS") == "true" else None
    )
    env = {**ctx.get_env(), "COPILOT_EXP_COPILOT_CLI_WEBSOCKET_RESPONSES": "true"}
    client = CopilotClient(
        connection=RuntimeConnection.for_stdio(path=ctx.cli_path),
        working_directory=ctx.work_dir,
        env=env,
        github_token=github_token,
        request_handler=handler,
    )
    try:
        yield _HandlerFixture(client=client, upstream=upstream, counters=counters)
    finally:
        try:
            await client.stop()
        except Exception:
            # Best-effort teardown during fixture cleanup.
            pass
        await handler.aclose()
        await upstream.close()


class TestCopilotRequestHandler: