def test_get_blob_resolver(self):
# Arrange
self.bsc.require_encryption = True
self.bsc.key_encryption_key = KeyWrapper('key1')
key_resolver = KeyResolver()
key_resolver.put_key(self.bsc.key_encryption_key)
self.bsc.key_resolver_function = key_resolver.resolve_key
blob = self._create_small_blob(BlobType.BlockBlob)
# Act
self.bsc.key_encryption_key = None
content = blob.download_blob().content_as_bytes()
# Assert
self.assertEqual(content, self.bytes)
async def _test_peek_messages_encrypted_resolver(self):
# Arrange
self.qsc.key_encryption_key = KeyWrapper('key1')
queue = await self._create_queue()
await queue.enqueue_message(u'encrypted_message_4')
key_resolver = KeyResolver()
key_resolver.put_key(self.qsc.key_encryption_key)
queue.key_resolver_function = key_resolver.resolve_key
queue.key_encryption_key = None # Ensure that the resolver is used
# Act
li = await queue.peek_messages()
# Assert
self.assertEqual(li[0].content, u'encrypted_message_4')
def test_get_messages_encrypted_resolver(self):
# Arrange
self.qsc.key_encryption_key = KeyWrapper('key1')
queue = self._create_queue()
queue.enqueue_message(u'encrypted_message_2')
key_resolver = KeyResolver()
key_resolver.put_key(self.qsc.key_encryption_key)
queue.key_resolver_function = key_resolver.resolve_key
queue.key_encryption_key = None # Ensure that the resolver is used
# Act
li = next(queue.receive_messages())
# Assert
self.assertEqual(li.content, u'encrypted_message_2')
async def test_get_blob_resolver_async(self, storage_account_name, storage_account_key):
await self._setup(storage_account_name, storage_account_key)
self.bsc.require_encryption = True
self.bsc.key_encryption_key = KeyWrapper('key1')
key_resolver = KeyResolver()
key_resolver.put_key(self.bsc.key_encryption_key)
self.bsc.key_resolver_function = key_resolver.resolve_key
blob = await self._create_small_blob(BlobType.BlockBlob)
# Act
self.bsc.key_encryption_key = None
content = await (await blob.download_blob()).content_as_bytes()
# Assert
self.assertEqual(content, self.bytes)
def test_get_messages_encrypted_resolver(self, storage_account_name, storage_account_key):
# Arrange
qsc = QueueServiceClient(self.account_url(storage_account_name, "queue"), storage_account_key)
qsc.key_encryption_key = KeyWrapper('key1')
queue = self._create_queue(qsc)
queue.send_message(u'encrypted_message_2')
key_resolver = KeyResolver()
key_resolver.put_key(qsc.key_encryption_key)
queue.key_resolver_function = key_resolver.resolve_key
queue.key_encryption_key = None # Ensure that the resolver is used
# Act
li = next(queue.receive_messages())
# Assert
self.assertEqual(li.content, u'encrypted_message_2')
async def test_peek_messages_encrypted_resolver(self, resource_group, location, storage_account, storage_account_key):
qsc = QueueServiceClient(self._account_url(storage_account.name), storage_account_key, transport=AiohttpTestTransport())
# Arrange
qsc.key_encryption_key = KeyWrapper('key1')
queue = await self._create_queue(qsc)
await queue.send_message(u'encrypted_message_4')
key_resolver = KeyResolver()
key_resolver.put_key(qsc.key_encryption_key)
queue.key_resolver_function = key_resolver.resolve_key
queue.key_encryption_key = None # Ensure that the resolver is used
# Act
li = await queue.peek_messages()
# Assert
self.assertEqual(li[0].content, u'encrypted_message_4')