def teardown(self): root_pg, _, flow_id = nf.set_environment() canvas.schedule_process_group(root_pg.id, False) while True: failed = False for controller in canvas.list_all_controllers(root_pg.id): try: canvas.schedule_controller(controller, False) LOG.debug('Controller %s stopped.', controller.component.name) except ApiException as exc: if exc.status == 409 and 'is referenced by' in exc.body: LOG.debug( 'Controller %s failed to stop. Will retry later.', controller.component.name) failed = True if not failed: break nf.delete_all(root_pg) efm.delete_all(flow_id) schreg.delete_all_schemas() reg_client = versioning.get_registry_client('NiFi Registry') if reg_client: versioning.delete_registry_client(reg_client) nifireg.delete_flows('SensorFlows') kudu.drop_table()
def test_delete_registry_client(fix_reg_client): r = versioning.delete_registry_client(fix_reg_client) assert isinstance(r, nifi.RegistryClientEntity) assert r.uri is None assert r.component.name == conftest.test_registry_client_name with pytest.raises(ValueError): _ = versioning.delete_registry_client('FakeClient')
def test_create_registry_client(): # First remove any leftover test client connections [ versioning.delete_registry_client(li) for li in versioning.list_registry_clients().registries if config.test_registry_client_name in li.component.name ] r = versioning.create_registry_client( name=config.test_registry_client_name, uri=config.test_docker_registry_endpoint, description='a test connection') assert isinstance(r, nifi.RegistryClientEntity) # test duplicate catch result with pytest.raises(ValueError): _ = versioning.create_registry_client( name=config.test_registry_client_name, uri=config.test_docker_registry_endpoint, description='who cares?')
def test_delete_registry_client(fixture_reg_client): r = versioning.delete_registry_client(fixture_reg_client) assert isinstance(r, nifi.RegistryClientEntity) assert r.uri is None assert r.component.name == config.test_registry_client_name