style: apply ruff formatting and move TYPE_CHECKING imports in tests
This commit is contained in:
@@ -8,7 +8,7 @@ from __future__ import annotations
|
||||
|
||||
import asyncio
|
||||
from contextlib import asynccontextmanager
|
||||
from typing import AsyncGenerator
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import pytest
|
||||
from fastapi import FastAPI, Request
|
||||
@@ -18,14 +18,17 @@ from httpx import ASGITransport, AsyncClient
|
||||
from fastapi_traffic import (
|
||||
Algorithm,
|
||||
MemoryBackend,
|
||||
RateLimitExceeded,
|
||||
RateLimiter,
|
||||
RateLimitExceeded,
|
||||
rate_limit,
|
||||
)
|
||||
from fastapi_traffic.core.config import RateLimitConfig
|
||||
from fastapi_traffic.core.limiter import set_limiter
|
||||
from fastapi_traffic.middleware import RateLimitMiddleware
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from collections.abc import AsyncGenerator
|
||||
|
||||
|
||||
class TestFullApplicationFlow:
|
||||
"""Integration tests for a complete application setup."""
|
||||
@@ -128,9 +131,7 @@ class TestFullApplicationFlow:
|
||||
)
|
||||
assert response.status_code == 200
|
||||
|
||||
async def test_basic_rate_limiting_works(
|
||||
self, client: AsyncClient
|
||||
) -> None:
|
||||
async def test_basic_rate_limiting_works(self, client: AsyncClient) -> None:
|
||||
"""Test that basic rate limiting is functional."""
|
||||
# Make a request and verify it works
|
||||
response = await client.get("/api/v1/users/1")
|
||||
|
||||
Reference in New Issue
Block a user