def test_close(self, mock_close):
self.mock_header.content_type = ContentType.NO_FRAMING
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=self.mock_input_stream)
test_decryptor.footer = sentinel.footer
test_decryptor.data_key = VALUES["data_key_obj"]
test_decryptor.close()
mock_close.assert_called_once_with()
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.footer = None
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
def test_close(self, mock_close):
self.mock_header.content_type = ContentType.NO_FRAMING
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=self.mock_input_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor.footer = sentinel.footer
test_decryptor.data_key = VALUES["data_key_obj"]
test_decryptor.close()
mock_close.assert_called_once_with()
def test_read_bytes_completed(self, mock_read_frame, mock_read_block):
ct_stream = io.BytesIO(VALUES["data_128"])
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=ct_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor.footer = None
test_decryptor._read_bytes(5)
assert not mock_read_frame.called
assert not mock_read_block.called
