How to make Enzyme or Jest wait until context work

Question

I am passing a function to the context that is called when the component that receives this context is mounted.This function set value to parent component state.

Exam