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