Skip to content

Commit 5eaf85f

Browse files
committed
add upper bounds check for zstd compression config
1 parent 398a86a commit 5eaf85f

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

tests/unit/test_middleware/test_compression_middleware.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
import sys
44
import zlib
55
from collections.abc import AsyncIterator, Callable
6-
from contextlib import nullcontext
76
from io import BytesIO
87
from typing import Literal, Union
98
from unittest.mock import MagicMock
@@ -183,7 +182,10 @@ def test_config_gzip_compress_level_validation(gzip_compress_level: int, should_
183182
),
184183
)
185184
def test_config_zstd_compress_level_validation(zstd_compress_level: int, should_raise: bool) -> None:
186-
with pytest.raises(ImproperlyConfiguredException) if should_raise else nullcontext():
185+
if should_raise:
186+
with pytest.raises(ImproperlyConfiguredException):
187+
CompressionConfig(backend="zstd", zstd_compress_level=zstd_compress_level)
188+
else:
187189
CompressionConfig(backend="zstd", zstd_compress_level=zstd_compress_level)
188190

189191

0 commit comments

Comments
 (0)