Files
routstr-core/tests/unit
redshift 60921cef20 fix: allow /v1/messages/count_tokens through the proxy allowlist
The exact-match endpoint allowlist in the proxy omitted
`messages/count_tokens`, so every Claude Code / Anthropic SDK request was
404'd with `Path '/v1/messages/count_tokens' not found` before it ever
reached the (fully supported) forwarding path.

Add the endpoint to `_ALLOWED_ENDPOINTS` and a regression test that
pins it as always reachable on POST.

Regression history:
- 933ba105 "add missing messages endpoint" (2026-04-01): added
  messages/count_tokens handling to the forwarding layer.
- 164ed775 "support /message/count_tokens endpoint" (2026-05-09): added
  the local count_tokens handler.
- 96661384 "update not found proxy" (2026-05-14): only GET was gated, so
  POST count_tokens passed implicitly.
- 0217002e "Gate proxy forwarding behind a segment-anchored API path
  allowlist" (2026-08-23): POST gated, but the "v1/" prefix still carried
  count_tokens.
- 5af04364 "Restrict proxy forwarding to an exact method/path allowlist"
  (2026-08-24): switched to an exact table, added "messages" but omitted
  "messages/count_tokens". This is where the endpoint got locked out.
2026-10-01 23:30:53 +08:00
..
2025-08-03 20:35:59 -03:00
2025-08-09 14:55:26 -03:00
2026-09-04 01:35:43 +02:00
2026-09-28 23:59:36 +02:00
2026-06-13 23:40:39 +02:00
2026-09-07 00:30:05 +02:00
2026-08-03 23:32:06 +02:00
2026-08-06 22:58:44 +02:00
2026-09-24 18:27:27 +02:00
2026-08-03 00:05:36 +02:00
2026-09-24 18:27:27 +02:00
2026-09-24 18:27:27 +02:00
2026-09-07 00:30:05 +02:00
2026-07-01 17:08:28 +02:00
2026-07-01 16:53:52 +02:00
2026-05-14 15:40:49 +02:00
2026-09-16 13:15:32 +02:00
2026-07-29 22:50:33 +02:00
2026-09-26 02:41:19 +02:00
2026-09-04 01:35:43 +02:00
2026-09-24 18:27:27 +02:00

FastAPI Async Unit Tests

This directory contains async unit tests for the Routstr proxy FastAPI application.

Installation

First, ensure you have the development dependencies installed:

uv pip install -e ".[dev]"

Running Tests

To run all tests:

pytest

To run tests with coverage:

pytest --cov=routstr --cov-report=html

To run specific test files:

pytest tests/test_main.py
pytest tests/test_models.py
pytest tests/test_proxy.py

To run only async tests:

pytest -m asyncio

Test Structure

  • conftest.py - Pytest fixtures and configuration
  • test_main.py - Tests for main app endpoints
  • test_account.py - Tests for wallet/account management endpoints
  • test_proxy.py - Tests for the proxy functionality with mocked upstream
  • test_models.py - Tests for model pricing and data structures

Key Fixtures

  • async_client - Async HTTP client for testing FastAPI endpoints
  • test_session - In-memory SQLite database session for tests
  • test_api_key - Pre-configured API key with balance
  • api_key_with_balance - API key with sufficient balance for proxy tests

Environment Variables

The tests automatically set up required environment variables in conftest.py. No manual configuration needed.

Writing New Tests

  1. Use @pytest.mark.asyncio for async tests
  2. Use the provided fixtures for database and client access
  3. Mock external dependencies (like upstream API calls)
  4. Test both success and error cases
  5. Verify database state changes when applicable