def test_add_pull_and_push_queue_tasks(self):
"""Ensure that push and pull tasks can be added with add_tasks()."""
from furious.async import Async
from furious.batcher import Message
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import get_tasks
from furious.test_stubs.appengine.queues import purge_tasks
# Add tasks the normal way so we can get them and test readding them
async = Async(target='time.ctime')
async.start()
async2 = Async(target='time.ctime')
async2.start()
Message(queue='default-pull').insert()
task_dict = get_tasks(self.queue_service)
# purge current tasks so we can verify how many we will add next.
purge_tasks(self.queue_service)
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to check how many tasks are in the queues
num_queued = purge_tasks(self.queue_service)
self.assertEqual(3, num_added)
self.assertEqual(3, num_queued)
def test_add_pull_and_push_queue_tasks(self):
"""Ensure that push and pull tasks can be added with add_tasks()."""
from furious.async import Async
from furious.batcher import Message
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import get_tasks
from furious.test_stubs.appengine.queues import purge_tasks
# Add tasks the normal way so we can get them and test readding them
async = Async(target='time.ctime')
async.start()
async2 = Async(target='time.ctime')
async2.start()
Message(queue='default-pull').insert()
task_dict = get_tasks(self.queue_service)
# purge current tasks so we can verify how many we will add next.
purge_tasks(self.queue_service)
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to check how many tasks are in the queues
num_queued = purge_tasks(self.queue_service)
self.assertEqual(3, num_added)
self.assertEqual(3, num_queued)
def test_add_tasks_when_there_are_no_tasks(self):
"""Ensure that no tasks are added to add_tasks() when the
task_dict is empty.
"""
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import purge_tasks
task_dict = {}
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to count if any tasks remained.
num_purged = purge_tasks(self.queue_service)
self.assertEqual(0, num_added)
self.assertEqual(0, num_purged)
def test_add_tasks_when_there_are_no_tasks(self):
"""Ensure that no tasks are added to add_tasks() when the
task_dict is empty.
"""
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import purge_tasks
task_dict = {}
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to count if any tasks remained.
num_purged = purge_tasks(self.queue_service)
self.assertEqual(0, num_added)
self.assertEqual(0, num_purged)
def test_add_async_and_message_tasks(self, ctime):
"""Ensure taskqueue.Task() instances from furious Asyncs and Messages
can be added.
"""
from google.appengine.api import taskqueue
from furious. async import Async
from furious.batcher import Message
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import run as run_queues
# Create asyncs
async = Async(target='time.ctime')
async2 = Async(target='time.ctime')
# Create a message
options = {'task_args': {'payload': 'abcdefg'}}
message = Message(payload='abc', **options)
message_task = message.to_task()
task_dict = {
'default': [async .to_task(), async2.to_task()],
'default-pull': [message_task]
}
num_added = add_tasks(self.queue_service, task_dict)
# Ensure three tasks were added.
self.assertEqual(3, num_added)
# Run the tasks to make sure they were inserted correctly.
run_queues(self.queue_service)
# Ensure both push queue tasks were executed.
self.assertEqual(2, ctime.call_count)
# Lease the pull queue task and make sure it has the correct payload.
tasks = taskqueue.Queue('default-pull').lease_tasks(3600, 100)
returned_task_message = tasks[0]
# Ensure pull queue task payload is the same as the original.
self.assertEqual(returned_task_message.payload, message_task.payload)
def test_add_tasks_with_empty_queues(self, queue_add):
"""Ensure qeueue.add() is not called when there are no tasks to queue.
In some cases adding an empty list causes an error in the taskqueue
stub.
"""
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import purge_tasks
task_dict = {'default': [], 'default-pull': []}
# Add using empty lists of tasks.
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to verify the count of tasks added.
num_purged = purge_tasks(self.queue_service)
# Ensure no tasks were added.
self.assertEqual(0, queue_add.call_count)
self.assertEqual(0, num_added)
self.assertEqual(0, num_purged)
def test_add_tasks_with_empty_queues(self, queue_add):
"""Ensure qeueue.add() is not called when there are no tasks to queue.
In some cases adding an empty list causes an error in the taskqueue
stub.
"""
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import purge_tasks
task_dict = {'default': [], 'default-pull': []}
# Add using empty lists of tasks.
num_added = add_tasks(self.queue_service, task_dict)
# Purge tasks to verify the count of tasks added.
num_purged = purge_tasks(self.queue_service)
# Ensure no tasks were added.
self.assertEqual(0, queue_add.call_count)
self.assertEqual(0, num_added)
self.assertEqual(0, num_purged)
def test_add_async_and_message_tasks(self, ctime):
"""Ensure taskqueue.Task() instances from furious Asyncs and Messages
can be added.
"""
from google.appengine.api import taskqueue
from furious.async import Async
from furious.batcher import Message
from furious.test_stubs.appengine.queues import add_tasks
from furious.test_stubs.appengine.queues import run as run_queues
# Create asyncs
async = Async(target='time.ctime')
async2 = Async(target='time.ctime')
# Create a message
options = {'task_args': {'payload': 'abcdefg'}}
message = Message(payload='abc', **options)
message_task = message.to_task()
task_dict = {'default': [async.to_task(), async2.to_task()],
'default-pull': [message_task]}
num_added = add_tasks(self.queue_service, task_dict)
# Ensure three tasks were added.
self.assertEqual(3, num_added)
# Run the tasks to make sure they were inserted correctly.
run_queues(self.queue_service)
# Ensure both push queue tasks were executed.
self.assertEqual(2, ctime.call_count)
# Lease the pull queue task and make sure it has the correct payload.
tasks = taskqueue.Queue('default-pull').lease_tasks(3600, 100)
returned_task_message = tasks[0]
# Ensure pull queue task payload is the same as the original.
self.assertEqual(returned_task_message.payload, message_task.payload)
