def setUpClass(cls): super(CanvasNavigatorTest, cls).setUpClass() print("Start of tests: preparing nifi objects") config.nifi_config.host = 'http://192.168.56.5:8080/nifi-api' root = canvas.get_process_group(canvas.get_root_pg_id(), 'id') # Create new process group in root CanvasNavigatorTest.pg_parent = canvas.create_process_group( root, "parent", CANVAS_CENTER) # Create new child process groups in unittest CanvasNavigatorTest.pg_child1 = canvas.create_process_group( CanvasNavigatorTest.pg_parent, "child1", CANVAS_CENTER) CanvasNavigatorTest.pg_child2 = canvas.create_process_group( CanvasNavigatorTest.pg_parent, "child2", CANVAS_CENTER) CanvasNavigatorTest.pg_child2_2 = canvas.create_process_group( CanvasNavigatorTest.pg_parent, "child2", CANVAS_CENTER) CanvasNavigatorTest.pg_grandchild1 = canvas.create_process_group( CanvasNavigatorTest.pg_child1, "grandchild1", CANVAS_CENTER) # Create other objects as well CanvasNavigatorTest.proc = canvas.create_processor( CanvasNavigatorTest.pg_parent, canvas.get_processor_type("GenerateFlowFile"), CANVAS_CENTER, "proc") CanvasNavigatorTest.input_port = canvas.create_port( CanvasNavigatorTest.pg_parent.component.id, "INPUT_PORT", "input_port", "STOPPED", CANVAS_CENTER) CanvasNavigatorTest.output_port = canvas.create_port( CanvasNavigatorTest.pg_parent.component.id, "OUTPUT_PORT", "output_port", "STOPPED", CANVAS_CENTER) CanvasNavigatorTest.controller = canvas.create_controller( CanvasNavigatorTest.pg_parent, canvas.list_all_controller_types()[0], "controller")
def get_controller_type(controller_type): types = [ ctype for ctype in canvas.list_all_controller_types() if ctype.type == controller_type ] if types: return types[0] return None
def test_create_controller(regress_nifi, fix_cont): root_pg = canvas.get_process_group(canvas.get_root_pg_id(), 'id') cont_type = canvas.list_all_controller_types()[0] r1 = canvas.create_controller(parent_pg=root_pg, controller=cont_type) assert isinstance(r1, nifi.ControllerServiceEntity) with pytest.raises(AssertionError): _ = canvas.create_controller('pie', cont_type) with pytest.raises(AssertionError): _ = canvas.create_controller(root_pg, 'pie')
def test_list_all_controller_types(regress_nifi): r1 = canvas.list_all_controller_types() assert len(r1) > 5 assert isinstance(r1[0], nifi.DocumentedTypeDTO)