I'm having difficulty getting screenshot testing to work.

When I run the updateDebugScreenshotTest target, the build fails with a message from Kotlin coroutines:

Exception in thread "Layoutlib Render Thread" java.lang.IllegalStateException: Module with the Main dispatcher is missing. Add dependency providing the Main dispatcher, e.g. 'kotlinx-coroutines-android' and ensure it has the same version as 'kotlinx-coroutines-core'

I've tried various things to make this work. But ultimately, I don't think I should have to do anything special to make this work: the app runs fine on a device, and screenshot previews do work fine inside the IDE. So there must be something wrong with the way we're transferring the dependency graph used by the IDE to the screenshot task, or something like that.

I've made my sample project available here: https://drive.google.com/file/d/1YIDqlLa1R1HWzzZ43pPGyCLM1bXfliTK/view?usp=sharing . The project doesn't make a lot of sense -- I've stripped it down from a real project -- but it reproduces this exact problem; you can open it up in Studio and see the preview, and you can build the project, but the update screenshot task fails with the full exception:

./gradlew  updateDebugScreenshotTest
Configuration on demand is an incubating feature.

> Configure project :

> Configure project :app
WARNING: The option setting 'android.experimental.enableScreenshotTest=true' is experimental.
The current default is 'false'.
WARNING: A terminally deprecated method in java.lang.System has been called
WARNING: System::setSecurityManager has been called by com.android.tools.rendering.security.RenderSecurityManager (file:/Users/tnorbye/.gradle/caches/modules-2/files-2.1/com.android.tools.compose/compose-preview-renderer/0.0.1-alpha07/dc658a9f98c48cd96d468e312d6e18923914c8b2/compose-preview-renderer-0.0.1-alpha07.jar)
WARNING: Please consider reporting this to the maintainers of com.android.tools.rendering.security.RenderSecurityManager
WARNING: System::setSecurityManager will be removed in a future release
Exception in thread "Layoutlib Render Thread" java.lang.IllegalStateException: Module with the Main dispatcher is missing. Add dependency providing the Main dispatcher, e.g. 'kotlinx-coroutines-android' and ensure it has the same version as 'kotlinx-coroutines-core'
        at kotlinx.coroutines.internal.MainDispatchersKt.throwMissingMainDispatcherException(MainDispatchers.kt:77)
        at kotlinx.coroutines.internal.MissingMainCoroutineDispatcher.missing(MainDispatchers.kt:108)
        at kotlinx.coroutines.internal.MissingMainCoroutineDispatcher.isDispatchNeeded(MainDispatchers.kt:92)
        at kotlinx.coroutines.internal.DispatchedContinuationKt.resumeCancellableWith(DispatchedContinuation.kt:315)
        at kotlinx.coroutines.intrinsics.CancellableKt.startCoroutineCancellable(Cancellable.kt:26)
        at kotlinx.coroutines.intrinsics.CancellableKt.startCoroutineCancellable$default(Cancellable.kt:21)
        at kotlinx.coroutines.CoroutineStart.invoke(CoroutineStart.kt:88)
        at kotlinx.coroutines.AbstractCoroutine.start(AbstractCoroutine.kt:123)
        at kotlinx.coroutines.BuildersKt__Builders_commonKt.launch(Builders.common.kt:52)
        at kotlinx.coroutines.BuildersKt.launch(Unknown Source)
        at kotlinx.coroutines.BuildersKt__Builders_commonKt.launch$default(Builders.common.kt:43)
        at kotlinx.coroutines.BuildersKt.launch$default(Unknown Source)
        at me.saket.telephoto.zoomable.internal.RememberWorker.onRemembered(RememberWorker.kt:17)
        at androidx.compose.runtime.CompositionImpl$RememberEventDispatcher.dispatchRememberObservers(Composition.kt:1364)
        at androidx.compose.runtime.CompositionImpl.applyChangesInLocked(Composition.kt:992)
        at androidx.compose.runtime.CompositionImpl.applyChanges(Composition.kt:1013)
        at androidx.compose.runtime.Recomposer.composeInitial$runtime_release(Recomposer.kt:1150)
        at androidx.compose.runtime.CompositionImpl.composeInitial(Composition.kt:649)
        at androidx.compose.runtime.CompositionImpl.setContent(Composition.kt:635)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:133)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.AndroidComposeView.setOnViewTreeOwnersAvailable(AndroidComposeView.android.kt:1626)
        at androidx.compose.ui.platform.WrappedComposition.setContent(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.WrappedComposition.onStateChanged(Wrapper.android.kt:180)
        at androidx.lifecycle.LifecycleRegistry$ObserverWithState.dispatchEvent(LifecycleRegistry.jvm.kt:320)
        at androidx.lifecycle.LifecycleRegistry.addObserver(LifecycleRegistry.jvm.kt:198)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:131)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.AndroidComposeView.onAttachedToWindow(AndroidComposeView.android.kt:1707)
        at android.view.View.dispatchAttachedToWindow(View.java:21980)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3490)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.AttachInfo_Accessor.setAttachInfo(AttachInfo_Accessor.java:51)
        at com.android.layoutlib.bridge.impl.RenderSessionImpl.inflate(RenderSessionImpl.java:372)
        at com.android.layoutlib.bridge.Bridge.createSession(Bridge.java:454)
        at com.android.tools.idea.layoutlib.LayoutLibrary.createSession(LayoutLibrary.java:120)
        at com.android.tools.rendering.RenderTask.createRenderSession(RenderTask.java:784)
        at com.android.tools.rendering.RenderTask.lambda$inflate$6(RenderTask.java:932)
        at com.android.tools.rendering.RenderExecutor$runAsyncActionWithTimeout$3.run(RenderExecutor.kt:204)
        at com.android.tools.rendering.RenderExecutor$PriorityRunnable.run(RenderExecutor.kt:321)
        at java.base/java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1136)
        at java.base/java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:635)
        at java.base/java.lang.Thread.run(Thread.java:840)
        Suppressed: kotlinx.coroutines.internal.DiagnosticCoroutineContextException: [StandaloneCoroutine{Cancelling}@109f0a8c, Dispatchers.Main[missing]]
