hi all! Trying to test tasks that make use of the run context with

Prefect 1

I could do this:

def test_my_task():
    with prefect.context(dict(flow_run_id="test_id")):
      my_task()
    # assertions

how c an this be done in

Prefect 2

?

I have issues with

get_run_context

being called from

task

.

@Deceivious have changed to

from prefect import runtime
# inside flow
flow_run_id = runtime.flow_run.id

i.e. use the context inside the flows and pass extracted variables to tasks this way the tasks can be tested independently (unit testing fashion) was looking for a neat way to test the task itself…