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)
