def test_create_and_get_task(self):
task = Task.create(name='foo', inputs=['foo','bar'])
self.assertEquals('foo', task.name)
self.assertEquals(2, len(task.inputs))
self.assertTrue(task.is_processing())
gotten = Task.get(task.id)
self.assertEquals('foo', gotten.name)
self.assertEquals(2, len(gotten.inputs))
self.assertTrue(task.is_processing())
def test_create_task(self):
data = {
'name' : 'foo',
'url' : 'https://s3.amazonaws.com/sonofjscompute-testing/sample.json'
}
resp = self.client.post('/tasks/', data=data)
self.assertEquals(201, resp.status_code)
json = loads(resp.data)
self.assertTrue(json['id'])
self.assertEquals('foo', json['name'])
inputs = Task.get(json['id']).get_requests()
self.assertEquals(2, len(inputs))
self.assertEquals('foo', inputs[0])
self.assertEquals('bar', inputs[1])
def run_worker(task_id):
task_id = int(task_id)
task = Task.get(task_id)
processes = multiprocessing.cpu_count() or 1
pool = multiprocessing.Pool(processes)
pool.map(_work, [task] * processes)
