def test_peek_messages_encrypted_resolver(self):
# Arrange
self.qs.key_encryption_key = KeyWrapper('key1')
queue_name = self._create_queue()
self.qs.put_message(queue_name, u'encrypted_message_4')
key_resolver = KeyResolver()
key_resolver.put_key(self.qs.key_encryption_key)
self.resolver = key_resolver.resolve_key
# Act
li = self.qs.peek_messages(queue_name)
# Assert
self.assertEqual(li[0].content, u'encrypted_message_4')
def test_peek_messages_encrypted_resolver(self):
# Arrange
self.qs.key_encryption_key = KeyWrapper('key1')
queue_name = self._create_queue()
self.qs.put_message(queue_name, u'encrypted_message_4')
key_resolver = KeyResolver()
key_resolver.put_key(self.qs.key_encryption_key)
self.resolver = key_resolver.resolve_key
# Act
li = self.qs.peek_messages(queue_name)
# Assert
self.assertEqual(li[0].content, u'encrypted_message_4')
def test_get_blob_resolver(self):
# Arrange
self.bbs.require_encryption = True
self.bbs.key_encryption_key = KeyWrapper('key1')
key_resolver = KeyResolver()
key_resolver.put_key(self.bbs.key_encryption_key)
self.bbs.key_resolver_function = key_resolver.resolve_key
blob_name = self._create_small_blob('block_blob')
# Act
self.bbs.key_encryption_key = None
blob = self.bbs.get_blob_to_bytes(self.container_name, blob_name)
# Assert
self.assertEqual(blob.content, self.bytes)
def test_get_messages_encrypted_resolver(self):
# Arrange
self.qs.key_encryption_key = KeyWrapper('key1')
queue_name = self._create_queue()
self.qs.put_message(queue_name, u'encrypted_message_2')
key_resolver = KeyResolver()
key_resolver.put_key(self.qs.key_encryption_key)
self.qs.key_resolver_function = key_resolver.resolve_key
self.qs.key_encryption_key = None # Ensure that the resolver is used
# Act
li = self.qs.get_messages(queue_name)
# Assert
self.assertEqual(li[0].content, u'encrypted_message_2')
def test_get_messages_encrypted_resolver(self):
# Arrange
self.qs.key_encryption_key = KeyWrapper('key1')
queue_name = self._create_queue()
self.qs.put_message(queue_name, u'encrypted_message_2')
key_resolver = KeyResolver()
key_resolver.put_key(self.qs.key_encryption_key)
self.qs.key_resolver_function = key_resolver.resolve_key
self.qs.key_encryption_key = None # Ensure that the resolver is used
# Act
li = self.qs.get_messages(queue_name)
# Assert
self.assertEqual(li[0].content, u'encrypted_message_2')
def test_get_blob_resolver(self):
# Arrange
self.bbs.require_encryption = True
self.bbs.key_encryption_key = KeyWrapper('key1')
key_resolver = KeyResolver()
key_resolver.put_key(self.bbs.key_encryption_key)
self.bbs.key_resolver_function = key_resolver.resolve_key
blob_name = self._create_small_blob('block_blob')
# Act
self.bbs.key_encryption_key = None
blob = self.bbs.get_blob_to_bytes(self.container_name, blob_name)
# Assert
self.assertEqual(blob.content, self.bytes)
def test_get_encrypted_entity_key_resolver(self):
# Arrange
self.ts.require_encryption = True
entity = self._create_default_entity_for_encryption()
self.ts.key_encryption_key = KeyWrapper('key1')
key_resolver = KeyResolver()
key_resolver.put_key(self.ts.key_encryption_key)
self.ts.key_resolver_function = key_resolver.resolve_key
self.ts.insert_entity(self.table_name, entity)
# Act
self.ts.key_encryption_key = None
new_entity = self.ts.get_entity(self.table_name, entity['PartitionKey'], entity['RowKey'])
# Assert
self._assert_default_entity(new_entity)