Screenshot Rule

screenshotrule.kt

class ScreenshotRule(
    private val screenshoter: Paparazzi,
) : TestRule {

    val layoutInflater: LayoutInflater
        get() = paparazzi.layoutInflater

    val context: Context
        get() = paparazzi.context

    override fun apply(base: Statement, description: Description): Statement {
        return RuleChain
            .emptyRuleChain()
            .around(MyCustomRule())
            .around(paparazzi)
            .apply(base, description)
    }

    fun snapshot(
        content: @Composable () -> Unit,
    ) {
        paparazzi.snapshot {
            MyTheme {
                content()
            }
        }
    }

    fun snapshot(view: View) {
        paparazzi.snapshot(view)
    }
}