def test_schedule_processor(fix_proc):
f_p1 = fix_proc.generate()
r1 = canvas.schedule_processor(f_p1, True)
status = canvas.get_processor(f_p1.id, 'id')
assert r1 is True
assert status.status.run_status == 'Running'
r2 = canvas.schedule_processor(f_p1, False)
status = canvas.get_processor(f_p1.id, 'id')
assert status.status.run_status == 'Stopped'
assert r2 is True
with pytest.raises(AssertionError):
_ = canvas.schedule_processor(f_p1, 'BANANA')
def test_schedule_processor(regress_nifi, fix_proc):
f_p1 = fix_proc.generate()
r1 = canvas.schedule_processor(f_p1, True)
processor_info = canvas.get_processor(f_p1.id, 'id')
assert r1 is True
assert isinstance(processor_info, nifi.ProcessorEntity)
assert processor_info.component.state == 'RUNNING'
r2 = canvas.schedule_processor(f_p1, False)
status = canvas.get_processor(f_p1.id, 'id')
assert status.component.state == 'STOPPED'
assert r2 is True
with pytest.raises(AssertionError):
_ = canvas.schedule_processor(f_p1, 'BANANA')
def test_delete_processor(fixture_processor, regress):
test_proc = fixture_processor.generate()
assert test_proc.status.name == config.test_processor_name
# try to delete running processor
canvas.schedule_processor(test_proc, 'RUNNING')
with pytest.raises(ValueError):
_ = canvas.delete_processor(test_proc)
canvas.schedule_processor(test_proc, 'STOPPED')
r = canvas.delete_processor(test_proc)
assert r.status is None
# try to delete twice
with pytest.raises(ValueError):
_ = canvas.delete_processor(test_proc)
def lab1_sensor_simulator(self):
# Create a processor to run the sensor simulator
gen_data = nf.create_processor(
self.context.root_pg, 'Generate Test Data',
'org.apache.nifi.processors.standard.ExecuteProcess', (0, 0), {
'properties': {
'Command': 'python3',
'Command Arguments': '/opt/demo/simulate.py',
},
'schedulingPeriod': '1 sec',
'schedulingStrategy': 'TIMER_DRIVEN',
'autoTerminatedRelationships': ['success'],
})
canvas.schedule_processor(gen_data, True)
def test_delete_processor(regress_nifi, fix_proc):
f_p1 = fix_proc.generate()
r1 = canvas.delete_processor(f_p1)
assert r1.status is None
assert isinstance(r1, nifi.ProcessorEntity)
# try to delete processor twice
with pytest.raises(ValueError):
_ = canvas.delete_processor(f_p1)
# try to delete running processor
f_p2 = fix_proc.generate()
canvas.schedule_processor(f_p2, True)
with pytest.raises(ValueError):
_ = canvas.delete_processor(f_p2)
# and once more with feeling, er, force
r2 = canvas.delete_processor(f_p2, force=True)
assert r2.status is None
root_pg = canvas.get_process_group(root_pg_id, 'id')
location_x = 2000
location_y = 2000
location = (location_x, location_y)
template_entity = nipyapi.templates.upload_template(root_pg_id,
'jangowave_demo.xml')
flow = nipyapi.templates.deploy_template(root_pg_id, template_entity.id, 2000,
2000)
jd = canvas.get_process_group('jangowave_demo')
for cs in canvas.list_all_processors(jd.id):
if cs.status.run_status != "ENABLED" and cs.component.name == "RecordIngest":
config_update = nifi.models.processor_config_dto.ProcessorConfigDTO(
properties={
"Accumulo User": "******",
"Instance Name": instance,
"Accumulo Password": "******",
"ZooKeeper Quorum": zookeeper_list
})
canvas.update_processor(cs, config_update)
for cs in canvas.list_all_controllers(jd.id):
canvas.schedule_controller(cs, True)
for cs in canvas.list_all_processors(jd.id):
canvas.schedule_processor(cs, True)
canvas.schedule_process_group(jd.id, True)
