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_forget(self): tb = DatabaseBackend(backend="memory://") tid = uuid() tb.mark_as_done(tid, {"foo": "bar"}) x = AsyncResult(tid) x.forget() self.assertIsNone(x.result)
def test_forget(self): tb = DatabaseBackend(backend='memory://') tid = uuid() tb.mark_as_done(tid, {'foo': 'bar'}) tb.mark_as_done(tid, {'foo': 'bar'}) x = AsyncResult(tid, backend=tb) x.forget() self.assertIsNone(x.result)
def test_forget(self): tb = DatabaseBackend(self.uri, backend='memory://', app=self.app) tid = uuid() tb.mark_as_done(tid, {'foo': 'bar'}) tb.mark_as_done(tid, {'foo': 'bar'}) x = self.app.AsyncResult(tid, backend=tb) x.forget() assert x.result is None
def test_forget(self): tb = DatabaseBackend(self.uri, backend="memory://", app=self.app) tid = uuid() tb.mark_as_done(tid, {"foo": "bar"}) tb.mark_as_done(tid, {"foo": "bar"}) x = self.app.AsyncResult(tid, backend=tb) x.forget() self.assertIsNone(x.result)
def test_is_pickled(self): tb = DatabaseBackend() tid2 = uuid() result = {'foo': 'baz', 'bar': SomeClass(12345)} tb.mark_as_done(tid2, result) # is serialized properly. rindb = tb.get_result(tid2) self.assertEqual(rindb.get('foo'), 'baz') self.assertEqual(rindb.get('bar').data, 12345)
def test_is_pickled(self): tb = DatabaseBackend() tid2 = uuid() result = {"foo": "baz", "bar": SomeClass(12345)} tb.mark_as_done(tid2, result) # is serialized properly. rindb = tb.get_result(tid2) self.assertEqual(rindb.get("foo"), "baz") self.assertEqual(rindb.get("bar").data, 12345)
def test_is_pickled(self): tb = DatabaseBackend(self.uri, app=self.app) tid2 = uuid() result = {'foo': 'baz', 'bar': SomeClass(12345)} tb.mark_as_done(tid2, result) # is serialized properly. rindb = tb.get_result(tid2) assert rindb.get('foo') == 'baz' assert rindb.get('bar').data == 12345
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(self.uri, app=self.app) tid = uuid() assert tb.get_state(tid) == states.PENDING assert tb.get_result(tid) is None tb.mark_as_done(tid, 42) assert tb.get_state(tid) == states.SUCCESS assert tb.get_result(tid) == 42
def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(uuid(), 42) tb.save_group(uuid(), {'foo': 'bar'}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.expires * 2 s.commit() s.close() tb.cleanup()
def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(uuid(), 42) tb.save_taskset(uuid(), {"foo": "bar"}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.expires * 2 s.commit() s.close() tb.cleanup()
def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(gen_unique_id(), 42) tb.save_taskset(gen_unique_id(), {"foo": "bar"}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.result_expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.result_expires * 2 s.commit() s.close() tb.cleanup()
def test_cleanup(self): tb = DatabaseBackend(self.uri, app=self.app) for i in range(10): tb.mark_as_done(uuid(), 42) tb.save_group(uuid(), {'foo': 'bar'}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.expires * 2 s.commit() s.close() tb.cleanup()
def test_cleanup(self): tb = DatabaseBackend() for i in range(10): tb.mark_as_done(gen_unique_id(), 42) tb.save_taskset(gen_unique_id(), {"foo": "bar"}) s = tb.ResultSession() for t in s.query(Task).all(): t.date_done = datetime.now() - tb.result_expires * 2 for t in s.query(TaskSet).all(): t.date_done = datetime.now() - tb.result_expires * 2 s.commit() s.close() tb.cleanup() s2 = tb.ResultSession() self.assertEqual(s2.query(Task).count(), 0) self.assertEqual(s2.query(TaskSet).count(), 0)