def test_on_render_hook_is_executed(tmp_directory):
hook.count = 0
dag = DAG()
PythonCallable(touch_root, File('file.txt'), dag, name='t')
dag.on_render = hook
dag.render()
assert hook.count == 1
def test_on_render_crashes(tmp_directory):
dag = DAG()
PythonCallable(touch_root, File('file.txt'), dag, name='t')
dag.on_render = hook_crashing
with pytest.raises(DAGRenderError) as excinfo:
dag.build()
msg = 'Exception when running on_render for DAG "No name": crash!'
assert str(excinfo.value) == msg
assert 'crash!' in str(excinfo.getrepr())
assert dag._exec_status == DAGStatus.ErroredRender
