FlakyTest.kt

// ViewState
class ViewState {
  ...
  var screenState by mutableStateOf(ScreenState.FORM)
  ...
}

// ViewModel
class ViewModel {
  ...
  fun test() {
    viewState.screenState = ScreenState.SUCCESS
    Log(${viewState.screenState}) // This prints correct value
  }
  ...
}

// Composable
fun Screen() {
    LaunchedEffect(Unit) {
        snapshotFlow { viewModel.viewState.screenState }.collect {
            Log($it) // This prints correct value
        }
    }
    
    Screen.Content(
       state = viewModel.viewState.screenState
    )
}

// Content
fun Content(state: ScreenState) {
    Log($state) // This doesn't print nothing

    LaunchedEffect(screenState) {
      Log($screenState) // This doesn't print nothing
    }
}