Explorer
/tmp/h005_diag/modify_test.py
← Zurück ↓ Download
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)