python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-06-30 18:03:07 +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: def test_frombytes() -> None:
with open("Tests/images/full_gimp_palette.gpl", "rb") as fp: # Test that __init__ stops reading after 260 lines
full_data = fp.read() 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 # 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" data = full_data.replace(b"#\n", b"") + b" 0 0 0 Index 256"
b = BytesIO(data) b = BytesIO(data)
palette = GimpPaletteFile(b) palette = GimpPaletteFile(b)