def test_async_function_scheduling(self):
# Check the message on the queue.
sqs_connection = boto.connect_sqs('', '')
sqs_connection.create_queue(self.queue_name)
queue = sqs_connection.get_queue(self.queue_name)
messages = queue.get_messages()
self.assertEquals(len(messages), 0)
# Schedule a function.
schedule_function(self.queue_name,
'a-function',
'1',
'2',
kwarg1=1,
kwarg2=2)
messages = queue.get_messages()
self.assertEquals(len(messages), 1)
# For some reason, boto base64-encodes the messages, but moto does
# not. Life.
self.assertEquals(
json.loads(messages[0].get_body()), {
FUNCTION: 'a-function',
ARGS: ['1', '2'],
KWARGS: {
'kwarg1': 1,
'kwarg2': 2
}
})
def schedule(self, project_id, step_slug):
if not settings.PRODUCTION and not settings.STAGING:
scheduler = SynchronousMachineStepScheduler()
scheduler.schedule(project_id, step_slug)
else:
schedule_function(settings.WORK_QUEUE, 'machine_task_executor',
project_id, step_slug)
def test_function_scheduling(self):
# Check the message on the queue.
sqs_connection = boto.connect_sqs('', '')
sqs_connection.create_queue(self.queue_name)
queue = sqs_connection.get_queue(self.queue_name)
messages = queue.get_messages()
self.assertEquals(len(messages), 0)
# Schedule a function.
schedule_function(
self.queue_name, 'a-function', '1', '2', kwarg1=1, kwarg2=2)
messages = queue.get_messages()
self.assertEquals(len(messages), 1)
# For some reason, boto base64-encodes the messages, but moto does
# not. Life.
self.assertEquals(
json.loads(messages[0].get_body()),
{FUNCTION: 'a-function', ARGS: ['1', '2'], KWARGS: {
'kwarg1': 1, 'kwarg2': 2}})
def test_sync_function_scheduling(self):
# Schedule a function.
schedule_function(self.queue_name, 'the_counter', 1, second_arg=5)
self.assertEquals(CALL_COUNTER, 6)
def test_sync_function_scheduling(self):
# Schedule a function.
schedule_function(self.queue_name, 'the_counter', 1, second_arg=5)
self.assertEquals(CALL_COUNTER, 6)
def schedule(self, project_id, step_slug):
if not settings.PRODUCTION and not settings.STAGING:
scheduler = SynchronousMachineStepScheduler()
scheduler.schedule(project_id, step_slug)
else:
schedule_function(settings.WORK_QUEUE, "machine_task_executor", project_id, step_slug)
