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