def test_backend(self):
b = DatabaseBackend()
tid = gen_unique_id()
self.assertFalse(b.is_successful(tid))
self.assertEquals(b.get_status(tid), states.PENDING)
self.assertTrue(b.get_result(tid) is None)
b.mark_as_done(tid, 42)
self.assertTrue(b.is_successful(tid))
self.assertEquals(b.get_status(tid), states.SUCCESS)
self.assertEquals(b.get_result(tid), 42)
self.assertTrue(b._cache.get(tid))
self.assertTrue(b.get_result(tid), 42)
tid2 = gen_unique_id()
result = {"foo": "baz", "bar": SomeClass(12345)}
b.mark_as_done(tid2, result)
# is serialized properly.
rindb = b.get_result(tid2)
self.assertEquals(rindb.get("foo"), "baz")
self.assertEquals(rindb.get("bar").data, 12345)
tid3 = gen_unique_id()
try:
raise KeyError("foo")
except KeyError, exception:
pass
def test_mark_as_done(self):
tb = DatabaseBackend()
tid = uuid()
self.assertEqual(tb.get_status(tid), states.PENDING)
self.assertIsNone(tb.get_result(tid))
tb.mark_as_done(tid, 42)
self.assertEqual(tb.get_status(tid), states.SUCCESS)
self.assertEqual(tb.get_result(tid), 42)
def test_mark_as_done(self):
tb = DatabaseBackend()
tid = uuid()
self.assertEqual(tb.get_status(tid), states.PENDING)
self.assertIsNone(tb.get_result(tid))
tb.mark_as_done(tid, 42)
self.assertEqual(tb.get_status(tid), states.SUCCESS)
self.assertEqual(tb.get_result(tid), 42)
def test_mark_as_retry(self):
tb = DatabaseBackend(self.uri, app=self.app)
tid = uuid()
try:
raise KeyError('foo')
except KeyError as exception:
import traceback
trace = '\n'.join(traceback.format_stack())
tb.mark_as_retry(tid, exception, traceback=trace)
self.assertEqual(tb.get_status(tid), states.RETRY)
self.assertIsInstance(tb.get_result(tid), KeyError)
self.assertEqual(tb.get_traceback(tid), trace)
def test_mark_as_retry(self):
tb = DatabaseBackend()
tid = uuid()
try:
raise KeyError('foo')
except KeyError as exception:
import traceback
trace = '\n'.join(traceback.format_stack())
tb.mark_as_retry(tid, exception, traceback=trace)
self.assertEqual(tb.get_status(tid), states.RETRY)
self.assertIsInstance(tb.get_result(tid), KeyError)
self.assertEqual(tb.get_traceback(tid), trace)
def test_mark_as_failure(self):
tb = DatabaseBackend()
tid3 = uuid()
try:
raise KeyError('foo')
except KeyError as exception:
import traceback
trace = '\n'.join(traceback.format_stack())
tb.mark_as_failure(tid3, exception, traceback=trace)
self.assertEqual(tb.get_status(tid3), states.FAILURE)
self.assertIsInstance(tb.get_result(tid3), KeyError)
self.assertEqual(tb.get_traceback(tid3), trace)
def test_mark_as_failure(self):
tb = DatabaseBackend(self.uri, app=self.app)
tid3 = uuid()
try:
raise KeyError('foo')
except KeyError as exception:
import traceback
trace = '\n'.join(traceback.format_stack())
tb.mark_as_failure(tid3, exception, traceback=trace)
self.assertEqual(tb.get_status(tid3), states.FAILURE)
self.assertIsInstance(tb.get_result(tid3), KeyError)
self.assertEqual(tb.get_traceback(tid3), trace)
def test_missing_task_id_is_PENDING(self):
tb = DatabaseBackend()
self.assertEqual(tb.get_status('xxx-does-not-exist'), states.PENDING)
def test_mark_as_revoked(self):
tb = DatabaseBackend()
tid = uuid()
tb.mark_as_revoked(tid)
self.assertEqual(tb.get_status(tid), states.REVOKED)
def test_mark_as_started(self):
tb = DatabaseBackend()
tid = uuid()
tb.mark_as_started(tid)
self.assertEqual(tb.get_status(tid), states.STARTED)
def test_missing_task_id_is_PENDING(self):
tb = DatabaseBackend()
self.assertEqual(tb.get_status("xxx-does-not-exist"), states.PENDING)
def test_mark_as_revoked(self):
tb = DatabaseBackend()
tid = uuid()
tb.mark_as_revoked(tid)
self.assertEqual(tb.get_status(tid), states.REVOKED)
def test_mark_as_started(self):
tb = DatabaseBackend()
tid = uuid()
tb.mark_as_started(tid)
self.assertEqual(tb.get_status(tid), states.STARTED)
def test_missing_task_id_is_PENDING(self):
tb = DatabaseBackend(self.uri, app=self.app)
self.assertEqual(tb.get_status('xxx-does-not-exist'), states.PENDING)
