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()