def test_get_variable_registry(fix_pg): test_pg = fix_pg.generate() r1 = canvas.get_variable_registry(test_pg) assert isinstance(r1, nifi.VariableRegistryEntity) with pytest.raises(ValueError, match='Unable to locate group with id'): canvas.delete_process_group(test_pg) _ = canvas.get_variable_registry(test_pg)
def __delete_test_group(self): nav = CanvasNavigator() nav.cd_to_id(self.base.component.id) pgs = nav.groups(self.name) for pg in pgs: canvas.delete_process_group(pg, True) self.test_group = None
def test_delete_process_group(fixture_pg, regress, fixture_processor): # Delete stopped PG pg_1 = fixture_pg.generate() r1 = canvas.delete_process_group(pg_1.id, pg_1.revision) assert r1.id == pg_1.id assert r1.status is None # Test deleting a running PG pg_2 = fixture_pg.generate() p_1 = fixture_processor.generate(parent_pg=pg_2) canvas.schedule_process_group(pg_2.id, 'RUNNING') with pytest.raises(ValueError): _ = canvas.delete_process_group(pg_2.id, pg_2.revision)
def setUpClass(cls): super(Test1ToNTest, cls).setUpClass() print("Start of tests: preparing nifi objects") config.nifi_config.host = 'http://192.168.56.5:8080/nifi-api' flow_name = "Test1ToNTest" nav = CanvasNavigator() # Delete all leftovers from previous (failed?) tests pgs_to_be_deleted = nav.groups(flow_name) for pg in pgs_to_be_deleted: canvas.delete_process_group(pg, force=True) # Create new process group in root Test1ToNTest.pg_test = canvas.create_process_group(nav.current, flow_name, (0, 0)) # Create simple flow to test Test1ToNTest.proc_start = canvas.create_processor( Test1ToNTest.pg_test, canvas.get_processor_type("GenerateFlowFile"), CANVAS_CENTER, "Start") Test1ToNTest.proc_2 = canvas.create_processor( Test1ToNTest.pg_test, canvas.get_processor_type("DebugFlow"), CANVAS_CENTER, "Processor 2") Test1ToNTest.proc_3 = canvas.create_processor( Test1ToNTest.pg_test, canvas.get_processor_type("DebugFlow"), CANVAS_CENTER, "Processor 3") Test1ToNTest.proc_end_1 = canvas.create_processor( Test1ToNTest.pg_test, canvas.get_processor_type("DebugFlow"), CANVAS_CENTER, "End 1") Test1ToNTest.proc_end_2 = canvas.create_processor( Test1ToNTest.pg_test, canvas.get_processor_type("DebugFlow"), CANVAS_CENTER, "End 2") canvas.update_processor(Test1ToNTest.proc_end_1, nifi.ProcessorConfigDTO(auto_terminated_relationships=["success", "failure"])) canvas.update_processor(Test1ToNTest.proc_end_2, nifi.ProcessorConfigDTO(auto_terminated_relationships=["success", "failure"])) Test1ToNTest.conn_1 = canvas.create_connection(Test1ToNTest.proc_start, Test1ToNTest.proc_2, ["success"]) Test1ToNTest.conn_2 = canvas.create_connection(Test1ToNTest.proc_2, Test1ToNTest.proc_3, ["success", "failure"]) Test1ToNTest.conn_3 = canvas.create_connection(Test1ToNTest.proc_3, Test1ToNTest.proc_end_1, ["success", "failure"]) Test1ToNTest.conn_4 = canvas.create_connection(Test1ToNTest.proc_3, Test1ToNTest.proc_end_2, ["success", "failure"]) canvas.schedule_process_group(Test1ToNTest.pg_test.component.id, scheduled=True)
def test_delete_process_group(regress_nifi, fix_pg, fix_proc): # Delete stopped PG f_pg1 = fix_pg.generate() r1 = canvas.delete_process_group(f_pg1) assert r1.id == f_pg1.id assert r1.status is None # Test deleting a running PG pg_2 = fix_pg.generate() _ = fix_proc.generate(parent_pg=pg_2) canvas.schedule_process_group(pg_2.id, True) with pytest.raises(ValueError): _ = canvas.delete_process_group(pg_2) # Once more with feeling r2 = canvas.delete_process_group(pg_2, force=True) assert r2.status is None
def test_delete_controller(regress_nifi, fix_pg, fix_cont): f_pg = fix_pg.generate() f_c1 = fix_cont(parent_pg=f_pg) r1 = canvas.delete_controller(f_c1) assert r1.revision is None f_c2 = fix_cont(parent_pg=f_pg) f_c2 = canvas.update_controller( f_c2, nifi.ControllerServiceDTO(properties={'Server Hostname': 'Bob'})) f_c2 = canvas.schedule_controller(f_c2, True) with pytest.raises(AssertionError): _ = canvas.delete_controller('pie') with pytest.raises(AssertionError): _ = canvas.delete_controller(f_c2, 'pie') with pytest.raises(ValueError): _ = canvas.delete_controller(f_c2) assert f_c2.revision is not None r2 = canvas.delete_controller(f_c2, True) assert r2.revision is None # Test for only delete within a PG f_c_root = fix_cont() f_c_pg = fix_cont(parent_pg=f_pg) r3 = canvas.delete_process_group(f_pg) assert r3.revision is None r4 = canvas.get_controller(identifier=f_c_root.id, identifier_type='id') assert r4.revision is not None
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_delete_process_group(fixture_pg, regress): single_pg = fixture_pg.generate() r = canvas.delete_process_group(single_pg.id, single_pg.revision) assert r.id == single_pg.id assert r.status is None
def tearDownClass(cls): print("Tests done: deleting nifi objects") canvas.delete_controller( canvas.get_controller(CanvasNavigatorTest.controller.component.id, 'id'), True) canvas.delete_process_group(CanvasNavigatorTest.pg_parent, True)
def tearDownClass(cls): print("Tests done: deleting nifi objects") canvas.delete_process_group(Test1To1Test.pg_test, force=True)