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 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 create_controller(pg, controller_type, properties, start, name=None): controller_type = get_controller_type(controller_type) controller = canvas.create_controller(pg, controller_type, name) controller = canvas.get_controller(controller.id, 'id') canvas.update_controller(controller, nifi.ControllerServiceDTO(properties=properties)) controller = canvas.get_controller(controller.id, 'id') canvas.schedule_controller(controller, start) return canvas.get_controller(controller.id, 'id')