Skip to content

Commit 05f9893

Browse files
committed
add tests to verify stream handler readable() behaviors
1 parent c2e53e1 commit 05f9893

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

test/functional/test_f_aws_encryption_sdk_client.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -574,3 +574,24 @@ def test_decrypt_oneshot_no_seek_input():
574574
ciphertext_no_seek = NoSeekBytesIO(ciphertext)
575575
decrypted, _header = aws_encryption_sdk.decrypt(source=ciphertext_no_seek, key_provider=key_provider)
576576
assert decrypted == VALUES["plaintext_128"]
577+
578+
579+
def test_stream_encryptor_readable():
580+
"""Verify that open StreamEncryptor instances report as readable."""
581+
key_provider = fake_kms_key_provider()
582+
plaintext = io.BytesIO(VALUES["plaintext_128"])
583+
with aws_encryption_sdk.StreamEncryptor(source=plaintext, key_provider=key_provider) as handler:
584+
assert handler.readable()
585+
handler.read()
586+
assert not handler.readable()
587+
588+
589+
def test_stream_decryptor_readable():
590+
"""Verify that open StreamEncryptor instances report as readable."""
591+
key_provider = fake_kms_key_provider()
592+
plaintext = io.BytesIO(VALUES["plaintext_128"])
593+
ciphertext, _header = aws_encryption_sdk.encrypt(source=plaintext, key_provider=key_provider)
594+
with aws_encryption_sdk.StreamDecryptor(source=ciphertext, key_provider=key_provider) as handler:
595+
assert handler.readable()
596+
handler.read()
597+
assert not handler.readable()

0 commit comments

Comments
 (0)