1 files changed, 6 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/intentresolver/ChooserActionFactoryTest.kt b/tests/unit/src/com/android/intentresolver/ChooserActionFactoryTest.kt
index 8dfbdbdd..c8e17de4 100644
--- a/tests/unit/src/com/android/intentresolver/ChooserActionFactoryTest.kt
+++ b/tests/unit/src/com/android/intentresolver/ChooserActionFactoryTest.kt
@@ -69,6 +69,8 @@ class ChooserActionFactoryTest {
                 latestReturn = resultCode
             }
         }
+    private val featureFlags =
+        FakeFeatureFlagsImpl().apply { setFlag(Flags.FLAG_FIX_PARTIAL_IMAGE_EDIT_TRANSITION, true) }
 
     @Before
     fun setup() {
@@ -119,6 +121,7 @@ class ChooserActionFactoryTest {
                 /* shareResultSender = */ null,
                 /* finishCallback = */ {},
                 /* clipboardManager = */ mock(),
+                /* featureFlags = */ featureFlags,
             )
         assertThat(testSubject.copyButtonRunnable).isNull()
     }
@@ -140,6 +143,7 @@ class ChooserActionFactoryTest {
                 /* shareResultSender = */ null,
                 /* finishCallback = */ {},
                 /* clipboardManager = */ mock(),
+                /* featureFlags = */ featureFlags,
             )
         assertThat(testSubject.copyButtonRunnable).isNull()
     }
@@ -162,6 +166,7 @@ class ChooserActionFactoryTest {
                 /* shareResultSender = */ resultSender,
                 /* finishCallback = */ {},
                 /* clipboardManager = */ mock(),
+                /* featureFlags = */ featureFlags,
             )
         assertThat(testSubject.copyButtonRunnable).isNotNull()
 
@@ -194,6 +199,7 @@ class ChooserActionFactoryTest {
             /* shareResultSender = */ null,
             /* finishCallback = */ resultConsumer,
             /* clipboardManager = */ mock(),
+            /* featureFlags = */ featureFlags,
         )
     }
 }