def test_node_control_barrier(get_node_id):
    get_node_id.side_effect = Exception("")  # return_value = None
    nodecontrol.node_control(CONTROL_URI, barrier=False)
    assert not get_node_id.called

    with pytest.raises(AssertionError):
        nodecontrol.node_control(CONTROL_URI, barrier=True)
    assert get_node_id.call_count == 20
    get_node_id.reset_mock()

    get_node_id.side_effect = None
    get_node_id.return_value = 1
    nodecontrol.node_control(CONTROL_URI, barrier=True)
    assert get_node_id.call_count == 1
def runtime(uri, control_uri, start_new):
    if start_new:
        rt = dispatch_node(uri=uri, control_uri=control_uri)
    else:
        rt = node_control(control_uri)
    return rt
Exemple #3
0
def runtime(uri, control_uri, start_new):
    if start_new:
        rt = dispatch_node(uri=uri, control_uri=control_uri)
    else:
        rt = node_control(control_uri)
    return rt