def test_producer_should_load_all_tasks(self):
responses.add(responses.GET,
'https://api.github.com',
body='something',
status=200)
responses.add(responses.GET,
'https://api.github.com/repos/bigodines/api-health/',
body='mocked',
status=200)
t1 = Task(url="https://api.github.com")
t2 = Task(url="https://api.github.com/repos/bigodines/api-health/")
session.add(t1)
session.add(t2)
session.commit()
yield cron.producer()
self.assertEquals(cron.queue.qsize(), 2)
def add_task(self, args):
"""
Stores a new Task() and returns a TaskForm() to be rendered
or Raise and exception with the invalid/missing fields
"""
task = Task()
# TODO: write a helper in the base class to mimic form.populate_obj()
# or better yet: make form.populate_obj() work with dicts.
if 'expected_fields' in args:
if isinstance(args['expected_fields'], list):
task.expected_fields = ','.join(str(args['expected_fields']))
else:
task.expected_fields = args['expected_fields']
if 'url' in args:
task.url = args["url"]
session.add(task)
session.commit()
raise gen.Return(task)
