def iter_fixture(self, _iter_chunk_offsets_, _ChunkFactory_, stream_rdr_,
chunk_, chunk_2_):
chunk_parser = _ChunkParser(stream_rdr_)
offsets = [2, 4, 6]
chunk_lst = [chunk_, chunk_2_]
return (chunk_parser, _iter_chunk_offsets_, _ChunkFactory_,
stream_rdr_, offsets, chunk_lst)
def iter_offsets_fixture(self):
bytes_ = b'-filler-\x00\x00\x00\x00IHDRxxxx\x00\x00\x00\x00IEND'
stream_rdr = StreamReader(BytesIO(bytes_), BIG_ENDIAN)
chunk_parser = _ChunkParser(stream_rdr)
expected_chunk_offsets = [
(PNG_CHUNK_TYPE.IHDR, 16),
(PNG_CHUNK_TYPE.IEND, 28),
]
return chunk_parser, expected_chunk_offsets
def iter_offsets_fixture(self):
bytes_ = b'-filler-\x00\x00\x00\x00IHDRxxxx\x00\x00\x00\x00IEND'
stream_rdr = StreamReader(BytesIO(bytes_), BIG_ENDIAN)
chunk_parser = _ChunkParser(stream_rdr)
expected_chunk_offsets = [
(PNG_CHUNK_TYPE.IHDR, 16),
(PNG_CHUNK_TYPE.IEND, 28),
]
return chunk_parser, expected_chunk_offsets
def iter_fixture(
self, _iter_chunk_offsets_, _ChunkFactory_, stream_rdr_, chunk_,
chunk_2_):
chunk_parser = _ChunkParser(stream_rdr_)
offsets = [2, 4, 6]
chunk_lst = [chunk_, chunk_2_]
return (
chunk_parser, _iter_chunk_offsets_, _ChunkFactory_, stream_rdr_,
offsets, chunk_lst
)
def it_can_iterate_over_the_chunks_in_its_png_stream(
self, _iter_chunk_offsets_, _ChunkFactory_, stream_rdr_, chunk_, chunk_2_
):
offsets = [2, 4, 6]
chunk_lst = [chunk_, chunk_2_]
chunk_parser = _ChunkParser(stream_rdr_)
chunks = [chunk for chunk in chunk_parser.iter_chunks()]
_iter_chunk_offsets_.assert_called_once_with(chunk_parser)
assert _ChunkFactory_.call_args_list == [
call(PNG_CHUNK_TYPE.IHDR, stream_rdr_, offsets[0]),
call(PNG_CHUNK_TYPE.pHYs, stream_rdr_, offsets[1]),
]
assert chunks == chunk_lst
def it_can_iterate_over_the_chunks_in_its_png_stream(
self, _iter_chunk_offsets_, _ChunkFactory_, stream_rdr_, chunk_,
chunk_2_):
offsets = [2, 4, 6]
chunk_lst = [chunk_, chunk_2_]
chunk_parser = _ChunkParser(stream_rdr_)
chunks = [chunk for chunk in chunk_parser.iter_chunks()]
_iter_chunk_offsets_.assert_called_once_with(chunk_parser)
assert _ChunkFactory_.call_args_list == [
call(PNG_CHUNK_TYPE.IHDR, stream_rdr_, offsets[0]),
call(PNG_CHUNK_TYPE.pHYs, stream_rdr_, offsets[1]),
]
assert chunks == chunk_lst
