From 698482546d192490f3cfef5f84a8871d8a022455 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci-lite[bot]" <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Date: Mon, 20 Apr 2026 13:33:25 +0000 Subject: [PATCH] =?UTF-8?q?=F0=9F=8E=A8=20Auto=20format?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- fastapi/middleware/__init__.py | 3 +-- tests/test_slow_request_middleware.py | 11 ++++++++--- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/fastapi/middleware/__init__.py b/fastapi/middleware/__init__.py index 1ea0da8850..1ae80d6830 100644 --- a/fastapi/middleware/__init__.py +++ b/fastapi/middleware/__init__.py @@ -1,3 +1,2 @@ -from starlette.middleware import Middleware as Middleware - from fastapi.middleware.slow_request import SlowRequestMiddleware +from starlette.middleware import Middleware as Middleware diff --git a/tests/test_slow_request_middleware.py b/tests/test_slow_request_middleware.py index 14476e2b05..f930460ce1 100644 --- a/tests/test_slow_request_middleware.py +++ b/tests/test_slow_request_middleware.py @@ -1,6 +1,5 @@ import asyncio import logging -import time from fastapi import FastAPI from fastapi.testclient import TestClient @@ -33,7 +32,9 @@ def test_slow_request_logging(caplog): assert response.status_code == 200 # 检查日志是否包含预期的警告信息 - slow_request_logs = [record for record in caplog.records if "Slow request" in record.message] + slow_request_logs = [ + record for record in caplog.records if "Slow request" in record.message + ] assert len(slow_request_logs) == 1 log_message = slow_request_logs[0].message @@ -41,8 +42,11 @@ def test_slow_request_logging(caplog): assert "/slow" in log_message # 检查是否包含两位小数的时间 import re + match = re.search(r"took (\d+\.\d{2})s", log_message) - assert match is not None, f"Log message should contain time with two decimal places: {log_message}" + assert match is not None, ( + f"Log message should contain time with two decimal places: {log_message}" + ) print("✅ 慢请求监控中间件测试通过!") print(f" 日志消息: {log_message}") @@ -50,4 +54,5 @@ def test_slow_request_logging(caplog): if __name__ == "__main__": import pytest + pytest.main([__file__, "-v", "-s"])