import sys
with open('test_visual_photo_pipeline.py', 'r') as f:
lines = f.readlines()
start = None
for i, line in enumerate(lines):
if line.strip().startswith('def test_extreme_image_dimensions_are_rejected(self):'):
start = i
break
if start is None:
print("Method not found")
sys.exit(1)
end = start + 1
while end < len(lines) and not lines[end].startswith(' def'):
end += 1
new_method = [
' def test_extreme_image_dimensions_are_rejected(self):\n',
' import visual_render\n',
' print("visual_render.__file__:", visual_render.__file__)\n',
' import hashlib\n',
' with open(visual_render.__file__, \'rb\') as f:\n',
' data = f.read()\n',
' print("SHA-256:", hashlib.sha256(data).hexdigest())\n',
' print("id(visual_render):", id(visual_render))\n',
' attrs = [\'_validated_image\', \'_validate_public_https_url\', \'select_photo_asset\', \'load_photo_data_uri\', \'render_template_html\', \'_PinnedHTTPSConnection\']\n',
' for a in attrs:\n',
' print(f"hasattr(visual_render, \\"{a}\\"):", hasattr(visual_render, a))\n',
' print("[name for name in dir(visual_render) if name in (\\"_validated_image\\", \\"_validate_public_https_url\\", \\"select_photo_asset\\", \\"load_photo_data_uri\\", \\"render_template_html\\", \\"_PinnedHTTPSConnection\\")]:", [name for name in dir(visual_render) if name in ("_validated_image", "_validate_public_https_url", "select_photo_asset", "load_photo_data_uri", "render_template_html", "_PinnedHTTPSConnection")])\n',
'\n',
' image = io.BytesIO()\n',
' Image.new("RGB", (20001, 1), "blue").save(image, format="PNG")\n',
'\n',
' with self.assertRaisesRegex(ValueError, "[Aa]bmessungen"):\n',
' visual_render._validated_image(image.getvalue())\n',
'\n'
]
new_lines = lines[:start] + new_method + lines[end:]
with open('test_visual_photo_pipeline_modified.py', 'w') as f:
f.writelines(new_lines)