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