def test_split_into_segments_fail2(self):
with self.assertRaises(ValueError):
_common.split_into_segments(b"\xff\xd8\xff\xe1\xff\xff")
def test_merge_segments(self):
# Remove APP0, when both APP0 and APP1 exists.
with open(INPUT_FILE1, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments)
segments = _common.split_into_segments(new_data)
self.assertFalse([1][0:2] == b"\xff\xe0"
and segments[2][0:2] == b"\xff\xe1")
self.assertEqual(segments[1][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
without_app0 = o.getvalue()
Image.open(o).close()
exif = _common.get_app1(segments)
# Remove APP1, when second 'merged_segments' arguments is None
# and no APP0.
segments = _common.split_into_segments(without_app0)
new_data = _common.merge_segments(segments, None)
segments = _common.split_into_segments(new_data)
self.assertNotEqual(segments[1][0:2], b"\xff\xe0")
self.assertNotEqual(segments[1][0:2], b"\xff\xe1")
self.assertNotEqual(segments[2][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
# Insert exif to jpeg that has APP0 and APP1.
o = io.BytesIO()
i = Image.new("RGBA", (8, 8))
i.save(o, format="jpeg", exif=exif)
o.seek(0)
segments = _common.split_into_segments(o.getvalue())
new_data = _common.merge_segments(segments, exif)
segments = _common.split_into_segments(new_data)
self.assertFalse(segments[1][0:2] == b"\xff\xe0"
and segments[2][0:2] == b"\xff\xe1")
self.assertEqual(segments[1], exif)
o = io.BytesIO(new_data)
Image.open(o).close()
# Insert exif to jpeg that doesn't have APP0 and APP1.
with open(NOAPP01_FILE, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments, exif)
segments = _common.split_into_segments(new_data)
self.assertEqual(segments[1][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
# Remove APP1, when second 'merged_segments' arguments is None
# and APP1 exists.
with open(INPUT_FILE1, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments, None)
segments = _common.split_into_segments(new_data)
self.assertNotEqual(segments[1][0:2], b"\xff\xe1")
self.assertNotEqual(segments[2][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
def test_split_into_segments_fail1(self):
with self.assertRaises(ValueError):
_common.split_into_segments(b"I'm not JPEG")
def test_split_into_segments_fail1(self):
with self.assertRaises(InvalidImageDataError):
_common.split_into_segments(b"I'm not JPEG")
def test_merge_segments(self):
# Remove APP0, when both APP0 and APP1 exists.
with open(INPUT_FILE1, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments)
segments = _common.split_into_segments(new_data)
self.assertFalse([1][0:2] == b"\xff\xe0"
and segments[2][0:2] == b"\xff\xe1")
self.assertEqual(segments[1][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
without_app0 = o.getvalue()
Image.open(o).close()
exif = _common.get_exif_seg(segments)
# Remove Exif, when second 'merged_segments' arguments is None
# and no APP0.
segments = _common.split_into_segments(without_app0)
new_data = _common.merge_segments(segments, None)
segments = _common.split_into_segments(new_data)
self.assertNotEqual(segments[1][0:2], b"\xff\xe0")
self.assertNotEqual(segments[1][0:2], b"\xff\xe1")
self.assertNotEqual(segments[2][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
# Insert exif to jpeg that has APP0 and Exif.
o = io.BytesIO()
i = Image.new("RGBA", (8, 8))
i.save(o, format="jpeg", exif=exif)
o.seek(0)
segments = _common.split_into_segments(o.getvalue())
new_data = _common.merge_segments(segments, exif)
segments = _common.split_into_segments(new_data)
self.assertFalse(segments[1][0:2] == b"\xff\xe0"
and segments[2][0:2] == b"\xff\xe1")
self.assertEqual(segments[1], exif)
o = io.BytesIO(new_data)
Image.open(o).close()
# Insert exif to jpeg that doesn't have APP0 and Exif.
with open(NOAPP01_FILE, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments, exif)
segments = _common.split_into_segments(new_data)
self.assertEqual(segments[1][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
# Remove Exif, when second 'merged_segments' arguments is None
# and Exif exists.
with open(INPUT_FILE1, "rb") as f:
original = f.read()
segments = _common.split_into_segments(original)
new_data = _common.merge_segments(segments, None)
segments = _common.split_into_segments(new_data)
self.assertNotEqual(segments[1][0:2], b"\xff\xe1")
self.assertNotEqual(segments[2][0:2], b"\xff\xe1")
o = io.BytesIO(new_data)
Image.open(o).close()
def test_split_into_segments_fail2(self):
with self.assertRaises(ValueError):
_common.split_into_segments(b"\xff\xd8\xff\xe1\xff\xff")
def test_split_into_segments_fail1(self):
with self.assertRaises(ValueError):
_common.split_into_segments(b"I'm not JPEG")