python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-04-14 22:34:14 +03:00
Code Issues Packages Projects Releases Wiki Activity

Test that an unlimited number of lines is not read by __init__

Browse Source
This commit is contained in:
Andrew Murray 2025-03-21 08:51:41 +11:00
parent 510bc05577
commit 21ff960c9c
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Tests/test_file_gimppalette.py
View File

@ -41,10 +41,17 @@ def test_get_palette(filename: str, size: int) -> None:
def test_frombytes() -> None:
with open("Tests/images/full_gimp_palette.gpl", "rb") as fp:
full_data = fp.read()
# Test that __init__ stops reading after 260 lines
with open("Tests/images/custom_gimp_palette.gpl", "rb") as fp:
custom_data = fp.read()
custom_data += b"#\n" * 300 + b" 0 0 0 Index 12"
b = BytesIO(custom_data)
palette = GimpPaletteFile(b)
assert len(palette.palette) / 3 == 8
# Test that __init__ only reads 256 entries
with open("Tests/images/full_gimp_palette.gpl", "rb") as fp:
full_data = fp.read()
data = full_data.replace(b"#\n", b"") + b" 0 0 0 Index 256"
b = BytesIO(data)
palette = GimpPaletteFile(b)