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_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_missing_task_id_is_PENDING(self): tb = DatabaseBackend(self.uri, app=self.app) self.assertEqual(tb.get_status('xxx-does-not-exist'), states.PENDING)