Composable screenshots with SideEffects

[St4B]

I am not sure whether I use Paparazzi wrongly or if there is a way to achieve the desired behavior, but I think that the API should provide a way to take snapshots after all side effects are executed. For example, I have a composable that renders differently when it has focus. In order to set focus programmatically in our screenshot test, we need to use SideEffect like that:

val focusRequester = remember { FocusRequester() }
MyAwesomeComposable(modifier = Modifier.focusRequester(focusRequester))
SideEffect { focusRequester.requestFocus() }

If I try to take a screenshot while the composable has focus:

paparazzi.snapshot("Awesome composable with focus") {
    val focusRequester = remember { FocusRequester() }
    MyAwesomeComposable(modifier = Modifier.focusRequester(focusRequester))
    SideEffect { focusRequester.requestFocus() }
}

then it seems that the screenshot was taken before the focus was gained (which I think is expected).

Maybe we need something like IdlingResource, or something to force all recompositions to happen.