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_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(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_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()
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_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()
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_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)
