async def _test_update_encrypted_raw_text_message(self):
# TODO: Recording doesn't work
if TestMode.need_recording_file(self.test_mode):
return
# Arrange
queue = await self._create_queue()
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
raw_text = u'Update Me'
await queue.enqueue_message(raw_text)
messages = []
async for m in queue.receive_messages():
messages.append(m)
list_result1 = messages[0]
# Act
raw_text = u'Updated'
list_result1.content = raw_text
async for m in queue.receive_messages():
messages.append(m)
list_result2 = messages[0]
# Assert
self.assertEqual(raw_text, list_result2.content)
async def _test_update_encrypted_json_message(self):
# TODO: Recording doesn't work
if TestMode.need_recording_file(self.test_mode):
return
# Arrange
queue = await self._create_queue()
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
message_dict = {'val1': 1, 'val2': '2'}
json_text = dumps(message_dict)
await queue.enqueue_message(json_text)
messages = []
async for m in queue.receive_messages():
messages.append(m)
list_result1 = messages[0]
# Act
message_dict['val1'] = 0
message_dict['val2'] = 'updated'
json_text = dumps(message_dict)
list_result1.content = json_text
await queue.update_message(list_result1)
async for m in queue.receive_messages():
messages.append(m)
list_result2 = messages[0]
# Assert
self.assertEqual(message_dict, loads(list_result2.content))
def test_update_encrypted_json_message(self, resource_group, location,
storage_account,
storage_account_key):
# TODO: Recording doesn't work
if not self.is_live:
return
# Arrange
qsc = QueueServiceClient(self._account_url(storage_account.name),
storage_account_key)
queue = self._create_queue(qsc)
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
message_dict = {'val1': 1, 'val2': '2'}
json_text = dumps(message_dict)
queue.enqueue_message(json_text)
messages = queue.receive_messages()
list_result1 = next(messages)
# Act
message_dict['val1'] = 0
message_dict['val2'] = 'updated'
json_text = dumps(message_dict)
list_result1.content = json_text
queue.update_message(list_result1)
list_result2 = next(messages)
# Assert
self.assertEqual(message_dict, loads(list_result2.content))
def test_update_encrypted_raw_text_message(self, resource_group, location,
storage_account,
storage_account_key):
# TODO: Recording doesn't work
if not self.is_live:
return
# Arrange
qsc = QueueServiceClient(self._account_url(storage_account.name),
storage_account_key)
queue = self._create_queue(qsc)
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
raw_text = u'Update Me'
queue.enqueue_message(raw_text)
messages = queue.receive_messages()
list_result1 = next(messages)
# Act
raw_text = u'Updated'
list_result1.content = raw_text
queue.update_message(list_result1)
list_result2 = next(messages)
# Assert
self.assertEqual(raw_text, list_result2.content)
async def test_update_encrypted_raw_text_message(self, resource_group,
location, storage_account,
storage_account_key):
qsc = QueueServiceClient(self._account_url(storage_account.name),
storage_account_key,
transport=AiohttpTestTransport())
# TODO: Recording doesn't work
if not self.is_live:
return
# Arrange
queue = await self._create_queue(qsc)
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
raw_text = u'Update Me'
await queue.enqueue_message(raw_text)
messages = []
async for m in queue.receive_messages():
messages.append(m)
list_result1 = messages[0]
# Act
raw_text = u'Updated'
list_result1.content = raw_text
async for m in queue.receive_messages():
messages.append(m)
list_result2 = messages[0]
# Assert
self.assertEqual(raw_text, list_result2.content)
async def test_update_encrypted_json_message(self, resource_group,
location, storage_account,
storage_account_key):
qsc = QueueServiceClient(self._account_url(storage_account.name),
storage_account_key,
transport=AiohttpTestTransport())
# TODO: Recording doesn't work
if not self.is_live:
return
# Arrange
queue = await self._create_queue(qsc)
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
message_dict = {'val1': 1, 'val2': '2'}
json_text = dumps(message_dict)
await queue.enqueue_message(json_text)
messages = []
async for m in queue.receive_messages():
messages.append(m)
list_result1 = messages[0]
# Act
message_dict['val1'] = 0
message_dict['val2'] = 'updated'
json_text = dumps(message_dict)
list_result1.content = json_text
await queue.update_message(list_result1)
async for m in queue.receive_messages():
messages.append(m)
list_result2 = messages[0]
# Assert
self.assertEqual(message_dict, loads(list_result2.content))
def test_update_encrypted_raw_text_message(self):
# TODO: Recording doesn't work
if TestMode.need_recording_file(self.test_mode):
return
# Arrange
queue = self._create_queue()
queue.key_encryption_key = KeyWrapper('key1')
queue._config.message_encode_policy = NoEncodePolicy()
queue._config.message_decode_policy = NoDecodePolicy()
raw_text = u'Update Me'
queue.enqueue_message(raw_text)
messages = queue.receive_messages()
list_result1 = next(messages)
# Act
raw_text = u'Updated'
list_result1.content = raw_text
queue.update_message(list_result1)
list_result2 = next(messages)
# Assert
self.assertEqual(raw_text, list_result2.content)