def test_get_processor(regress_nifi, fix_proc):
f_p1 = fix_proc.generate()
r1 = canvas.get_processor(f_p1.status.name)
assert isinstance(r1, nifi.ProcessorEntity)
r2 = canvas.get_processor('ClearlyNotAProcessor')
assert r2 is None
f_p2 = fix_proc.generate()
r3 = canvas.get_processor(f_p1.status.name)
assert isinstance(r3, list)
r4 = canvas.get_processor(f_p2.id, 'id')
assert isinstance(r4, nifi.ProcessorEntity)
assert r4.id != r1.id
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')
