def test_soft_stop(simple_plan):
uid = simple_plan.graph["uid"]
scheduler = Scheduler(None)
scheduler.soft_stop({}, uid)
plan = graph.get_graph(uid)
for n in plan:
assert n.status == states.SKIPPED.name
def test_soft_stop(simple_plan):
# graph.save_graph(simple_plan)
uid = simple_plan.graph['uid']
scheduler = Scheduler(None)
scheduler.soft_stop({}, uid)
plan = graph.get_graph(uid)
for n in plan:
assert plan.node[n]['status'] == states.SKIPPED.name
def test_scheduler_next_fails_with_empty_plan():
scheduler = Scheduler(None)
with pytest.raises(ValueError):
scheduler.next({}, "nonexistent_uid")
def test_scheduler_next_fails_with_empty_plan():
scheduler = Scheduler(None)
with pytest.raises(ValueError):
scheduler.next({}, 'nonexistent_uid')