remove source_stream.closed from StreamDecryptor._read_bytes()

Author: mattsb42-aws

src/aws_encryption_sdk/streaming_client.py

@@ -904,8 +904,8 @@ def _read_bytes(self, b):
         :param int b: Number of bytes to read
         :raises NotSupportedError: if content type is not supported
         """
-        if self.source_stream.closed:
-            _LOGGER.debug("Source stream closed")
+        if hasattr(self, "footer"):
+            _LOGGER.debug("Source stream processing complete")
             return
 
         buffer_length = len(self.output_buffer)

test/unit/test_streaming_client_stream_decryptor.py

@@ -502,10 +502,10 @@ def test_read_bytes_from_framed_body_bad_sequence_number(self):
 
     @patch("aws_encryption_sdk.streaming_client.StreamDecryptor._read_bytes_from_non_framed_body")
     @patch("aws_encryption_sdk.streaming_client.StreamDecryptor._read_bytes_from_framed_body")
-    def test_read_bytes_closed(self, mock_read_frame, mock_read_block):
+    def test_read_bytes_completed(self, mock_read_frame, mock_read_block):
         ct_stream = io.BytesIO(VALUES["data_128"])
         test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=ct_stream)
-        test_decryptor.source_stream.close()
+        test_decryptor.footer = None
         test_decryptor._read_bytes(5)
         assert not mock_read_frame.called
         assert not mock_read_block.called