def test_pickle_with_badid(tmpdir):
tsk = Task(str, tid="/tmp/foobar")
tsk.pickle(str(tmpdir))
assert 1 == len(tmpdir.listdir())
assert tmpdir.listdir()[0].isfile()
utsk = Task.unpickle(tsk.tid, str(tmpdir))
assert utsk.tid == tsk.tid
def get_task(self, tid):
if tid not in self.tasks:
log.debug("Missing task %s, checking filesystem.", tid)
task = Task.unpickle(tid, self.work_dir)
self._enqueue(task)
return self.tasks[tid]
def test_unpickle_missing(tmpdir):
with pytest.raises(NoSuchTaskError):
Task.unpickle('asdf', str(tmpdir))
def test_task_pickle_unpickle_id(tmpdir, idtask):
tid = idtask.pickle(str(tmpdir))
tsk = Task.unpickle(tid, str(tmpdir))
assert 1 == len(tmpdir.listdir())
assert tid == tsk.tid
assert '' == tsk.run()
