class GatorTestCase(unittest.TestCase):
def setUp(self):
super(GatorTestCase, self).setUp()
self.conn_string = os.environ.get('ALLIGATOR_CONN')
self.gator = Gator(self.conn_string)
# Just reach in & clear things out.
self.gator.backend.drop_all(ALL)
def test_init(self):
gator = Gator('whatever://', backend_class=CustomClient)
self.assertEqual(gator.conn_string, 'whatever://')
self.assertEqual(gator.queue_name, ALL)
self.assertEqual(gator.task_class, Task)
self.assertEqual(gator.backend_class, CustomClient)
self.assertTrue(isinstance(gator.backend, CustomClient))
gator = Gator('locmem://',
queue_name='hard_things',
task_class=CustomTask)
self.assertEqual(gator.conn_string, 'locmem://')
self.assertEqual(gator.queue_name, 'hard_things')
self.assertEqual(gator.task_class, CustomTask)
self.assertEqual(gator.backend_class, None)
self.assertTrue(isinstance(gator.backend, LocmemClient))
def test_build_backend(self):
backend = self.gator.build_backend('locmem://')
self.assertTrue(isinstance(backend, LocmemClient))
backend = self.gator.build_backend('redis://localhost:6379/0')
self.assertTrue(isinstance(backend, RedisClient))
def test_push_async(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task = Task(async=True)
self.gator.push(task, so_computationally_expensive, 1, 1)
self.assertEqual(self.gator.backend.len(ALL), 1)
def test_push_sync(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
def success(t, result):
t.result = result
task = Task(async=False, on_success=success)
res = self.gator.push(task, so_computationally_expensive, 1, 1)
self.assertEqual(self.gator.backend.len(ALL), 0)
self.assertEqual(res.result, 2)
def test_pop(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task = Task(async=True)
self.gator.push(task, so_computationally_expensive, 1, 1)
self.assertEqual(self.gator.backend.len(ALL), 1)
res = self.gator.pop()
self.assertEqual(res, 2)
def test_get(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task_1 = Task(async=True)
task_2 = Task(task_id='hello', async=True)
self.gator.push(task_1, so_computationally_expensive, 1, 1)
self.gator.push(task_2, so_computationally_expensive, 3, 5)
self.assertEqual(self.gator.backend.len(ALL), 2)
res = self.gator.get(task_2.task_id)
self.assertEqual(res, 8)
res = self.gator.get(task_1.task_id)
self.assertEqual(res, 2)
def test_cancel(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
task_1 = Task(async=True)
task_2 = Task(task_id='hello', async=True)
self.gator.push(task_1, so_computationally_expensive, 1, 1)
self.gator.push(task_2, so_computationally_expensive, 3, 5)
self.assertEqual(self.gator.backend.len(ALL), 2)
task = self.gator.cancel(task_2.task_id)
self.assertEqual(task.status, CANCELED)
self.assertEqual(self.gator.backend.len(ALL), 1)
def test_execute_success(self):
task = Task(retries=3, async=True)
task.to_call(so_computationally_expensive, 2, 7)
res = self.gator.execute(task)
self.assertEqual(res, 9)
self.assertEqual(task.retries, 3)
self.assertEqual(task.status, SUCCESS)
def test_execute_failed(self):
task = Task(retries=3, async=True)
task.to_call(fail_task, 2, 7)
self.assertEqual(task.status, WAITING)
try:
self.gator.execute(task)
self.assertEqual(task.status, RETRYING)
self.gator.execute(task)
self.gator.execute(task)
self.gator.execute(task)
self.fail()
except IOError:
self.assertEqual(task.retries, 0)
self.assertEqual(task.status, FAILED)
def test_execute_retries(self):
task = Task(retries=3, async=True)
task.to_call(eventual_success(), 2, 7)
try:
self.gator.execute(task)
except IOError:
pass
try:
self.gator.execute(task)
except IOError:
pass
res = self.gator.execute(task)
self.assertEqual(res, 9)
self.assertEqual(task.retries, 1)
def test_task(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
self.gator.task(so_computationally_expensive, 1, 1)
self.assertEqual(self.gator.backend.len(ALL), 1)
def test_options(self):
self.assertEqual(self.gator.backend.len(ALL), 0)
def success(t, result):
t.result = result
with self.gator.options(retries=4, async=False,
on_success=success) as opts:
res = opts.task(eventual_success(), 3, 9)
self.assertEqual(res.retries, 2)
self.assertEqual(res.result, 12)