java.lang.IllegalStateException: Module with the Main dispatcher is missing. Add dependency providing the Main dispatcher, e.g. 'kotlinx-coroutines-android' and ensure it has the same version as 'kotlinx-coroutines-core'
        at kotlinx.coroutines.internal.MainDispatchersKt.throwMissingMainDispatcherException(MainDispatchers.kt:77)
        at kotlinx.coroutines.internal.MissingMainCoroutineDispatcher.missing(MainDispatchers.kt:108)
        at kotlinx.coroutines.internal.MissingMainCoroutineDispatcher.isDispatchNeeded(MainDispatchers.kt:92)
        at kotlinx.coroutines.internal.DispatchedContinuationKt.resumeCancellableWith(DispatchedContinuation.kt:315)
        at kotlinx.coroutines.intrinsics.CancellableKt.startCoroutineCancellable(Cancellable.kt:26)
        at kotlinx.coroutines.intrinsics.CancellableKt.startCoroutineCancellable$default(Cancellable.kt:21)
        at kotlinx.coroutines.CoroutineStart.invoke(CoroutineStart.kt:88)
        at kotlinx.coroutines.AbstractCoroutine.start(AbstractCoroutine.kt:123)
        at kotlinx.coroutines.BuildersKt__Builders_commonKt.launch(Builders.common.kt:52)
        at kotlinx.coroutines.BuildersKt.launch(Unknown Source)
        at kotlinx.coroutines.BuildersKt__Builders_commonKt.launch$default(Builders.common.kt:43)
        at kotlinx.coroutines.BuildersKt.launch$default(Unknown Source)
        at me.saket.telephoto.zoomable.internal.RememberWorker.onRemembered(RememberWorker.kt:17)
        at androidx.compose.runtime.CompositionImpl$RememberEventDispatcher.dispatchRememberObservers(Composition.kt:1364)
        at androidx.compose.runtime.CompositionImpl.applyChangesInLocked(Composition.kt:992)
        at androidx.compose.runtime.CompositionImpl.applyChanges(Composition.kt:1013)
        at androidx.compose.runtime.Recomposer.composeInitial$runtime_release(Recomposer.kt:1150)
        at androidx.compose.runtime.CompositionImpl.composeInitial(Composition.kt:649)
        at androidx.compose.runtime.CompositionImpl.setContent(Composition.kt:635)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:133)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.AndroidComposeView.setOnViewTreeOwnersAvailable(AndroidComposeView.android.kt:1626)
        at androidx.compose.ui.platform.WrappedComposition.setContent(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.WrappedComposition.onStateChanged(Wrapper.android.kt:180)
        at androidx.lifecycle.LifecycleRegistry$ObserverWithState.dispatchEvent(LifecycleRegistry.jvm.kt:320)
        at androidx.lifecycle.LifecycleRegistry.addObserver(LifecycleRegistry.jvm.kt:198)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:131)
        at androidx.compose.ui.platform.WrappedComposition$setContent$1.invoke(Wrapper.android.kt:124)
        at androidx.compose.ui.platform.AndroidComposeView.onAttachedToWindow(AndroidComposeView.android.kt:1707)
        at android.view.View.dispatchAttachedToWindow(View.java:21980)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3490)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.ViewGroup.dispatchAttachedToWindow(ViewGroup.java:3497)
        at android.view.AttachInfo_Accessor.setAttachInfo(AttachInfo_Accessor.java:51)
        at com.android.layoutlib.bridge.impl.RenderSessionImpl.inflate(RenderSessionImpl.java:372)
        at com.android.layoutlib.bridge.Bridge.createSession(Bridge.java:454)
        at com.android.tools.idea.layoutlib.LayoutLibrary.createSession(LayoutLibrary.java:120)
        at com.android.tools.rendering.RenderTask.createRenderSession(RenderTask.java:784)
        at com.android.tools.rendering.RenderTask.lambda$inflate$6(RenderTask.java:932)
        at com.android.tools.rendering.RenderExecutor$runAsyncActionWithTimeout$3.run(RenderExecutor.kt:204)
        at com.android.tools.rendering.RenderExecutor$PriorityRunnable.run(RenderExecutor.kt:321)
        at java.base/java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1136)
        at java.base/java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:635)
        at java.base/java.lang.Thread.run(Thread.java:840)
        Suppressed: kotlinx.coroutines.internal.DiagnosticCoroutineContextException: [StandaloneCoroutine{Cancelled}@109f0a8c, Dispatchers.Main[missing]]

> Task :app:debugPreviewScreenshotRender FAILED

FAILURE: Build failed with an exception.

* What went wrong:
Execution failed for task ':app:debugPreviewScreenshotRender'.
> A failure occurred while executing com.android.compose.screenshot.tasks.PreviewRenderWorkAction
   > Rendering failed. For more details, check /Users/tnorbye/AndroidStudioProjects/ur/app/build/outputs/screenshotTest-results/preview/debug/results.json

* Try:
> Run with --stacktrace option to get the stack trace.
> Run with --info or --debug option to get more log output.
> Run with --scan to get full insights.
> Get more help at https://help.gradle.org.

BUILD FAILED in 2s
26 actionable tasks: 8 executed, 18 up-to-date