def test_close_no_footer(self, mock_close):
self.mock_header.content_type = ContentType.FRAMED_DATA
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider,
source=self.mock_input_stream)
with six.assertRaisesRegex(self, SerializationError,
'Footer not read'):
test_decryptor.close()
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_close_no_footer(self, mock_close):
self.mock_header.content_type = ContentType.FRAMED_DATA
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider,
source=self.mock_input_stream)
with pytest.raises(SerializationError) as excinfo:
test_decryptor.close()
excinfo.match("Footer not read")
def test_close_no_footer(self, mock_close):
self.mock_header.content_type = ContentType.FRAMED_DATA
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=self.mock_input_stream,
commitment_policy=self.mock_commitment_policy,
)
with pytest.raises(SerializationError) as excinfo:
test_decryptor.close()
excinfo.match("Footer not read")
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()
