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)
