def Generate() -> Iterator[bytes]:
chunks = ioutil.Chunk(stream, size=_AEAD_CHUNK_SIZE)
chunks = iterator.Lookahead(enumerate(chunks))
for idx, chunk in chunks:
nonce = os.urandom(_AEAD_NONCE_SIZE)
adata = _AEAD_ADATA_FORMAT.pack(idx, chunks.done)
yield nonce + aesgcm.encrypt(nonce, chunk, adata)
def testSizeZero(self):
with self.assertRaises(ValueError):
list(ioutil.Chunk(io.BytesIO(b"foobar"), size=0))
def testManyUnevenChunks(self):
chunks = list(ioutil.Chunk(io.BytesIO(b"foobarbaz"), size=2))
self.assertEqual(chunks, [b"fo", b"ob", b"ar", b"ba", b"z"])
def testSingleChunk(self):
chunks = list(ioutil.Chunk(io.BytesIO(b"foobar"), size=6))
self.assertEqual(chunks, [b"foobar"])
def testEmpty(self):
chunks = list(ioutil.Chunk(io.BytesIO(b""), size=3))
self.assertEmpty(chunks)
def testSizeNegative(self):
with self.assertRaises(ValueError):
list(ioutil.Chunk(io.BytesIO(b"foobar"), size=-42))