From 70bf9101c4408b03110b0840e87c8828577fb9b5 Mon Sep 17 00:00:00 2001
From: Honza <rychnovsky.honza@gmail.com>
Date: Wed, 12 Mar 2025 14:30:50 +0100
Subject: [PATCH] Screenshot testing fix

---
 .../zcash/ui/screenshot/ScreenshotTest.kt              | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/ui-screenshot-test/src/main/java/co/electroniccoin/zcash/ui/screenshot/ScreenshotTest.kt b/ui-screenshot-test/src/main/java/co/electroniccoin/zcash/ui/screenshot/ScreenshotTest.kt
index fb8fb9e3a..df25c77c7 100644
--- a/ui-screenshot-test/src/main/java/co/electroniccoin/zcash/ui/screenshot/ScreenshotTest.kt
+++ b/ui-screenshot-test/src/main/java/co/electroniccoin/zcash/ui/screenshot/ScreenshotTest.kt
@@ -3,6 +3,7 @@
 package co.electroniccoin.zcash.ui.screenshot
 
 import android.content.Context
+import android.graphics.Bitmap
 import android.os.Build
 import android.os.LocaleList
 import androidx.activity.viewModels
@@ -24,8 +25,8 @@ import androidx.test.core.app.ApplicationProvider
 import androidx.test.core.graphics.writeToTestStorage
 import androidx.test.espresso.Espresso
 import androidx.test.espresso.Espresso.onView
+import androidx.test.espresso.action.ViewActions.captureToBitmap
 import androidx.test.espresso.matcher.ViewMatchers.isRoot
-import androidx.test.espresso.screenshot.captureToBitmap
 import androidx.test.ext.junit.rules.ActivityScenarioRule
 import androidx.test.filters.LargeTest
 import androidx.test.filters.SdkSuppress
@@ -87,8 +88,11 @@ class ScreenshotTest : UiTestPrerequisites() {
             screenshotName: String
         ) {
             onView(isRoot())
-                .captureToBitmap()
-                .writeToTestStorage("$screenshotName - $tag")
+                .perform(
+                    captureToBitmap { bitmap: Bitmap ->
+                        bitmap.writeToTestStorage("$screenshotName - $tag")
+                    }
+                )
         }
     }