Skip to content

Commit f47f942

Browse files
committed
actually test InsistentReaderBytesIO
1 parent de4d7f0 commit f47f942

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/unit/test_util_streams.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717

1818
from aws_encryption_sdk.exceptions import ActionNotAllowedError
1919
from aws_encryption_sdk.internal.str_ops import to_bytes, to_str
20-
from aws_encryption_sdk.internal.utils.streams import ROStream, TeeStream
20+
from aws_encryption_sdk.internal.utils.streams import InsistentReaderBytesIO, ROStream, TeeStream
2121

2222
from .unit_test_utils import NothingButRead, SometimesIncompleteReaderIO
2323

@@ -67,7 +67,7 @@ def test_teestream_full():
6767
@pytest.mark.parametrize("bytes_to_read", range(1, 102))
6868
@pytest.mark.parametrize("source_length", (1, 11, 100))
6969
def test_insistent_stream(source_length, bytes_to_read, stream_type, converter):
70-
source = data(length=source_length, stream_type=stream_type, converter=converter)
70+
source = InsistentReaderBytesIO(data(length=source_length, stream_type=stream_type, converter=converter))
7171

7272
test = source.read(bytes_to_read)
7373

0 commit comments

Comments
 (0)