async def test_peek_messages_encrypted_kek(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_3')
# Act
li = await queue.peek_messages()
# Assert
self.assertEqual(li[0].content, u'encrypted_message_3')
async def test_get_messages_encrypted_kek(self, resource_group, location, storage_account, storage_account_key):
# Arrange
qsc = QueueServiceClient(self._account_url(storage_account.name), storage_account_key, transport=AiohttpTestTransport())
qsc.key_encryption_key = KeyWrapper('key1')
queue = await self._create_queue(qsc)
await queue.send_message(u'encrypted_message_2')
# Act
li = None
async for m in queue.receive_messages():
li = m
# 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')
async def test_peek_messages_encrypted_kek_RSA(self, resource_group, location, storage_account, storage_account_key):
qsc = QueueServiceClient(self._account_url(storage_account.name), storage_account_key, transport=AiohttpTestTransport())
# We can only generate random RSA keys, so this must be run live or
# the playback test will fail due to a change in kek values.
if not self.is_live:
return
# Arrange
qsc.key_encryption_key = RSAKeyWrapper('key2')
queue = await self._create_queue(qsc)
await queue.send_message(u'encrypted_message_3')
# Act
li = await queue.peek_messages()
# Assert
self.assertEqual(li[0].content, u'encrypted_message_3')
async 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,
transport=AiohttpTestTransport())
qsc.key_encryption_key = KeyWrapper('key1')
queue = await self._create_queue(qsc)
await 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 = None
async for m in queue.receive_messages():
li = m
# Assert
self.assertEqual(li.content, u'encrypted_message_2')
