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')
