def test_simpleencryptio():
key = b"123"
buf = os.urandom(1024 * 1024 * 50)
bio = io.BytesIO(buf)
c = SimpleEncryptIO(bio, key, len(buf))
assert total_len(c) == len(buf) + PADDED_ENCRYPT_HEAD_WITH_SALT_LEN
enc = c.read()
d = to_decryptio(io.BytesIO(enc), key)
assert total_len(d) == len(buf)
dec = d.read()
assert buf == dec
def test_simpleencryptio():
key = "123"
nonce_or_iv = os.urandom(16)
buf = os.urandom(1024 * 1024 * 50)
bio = io.BytesIO(buf)
c = SimpleEncryptIO(bio, key, nonce_or_iv, len(buf))
assert total_len(c) == len(buf) + ENCRYPT_HEAD_LEN
enc = c.read()
d = to_decryptio(io.BytesIO(enc), key)
assert total_len(d) == len(buf)
dec = d.read()
assert buf == dec