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