Hey Prefect, I have a task in 2,0 that I want to test, in that task I use the

get_run_logger()

. In order to test that function I am using

.fn()

but I can not test it because

get_run_logger()

failed every time because

E           prefect.exceptions.MissingContextError: There is no active flow or task run context.

Besides possibly using an optional kwarg in every single task that I have, how can I get around this to test my tasks logic properly?

hi @Ben Muller - you can disable the logger

from prefect.logging import disable_run_logger

def test_some_flow():
   with disable_run_logger():
     call_some_flow_that_uses_logger.fn()

huh

i wrote it wrong at first, but updated - I'm just suggesting calling

your_flow.fn()

within the

disable_run_logger

context manager block in tests

ahhhh

Hey Ben, Our approach to this is to separate the “class task” from the prefect function task. Our code typically takes this form:

@task
def some_task_here(something: str):
  some_task = SomeTask() # can move outside of this function to avoid instantiation each time if desired
  some_task.run(something)

This makes it quite easy to test your code, as it lives outside of prefect context.

Thanks Cole, I actually did this quite a bit in 1.0 but now moving to 2.0 it felt like a bit of a mistake tbh, lead to a lot of duplication in my code base. Appreciate you weighing in and helping out 🙂