def test_prep_message_non_framed_message(self, mock_read_header,
mock_prep_non_framed):
self.mock_header.content_type = ContentType.NO_FRAMING
mock_read_header.return_value = self.mock_header, sentinel.header_auth
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider,
source=self.mock_input_stream)
test_decryptor._prep_message()
mock_prep_non_framed.assert_called_once_with()
def test_prep_message_framed_message(self, mock_read_header, mock_prep_non_framed):
self.mock_header.content_type = ContentType.FRAMED_DATA
mock_read_header.return_value = self.mock_header, sentinel.header_auth
test_decryptor = StreamDecryptor(key_provider=self.mock_key_provider, source=self.mock_input_stream)
test_decryptor._prep_message()
mock_read_header.assert_called_once_with()
assert test_decryptor._header is self.mock_header
assert test_decryptor.header_auth is sentinel.header_auth
assert not mock_prep_non_framed.called
assert test_decryptor._message_prepped
def test_prep_message_non_framed_message(self, mock_read_header,
mock_prep_non_framed):
self.mock_header.content_type = ContentType.NO_FRAMING
mock_read_header.return_value = self.mock_header, sentinel.header_auth
test_decryptor = StreamDecryptor(
materials_manager=self.mock_materials_manager,
source=self.mock_input_stream,
commitment_policy=self.mock_commitment_policy,
)
test_decryptor._prep_message()
mock_prep_non_framed.assert_called_once_with()
