def close_work_item(work_item, delete_from_queue):
"""
Finish handling the given work_item, deleting it from the input
queue if delete_from_queue is true
"""
logging.info("%s asset %s" % ((delete_from_queue and 'closing'
or 'aborting'), work_item['Asset-ID']))
local_path = work_item.get('Local-Path')
if delete_from_queue:
sqs.delete_message(work_item)
if local_path and os.path.exists(local_path):
shutil.rmtree(os.path.dirname(local_path))
def test_post_message(self):
sqs.clear_all_messages()
received = sqs.get_message(max_wait_time=10)
self.assert_(received is None, 'Queue should be empty')
original = {'body': r'This is a test message\nWith two lines\n'}
sqs.post_message(original)
received = sqs.get_message(max_wait_time=10, visibility_timeout=5)
self.assert_(received is not None)
self.assert_(received['body'] == original['body'])
sqs.delete_message(received)
self.assert_(sqs.get_message(max_wait_time=15) is None)
def close_work_item(work_item, delete_from_queue):
"""
Finish handling the given work_item, deleting it from the input
queue if delete_from_queue is true
"""
logging.info(
"%s asset %s" % (
(delete_from_queue and 'closing' or 'aborting'),
work_item['Asset-ID'] ))
local_path = work_item.get('Local-Path')
if delete_from_queue:
sqs.delete_message(work_item)
if local_path and os.path.exists(local_path):
shutil.rmtree(os.path.dirname(local_path))
def test_post_message(self):
sqs.clear_all_messages()
received = sqs.get_message( max_wait_time = 10)
self.assert_( received is None, 'Queue should be empty' )
original = { 'body' : r'This is a test message\nWith two lines\n' }
sqs.post_message( original )
received = sqs.get_message( max_wait_time = 10, visibility_timeout = 5 )
self.assert_( received is not None )
self.assert_( received['body'] == original['body'] )
sqs.delete_message(received)
self.assert_( sqs.get_message( max_wait_time = 15 ) is None )
