Expose Roborazzi options (#2458)

* Expose Roborazzi options
google · Nov 12, 2024 · 42a9990 · 42a9990
1 parent 9acf67a
commit 42a9990
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/roboscreenshots/api/current.api b/roboscreenshots/api/current.api
@@ -129,6 +129,7 @@ package com.google.android.horologist.screenshots.rng {
     method public coil.test.FakeImageLoaderEngine? getImageLoader();
     method @org.junit.Rule public final org.junit.rules.TestName getTestInfo();
     method public float getTolerance();
+    method public com.github.takahirom.roborazzi.RoborazziOptions roborazziOptions(optional boolean applyDeviceCrop);
     method public final void runTest(optional String? suffix, optional com.google.android.horologist.screenshots.rng.WearDevice? device, optional boolean applyDeviceConfig, optional boolean captureScreenshot, kotlin.jvm.functions.Function0<kotlin.Unit> content);
     method public String testName(String suffix);
     property @org.junit.Rule public final androidx.compose.ui.test.junit4.ComposeContentTestRule composeRule;

diff --git a/...enshots/src/main/java/com/google/android/horologist/screenshots/rng/WearScreenshotTest.kt b/...enshots/src/main/java/com/google/android/horologist/screenshots/rng/WearScreenshotTest.kt
@@ -74,6 +74,16 @@ public abstract class WearScreenshotTest {
 
     public open val imageLoader: FakeImageLoaderEngine? = null
 
+    public open fun roborazziOptions(applyDeviceCrop: Boolean = true): RoborazziOptions =
+        RoborazziOptions(
+            recordOptions = RoborazziOptions.RecordOptions(
+                applyDeviceCrop = applyDeviceCrop,
+            ),
+            compareOptions = RoborazziOptions.CompareOptions(
+                resultValidator = ThresholdValidator(tolerance),
+            ),
+        )
+
     public fun runTest(
         suffix: String? = null,
         device: WearDevice? = this.device,
@@ -101,14 +111,7 @@ public abstract class WearScreenshotTest {
     public fun captureScreenshot(suffix: String = "") {
         captureScreenRoboImage(
             filePath = testName(suffix),
-            roborazziOptions = RoborazziOptions(
-                recordOptions = RoborazziOptions.RecordOptions(
-                    applyDeviceCrop = true,
-                ),
-                compareOptions = RoborazziOptions.CompareOptions(
-                    resultValidator = ThresholdValidator(tolerance),
-                ),
-            ),
+            roborazziOptions = roborazziOptions(),
         )
     }