def test_create_test_input(self):
nav = CanvasNavigator()
loc = Location()
name = "nipytest - unit test - test_create_test_input"
# Testing incorrect inputs
with self.assertRaises(AssertionError):
canvas_ext.create_test_input("id", loc, name)
# noinspection PyTypeChecker
canvas_ext.create_test_input(nav.current, 1, name)
# noinspection PyTypeChecker
canvas_ext.create_test_input(nav.current, loc, 1)
# Run function
output_port = canvas_ext.create_test_input(nav.current, loc, name)
# If output is ok type, function was successful
self.assertIsInstance(output_port, nifi.PortEntity)
self.assertEqual(output_port.component.name, name)
# Remove temporary created object(s)
canvas.delete_port(output_port)
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 test_create_input_router(self):
nav = CanvasNavigator()
loc = Location()
name = "nipytest - unit test - test_create_input_router"
# Testing incorrect inputs
with self.assertRaises(AssertionError):
canvas_ext.create_input_router("id", loc, [])
# noinspection PyTypeChecker
canvas_ext.create_input_router(nav.current, 1, [])
# canvas_ext.create_input_router(nav.current, loc, 1)
# Run function
output_port = canvas_ext.create_test_input(nav.current, loc, name)
input_router = canvas_ext.create_input_router(nav.current, loc,
[output_port])
# If output is ok type, function was successful
self.assertIsInstance(input_router, nifi.ProcessorEntity)
self.assertIsNotNone(input_router.component.config.properties[name])
# Remove temporary created object(s)
canvas.delete_processor(input_router, force=True)
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 test_input_output_ports(regress_nifi, fix_pg):
root_input_port = canvas.create_port(pg_id=canvas.get_root_pg_id(),
port_type='INPUT_PORT',
name=conftest.test_basename +
'input_port',
state='STOPPED')
assert isinstance(root_input_port, nifi.PortEntity)
root_output_port = canvas.create_port(pg_id=canvas.get_root_pg_id(),
port_type='OUTPUT_PORT',
name=conftest.test_basename +
'output_port',
state='STOPPED')
assert isinstance(root_output_port, nifi.PortEntity)
input_ports = [
x for x in canvas.list_all_by_kind('input_ports')
if conftest.test_basename in x.status.name
]
assert len(input_ports) == 1
output_ports = [
x for x in canvas.list_all_by_kind('output_ports')
if conftest.test_basename in x.status.name
]
assert len(output_ports) == 1
f_pg = fix_pg.generate()
f_pg_input_port = canvas.create_port(pg_id=f_pg.id,
port_type='INPUT_PORT',
name=conftest.test_basename +
'input_port',
state='STOPPED')
assert isinstance(f_pg_input_port, nifi.PortEntity)
f_pg_output_port = canvas.create_port(pg_id=f_pg.id,
port_type='OUTPUT_PORT',
name=conftest.test_basename +
'output_port',
state='STOPPED')
assert isinstance(f_pg_output_port, nifi.PortEntity)
input_ports = [
x for x in canvas.list_all_by_kind('input_ports')
if conftest.test_basename in x.status.name
]
assert len(input_ports) == 2
output_ports = [
x for x in canvas.list_all_by_kind('output_ports')
if conftest.test_basename in x.status.name
]
assert len(output_ports) == 2
d1 = canvas.delete_port(root_input_port)
assert isinstance(d1, nifi.PortEntity)
assert d1.status is None
