@OptIn(ExperimentalMaterial3Api::class)
@Composable
fun MyBottomSheet(onDismiss: () -> Unit) {
    val sheetState = rememberModalBottomSheetState()
    val scope = rememberCoroutineScope()
    ModalBottomSheet(
        onDismissRequest = onDismiss,
        sheetState = sheetState
    ) {
        Image(
            imageVector = LocalContext.current.CloseIcon,
            contentDescription = "Close this",
            modifier = Modifier
                .padding(dimensionResource(id = R.dimen.padding))
                .clickable {
                    println("Click was called")
                    scope
                        .launch { sheetState.hide() }
                        .invokeOnCompletion {
                            if (!sheetState.isVisible) {
                                onDismiss()
                            }
                        }
                }
        )
    }
}

@RunWith(AndroidJUnit4::class)
class ExampleUnitTest {

    @get:Rule
    val composeTestRule = createComposeRule()

    @Test
    fun openCloseBottomSheet() {
        with(composeTestRule) {
            setContent {
                MainScreen()
            }
            // Open bottom sheet
            onNode(hasText("Hit me") and hasClickAction())
                .assertIsDisplayed()
                .performClick()

            val closeButton = onNodeWithContentDescription("Close this")

            // Click the close button in the bottom sheet
            closeButton
                .assertIsDisplayed()
                .assertHasClickAction()
                .performClick()

            // Check bottom sheet is indeed gone: this fails on API 27 and 29+
            closeButton.assertIsNotDisplayed()
        }
    }
}