Ejemplo n.º 1
0
 def test_read_bytes_completed(self, mock_read_non_framed, mock_read_framed):
     pt_stream = io.BytesIO(self.plaintext)
     test_encryptor = StreamEncryptor(source=pt_stream, key_provider=self.mock_key_provider)
     test_encryptor._StreamEncryptor__message_complete = True
     test_encryptor._read_bytes(5)
     assert not mock_read_non_framed.called
     assert not mock_read_framed.called
Ejemplo n.º 2
0
 def test_read_bytes_completed(self, mock_read_non_framed,
                               mock_read_framed):
     pt_stream = io.BytesIO(self.plaintext)
     test_encryptor = StreamEncryptor(
         source=pt_stream,
         materials_manager=self.mock_materials_manager,
         commitment_policy=self.mock_commitment_policy,
     )
     test_encryptor._StreamEncryptor__message_complete = True
     test_encryptor._read_bytes(5)
     assert not mock_read_non_framed.called
     assert not mock_read_framed.called