FusionAGI/tests/test_security_middleware.py

"""Tests for CSRF and CSP security middleware."""

from fusionagi.api.security import get_csp_middleware, get_csrf_middleware


def test_csrf_middleware_class():
    """CSRF middleware should be a valid class."""
    cls = get_csrf_middleware()
    assert cls is not None
    assert cls.__name__ == "CSRFMiddleware"


def test_csp_middleware_class():
    """CSP middleware should be a valid class."""
    cls = get_csp_middleware()
    assert cls is not None
    assert cls.__name__ == "CSPMiddleware"