Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Adding optional block to run before screenshot" into udc-dev

Browse Source
This commit is contained in:
Grace Cheng
2023-06-01 22:05:28 +00:00
committed by Android (Google) Code Review
parent 7515404afb 2336a16211
commit a6550a836b
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
packages/SystemUI/screenshot/src/com/android/systemui/testing/screenshot/ViewScreenshotTestRule.kt
View File

@@ -78,6 +78,7 @@ open class ViewScreenshotTestRule(
protected fun takeScreenshot(
mode: Mode = Mode.WrapContent,
viewProvider: (ComponentActivity) -> View,
beforeScreenshot: (ComponentActivity) -> Unit = {}
): Bitmap {
activityRule.scenario.onActivity { activity ->
// Make sure that the activity draws full screen and fits the whole display instead of
@@ -103,6 +104,7 @@ open class ViewScreenshotTestRule(
val content = activity.requireViewById<ViewGroup>(android.R.id.content)
assertEquals(1, content.childCount)
contentView = content.getChildAt(0)
beforeScreenshot(activity)
}
return if (isRobolectric) {
@@ -120,9 +122,10 @@ open class ViewScreenshotTestRule(
fun screenshotTest(
goldenIdentifier: String,
mode: Mode = Mode.WrapContent,
viewProvider: (ComponentActivity) -> View,
beforeScreenshot: (ComponentActivity) -> Unit = {},
viewProvider: (ComponentActivity) -> View
) {
val bitmap = takeScreenshot(mode, viewProvider)
val bitmap = takeScreenshot(mode, viewProvider, beforeScreenshot)
screenshotRule.assertBitmapAgainstGolden(
bitmap,
goldenIdentifier,