在视图模型上有任何UI的上下文都是不好的，因为视图模型应该只处理状态。
但是，如果您想保持代码的这种方式，您可以使用MVI设计模式。
使用MVI，UI将仅发送事件（actions）到视图模型，视图模型将负责执行任务，必要时调用任何方法。UI将只从视图模型接收状态并随其更改。这样，很容易将视图模型与屏幕分离。您必须有一个可组合的内容，只接收状态模型和回调发送您的事件（动作）可以预览此内容组合。现在很容易模拟此值。
这是一个如何在Jetpack Compose中使用MVI模式的例子。

TimeViewState.kt

data class TimeViewState(
   val name: String = "",
   val error: Throwable? = null
)

TimeViewEvent.kt

sealed class TimeViewEvent {
    object Set {
        data class Name(val value: String): TimeViewEvent()
        data class Error(val value: Throwable?): TimeViewEvent()
    }
}

TimeViewModel.kt

class TimeViewModel(appObj: Application): AndroidViewModel(appObj) {
    private val _state = MutableStateFlow(TimeViewState())
    val state = _uiState.asStateFlow()

    fun dispatchEvent(event: TimeViewEvent) {
        when (event) {
            is TimeViewEvent.Set.Name -> setName(event.value)
            is TimeViewEvent.Set.Error -> setError(event.value)
        }
    }

    private fun setName(value: String) {
        _state.update { it.copy(name = value) }
    }

    private fun setError(value: Throwable?) {
        _state.update { it.copy(error = value) }
    }
}

ProgressView.kt

@Composable
fun ProgressView(viewModel: TimeViewModel) {
    val state by viewModel.state.collectAsState()

    Content(
        state = state,
        event = viewModel::dispatchEvent
    )
}

@Composable
private fun Content(
    state: TimeViewState,
    event: (TimeViewAction) -> Unit
) {
   // your ui here
}

@Preview
@Composable
private fun ContentPreview() {
    Content(TimeViewState()) {}
}