def test_get_registry_client(fix_reg_client):
r1 = versioning.get_registry_client(conftest.test_registry_client_name)
assert isinstance(r1, nifi.RegistryClientEntity)
assert r1.component.name == conftest.test_registry_client_name
r2 = versioning.get_registry_client(r1.id, 'id')
assert r2.id == r1.id
with pytest.raises(ValueError):
_ = versioning.get_registry_client('', 'NotIDorName')
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():
f_reg_client = versioning.get_registry_client(
conftest.test_registry_client_name)
r = versioning.delete_registry_client(f_reg_client)
assert isinstance(r, nifi.RegistryClientEntity)
assert r.uri is None
assert r.component.name == conftest.test_registry_client_name
with pytest.raises(AssertionError):
_ = versioning.delete_registry_client('FakeClient')
def test_get_registry_client(fixture_reg_client):
r1 = versioning.get_registry_client(config.test_registry_client_name)
assert isinstance(r1, nifi.RegistryClientEntity)
assert r1.component.name == config.test_registry_client_name
r2 = versioning.get_registry_client(r1.id, 'id')
assert r2.id == r1.id