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)