def test_close(self, mock_close): self.mock_data_encryption_key.key_provider = VALUES["key_provider"] self.mock_data_encryption_key.encrypted_data_key = VALUES["encrypted_data_key"] pt_stream = io.BytesIO(self.plaintext) test_encryptor = StreamEncryptor(source=pt_stream, key_provider=self.mock_key_provider) test_encryptor._derived_data_key = sentinel.derived_data_key test_encryptor.close() mock_close.assert_called_once_with()
def test_close(self, mock_close): self.mock_encryption_data_key.key_provider = VALUES['key_provider'] self.mock_encryption_data_key.encrypted_data_key = VALUES['encrypted_data_key'] pt_stream = io.BytesIO(self.plaintext) test_encryptor = StreamEncryptor( source=pt_stream, key_provider=self.mock_key_provider ) test_encryptor.encryption_data_key = self.mock_encryption_data_key test_encryptor.close() mock_close.assert_called_once_with()
def test_close(self, mock_close): self.mock_data_encryption_key.key_provider = VALUES["key_provider"] self.mock_data_encryption_key.encrypted_data_key = VALUES[ "encrypted_data_key"] pt_stream = io.BytesIO(self.plaintext) test_encryptor = StreamEncryptor( source=pt_stream, materials_manager=self.mock_materials_manager, commitment_policy=self.mock_commitment_policy, signature_policy=self.mock_signature_policy, ) test_encryptor._derived_data_key = sentinel.derived_data_key test_encryptor.close() mock_close.assert_called_once_with()