def test_create_request_handler(self):
nav = CanvasNavigator()
loc = Location()
name = "nipytest - unit test - test_create_request_handler"
controller = canvas_ext.create_http_context_map(nav.current, name)
# Testing incorrect inputs
with self.assertRaises(AssertionError):
canvas_ext.create_request_handler("id", loc, controller, PORT)
# noinspection PyTypeChecker
canvas_ext.create_request_handler(nav.current, 1, controller, PORT)
canvas_ext.create_request_handler(nav.current, loc, 1, PORT)
# noinspection PyTypeChecker
canvas_ext.create_request_handler(nav.current, loc, controller,
"string")
# Run function
request_handler = canvas_ext.create_request_handler(
nav.current, loc, controller, PORT)
# If output is ok type, function was successful
self.assertIsInstance(request_handler, nifi.ProcessorEntity)
self.assertEqual(
request_handler.component.config.properties["Listening Port"],
str(PORT))
# Remove temporary created object(s)
canvas.delete_processor(request_handler, force=True)
canvas.delete_controller(controller, True)
def test_delete_processor(fixture_processor, regress):
test_proc = fixture_processor.generate()
assert test_proc.status.name == config.test_processor_name
# try to delete running processor
canvas.schedule_processor(test_proc, 'RUNNING')
with pytest.raises(ValueError):
_ = canvas.delete_processor(test_proc)
canvas.schedule_processor(test_proc, 'STOPPED')
r = canvas.delete_processor(test_proc)
assert r.status is None
# try to delete twice
with pytest.raises(ValueError):
_ = canvas.delete_processor(test_proc)
def test_delete_processor(regress_nifi, fix_proc):
f_p1 = fix_proc.generate()
r1 = canvas.delete_processor(f_p1)
assert r1.status is None
assert isinstance(r1, nifi.ProcessorEntity)
# try to delete processor twice
with pytest.raises(ValueError):
_ = canvas.delete_processor(f_p1)
# try to delete running processor
f_p2 = fix_proc.generate()
canvas.schedule_processor(f_p2, True)
with pytest.raises(ValueError):
_ = canvas.delete_processor(f_p2)
# and once more with feeling, er, force
r2 = canvas.delete_processor(f_p2, force=True)
assert r2.status is None
def test_create_output_router(self):
nav = CanvasNavigator()
loc = Location()
name = "nipytest - unit test - test_create_output_router"
# Testing incorrect inputs
with self.assertRaises(AssertionError):
canvas_ext.create_output_router("id", loc, name)
# noinspection PyTypeChecker
canvas_ext.create_output_router(nav.current, 1, name)
# noinspection PyTypeChecker
canvas_ext.create_output_router(nav.current, loc, 1)
# Run function
output_router = canvas_ext.create_output_router(nav.current, loc, name)
# If output is ok type, function was successful
self.assertIsInstance(output_router, nifi.ProcessorEntity)
self.assertIsNotNone(output_router.component.name, name)
# Remove temporary created object(s)
canvas.delete_processor(output_router, force=True)
def test_create_output_attribute(self):
nav = CanvasNavigator()
loc = Location()
name = "nipytest - unit test - test_create_output_attribute"
# Testing incorrect inputs
with self.assertRaises(AssertionError):
canvas_ext.create_output_attribute("id", loc, name)
# noinspection PyTypeChecker
canvas_ext.create_output_attribute(nav.current, 1, name)
# noinspection PyTypeChecker
canvas_ext.create_output_attribute(nav.current, loc, 1)
# Run function
output_attribute = canvas_ext.create_output_attribute(
nav.current, loc, name)
# If output is ok type, function was successful
self.assertIsInstance(output_attribute, nifi.ProcessorEntity)
self.assertEqual(output_attribute.component.name,
"Set output name '" + name + "'")
# Remove temporary created object(s)
canvas.delete_processor(output_attribute)
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)
