diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 81a3571..2c0456b 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -16,7 +16,7 @@ repos: - id: black args: ["--check"] - repo: https://github.com/astral-sh/ruff-pre-commit - rev: v0.0.270 + rev: v0.0.277 hooks: - id: ruff args: ["--target-version", "py39"] diff --git a/tests_integration/tests/conftest.py b/tests_integration/tests/conftest.py index 4a56d87..640363d 100644 --- a/tests_integration/tests/conftest.py +++ b/tests_integration/tests/conftest.py @@ -3,7 +3,7 @@ import math import os import time -from enum import Enum +from enum import StrEnum from typing import Any import httpx @@ -14,7 +14,7 @@ TIMEOUT = 10 URL = "http://spoolman:8000" -class DbType(str, Enum): +class DbType(StrEnum): """Enum for database types.""" SQLITE = "sqlite" @@ -43,7 +43,7 @@ def _wait_for_server(): # noqa: ANN202 try: response = httpx.get("http://spoolman:8000", timeout=1) response.raise_for_status() - except httpx.HTTPError: + except httpx.HTTPError: # noqa: PERF203 if time.time() - start_time > TIMEOUT: raise else: diff --git a/tests_integration/tests/test_spool.py b/tests_integration/tests/test_spool.py index 024cdf9..1ecab0c 100644 --- a/tests_integration/tests/test_spool.py +++ b/tests_integration/tests/test_spool.py @@ -225,7 +225,7 @@ def test_get_spool_not_found(): assert "123456789" in message -def test_find_spools(random_filament: dict[str, Any]): # noqa: PLR0915 +def test_find_spools(random_filament: dict[str, Any]): """Test finding spools in the database.""" # Setup result = httpx.post( @@ -628,18 +628,18 @@ async def test_use_spool_concurrent(random_filament: dict[str, Any]): requests = 100 used_weight = 0.5 async with httpx.AsyncClient() as client: - tasks = [] - for _ in range(requests): - tasks.append( + await asyncio.gather( + *( client.put( f"{URL}/api/v1/spool/{spool['id']}/use", json={ "use_weight": used_weight, }, timeout=60, - ), - ) - await asyncio.gather(*tasks) + ) + for _ in range(requests) + ), + ) # Verify result = httpx.get(f"{URL}/api/v1/spool/{spool['id']}") # noqa: ASYNC100