def test_list_all_connections(regress_nifi, fix_proc): f_p1 = fix_proc.generate() f_p2 = fix_proc.generate() r1 = [ x for x in canvas.list_all_connections() if conftest.test_basename in x.component.name ] assert not r1 # connect single relationship c1 = canvas.create_connection(f_p1, f_p2, ['success'], conftest.test_basename) r2 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert len(r2) == 1 r3 = [ x for x in canvas.list_all_connections(canvas.get_root_pg_id()) if conftest.test_basename in x.component.name ] assert len(r3) == 1 assert isinstance(r2[0], nifi.ConnectionEntity) c2 = canvas.create_connection(f_p1, f_p2, name=conftest.test_basename) r2 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert len(r2) == 2 _ = canvas.delete_connection(c1) _ = canvas.delete_connection(c2) r3 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert not r3
def test_list_all_connections(regress_nifi, fix_pg, fix_proc): f_p1 = fix_proc.generate() f_p2 = fix_proc.generate() r1 = [ x for x in canvas.list_all_connections() if conftest.test_basename in x.component.name ] assert not r1 # connect single relationship c1 = canvas.create_connection(f_p1, f_p2, ['success'], conftest.test_basename) r2 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert len(r2) == 1 r3 = [ x for x in canvas.list_all_connections(canvas.get_root_pg_id()) if conftest.test_basename in x.component.name ] assert len(r3) == 1 assert isinstance(r2[0], nifi.ConnectionEntity) c2 = canvas.create_connection(f_p1, f_p2, name=conftest.test_basename) r2 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert len(r2) == 2 _ = canvas.delete_connection(c1) _ = canvas.delete_connection(c2) r4 = [ x for x in canvas.list_all_connections('root') if conftest.test_basename in x.component.name ] assert not r4 # Test Issue #129 - nested PGs with descendents missing nested content f_pg1 = fix_pg.generate() f_pg2 = fix_pg.generate(parent_pg=f_pg1) f_p3 = fix_proc.generate(parent_pg=f_pg2) f_p4 = fix_proc.generate(parent_pg=f_pg2) c2 = canvas.create_connection(f_p3, f_p4, ['success'], conftest.test_basename) r5 = [ x for x in canvas.list_all_connections(f_pg2.id) if conftest.test_basename in x.component.name ] assert len(r5) == 1 assert r5[0].id == c2.id
def delete_all_connections(process_group, purge=True, descendants=True): """ Purges and deletes all connections inside a process group Args: process_group (ProcessGroupEntity): Process group where connections should be purged and deleted purge (bool): True to Purge, Defaults to True descendants (bool): True to recurse child PGs, False to not """ assert isinstance(process_group, nifi.ProcessGroupEntity) connections = canvas.list_all_connections(process_group.component.id, descendants=descendants) print("Listing connections in " + process_group.component.id) print(connections) for connection in connections: canvas.delete_connection(connection, purge=purge)
def test_delete_connection(regress_nifi, fix_proc): f_p1 = fix_proc.generate() f_p2 = fix_proc.generate() # connect single relationship c1 = canvas.create_connection(f_p1, f_p2, ['success'], conftest.test_basename) r1 = canvas.delete_connection(c1) assert isinstance(r1, nifi.ConnectionEntity) assert r1.status is None
def test_recreate_connection(self): nav = CanvasNavigator() loc = Location() name = "nipytest - unit test - test_recreate_connection" # Testing incorrect inputs with self.assertRaises(AssertionError): canvas_ext.recreate_connection(1) # Create connection input_port = canvas_ext.create_test_output(nav.current, loc, name) output_port = canvas_ext.create_test_input(nav.current, loc, name) connection = canvas.create_connection(input_port, output_port) canvas.delete_connection(connection) # Run function new_connection = canvas_ext.recreate_connection(connection) # If output is ok type, function was successful self.assertIsInstance(new_connection, nifi.ConnectionEntity) self.assertNotEqual(connection.component.id, new_connection.component.id) # Remove temporary created object(s) canvas.delete_connection(new_connection) canvas.delete_port(input_port) canvas.delete_port(output_port)
def nifi_delete_all(pg): canvas.schedule_process_group(pg.id, False) for conn in canvas.list_all_connections(pg.id): LOG.debug('Connection: ' + conn.id) canvas.delete_connection(conn, purge=True) for input_port in canvas.list_all_input_ports(pg.id): LOG.debug('Input Port: ' + input_port.id) canvas.delete_port(input_port) for output_port in canvas.list_all_output_ports(pg.id): LOG.debug('Output Port: ' + output_port.id) canvas.delete_port(output_port) for funnel in canvas.list_all_funnels(pg.id): LOG.debug('Funnel: ' + funnel.id) canvas.delete_funnel(funnel) for processor in canvas.list_all_processors(pg.id): LOG.debug('Processor: ' + processor.id) canvas.delete_processor(processor, force=True) for process_group in canvas.list_all_process_groups(pg.id): if pg.id == process_group.id: continue LOG.debug('Process Group: ' + process_group.id) nifi_delete_all(process_group) canvas.delete_process_group(process_group, force=True)
def __remove_outgoing_connections(self): for connection in self.connections_to_remove: canvas.delete_connection(connection, purge=True)