MCP Agent Mail

from __future__ import annotations from fastapi import FastAPI from mcp_agent_mail import config as _config from mcp_agent_mail.http import SecurityAndRateLimitMiddleware, _decode_jwt_header_segment def test_decode_jwt_header_segment_variants(): # Well-formed header import base64 import json hdr = base64.urlsafe_b64encode(json.dumps({"alg": "HS256"}).encode("utf-8")).rstrip(b"=") token = hdr.decode("ascii") + ".x.y" decoded = _decode_jwt_header_segment(token) assert decoded and decoded.get("alg") == "HS256" # Malformed returns None assert _decode_jwt_header_segment("nope") is None def test_rate_limits_for_branches(monkeypatch): _config.clear_settings_cache() settings = _config.get_settings() app = FastAPI() mw = SecurityAndRateLimitMiddleware(app, settings) assert mw._rate_limits_for("tools")[0] >= 1 assert mw._rate_limits_for("resources")[0] >= 1 assert mw._rate_limits_for("other")[0] >= 1