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