def test_queue_up(self):
inputs = {'a': Int(2), 'b': Str('test')}
# Queue up the process
pid = queue_up(DummyProcess, inputs, self.storage)
# Then load the checkpoint and instantiate the class
cp = self.storage.load_checkpoint(pid)
dp = DummyProcess.create_from(cp)
self.assertIsInstance(dp, DummyProcess)
self.assertEqual(dp.raw_inputs, inputs)
dp.run_until_complete()
def test_seal(self):
rinfo = run(DummyProcess, _return_pid=True)[1]
self.assertTrue(load_node(pk=rinfo).is_sealed)
storedir = tempfile.mkdtemp()
storage = Persistence.create_from_basedir(storedir)
rinfo = submit(DummyProcess, _jobs_store=storage)
n = load_node(pk=rinfo.pid)
self.assertFalse(n.is_sealed)
dp = DummyProcess.create_from(storage.load_checkpoint(rinfo.pid))
dp.run_until_complete()
self.assertTrue(n.is_sealed)
shutil.rmtree(storedir)