Beispiel #1
0
def test_reset_all_states(simple_plan):
    for n in simple_plan:
        n.status == states.ERROR.name
    graph.reset(simple_plan)

    for n in simple_plan:
        assert n.status == states.PENDING.name
Beispiel #2
0
def test_reset_resets_times():
    g = nx.MultiDiGraph()
    g.add_node('task1', task=mock.Mock(), status='status', errmsg='',
               start_time=1, end_time=4)
    graph.reset(g)
    assert int(g.node['task1']['start_time']) == 0
    assert int(g.node['task1']['start_time']) == 0
Beispiel #3
0
def test_reset_all_states(simple_plan):
    for n in simple_plan:
        simple_plan.node[n]['status'] == states.ERROR.name
    graph.reset(simple_plan)

    for n in simple_plan:
        assert simple_plan.node[n]['status'] == states.PENDING.name
Beispiel #4
0
def test_reset_all_states(simple):
    for n in simple:
        simple.node[n]["status"] == states.ERROR.name
    graph.reset(simple)

    for n in simple:
        assert simple.node[n]["status"] == states.PENDING.name
Beispiel #5
0
def test_reset_only_provided(simple_plan):
    simple_plan.node['just_fail']['status'] = states.ERROR.name
    simple_plan.node['echo_stuff']['status'] = states.SUCCESS.name

    graph.reset(simple_plan, [states.ERROR.name])

    assert simple_plan.node['just_fail']['status'] == states.PENDING.name
    assert simple_plan.node['echo_stuff']['status'] == states.SUCCESS.name
Beispiel #6
0
def test_reset_only_provided(simple):
    simple.node["just_fail"]["status"] = states.ERROR.name
    simple.node["echo_stuff"]["status"] = states.SUCCESS.name

    graph.reset(simple, [states.ERROR.name])

    assert simple.node["just_fail"]["status"] == states.PENDING.name
    assert simple.node["echo_stuff"]["status"] == states.SUCCESS.name
Beispiel #7
0
def test_reset_resets_times():
    g = nx.MultiDiGraph()
    g.add_node('task1',
               task=mock.Mock(),
               status='status',
               errmsg='',
               start_time=1,
               end_time=4)
    graph.reset(g)
    assert int(g.node['task1']['start_time']) == 0
    assert int(g.node['task1']['start_time']) == 0
Beispiel #8
0
def test_reset_resets_times():
    g = nx.MultiDiGraph()
    task = mock.Mock(
        name='task1',
        status='status',
        errmsg='',
        start_time=1, end_time=4)
    g.add_node(task)
    graph.reset(g)
    for n in g.nodes():
        assert n.start_time == 0
Beispiel #9
0
def test_reset_only_provided(simple_plan):
    for n in simple_plan.nodes():
        if n.name == 'just_fail':
            n.status = states.ERROR.name
        elif n.name == 'echo_stuff':
            n.status = states.SUCCESS.name

    graph.reset(simple_plan, [states.ERROR.name])

    for n in simple_plan.nodes():
        if n.name == 'just_fail':
            assert n.status == states.PENDING.name
        elif n.name == 'echo_stuff':
            assert n.status == states.SUCCESS.name
Beispiel #10
0
def reset(uid):
    graph.reset(uid)
Beispiel #11
0
def restart(uid):
    graph.reset(uid)
    tasks.schedule_start.apply_async(args=[uid], queue='scheduler')
Beispiel #12
0
def retry(uid):
    graph.reset(uid, ['ERROR'])
    tasks.schedule_start.apply_async(args=[uid], queue='scheduler')
Beispiel #13
0
def resume(uid):
    graph.reset(uid, ['SKIPPED'])
    tasks.schedule_start.apply_async(args=[uid], queue='scheduler')