Initialize PlayloadToggleInteractor in PreviewViewModel

Initialize PlayloadToggleInteractor in PreviewViewModel when the flag is
enabled.

Bug: 302691505
Test: IntentResolver-tests-unit (Both V2 flag values)
Test: IntentResolver-tests-integration (Both V2 flag values)
Test: IntentResolver-tests-activity (Both V2 flag values)
Test: Functinality smoke test (orentation change, different preview
 types, target selection flow) for both V2 flag values.
Change-Id: I5899b0dd25b9482e56d17bdcad57a0aaa4600734

1 files changed, 20 insertions, 6 deletions

diff --git a/tests/shared/src/com/android/intentresolver/TestContentPreviewViewModel.kt b/tests/shared/src/com/android/intentresolver/TestContentPreviewViewModel.kt
index 998c0802..b352f360 100644
--- a/tests/shared/src/com/android/intentresolver/TestContentPreviewViewModel.kt
+++ b/tests/shared/src/com/android/intentresolver/TestContentPreviewViewModel.kt
@@ -17,28 +17,42 @@
 package com.android.intentresolver
 
 import android.content.Intent
+import android.net.Uri
 import androidx.lifecycle.ViewModel
 import androidx.lifecycle.ViewModelProvider
 import androidx.lifecycle.viewmodel.CreationExtras
 import com.android.intentresolver.contentpreview.BasePreviewViewModel
 import com.android.intentresolver.contentpreview.ImageLoader
 import com.android.intentresolver.contentpreview.PayloadToggleInteractor
-import com.android.intentresolver.contentpreview.PreviewDataProvider
 
 /** A test content preview model that supports image loader override. */
 class TestContentPreviewViewModel(
     private val viewModel: BasePreviewViewModel,
-    private val imageLoaderDelegate: ImageLoader?,
+    override val imageLoader: ImageLoader,
 ) : BasePreviewViewModel() {
-    override fun createOrReuseProvider(targetIntent: Intent): PreviewDataProvider =
-        viewModel.createOrReuseProvider(targetIntent)
 
-    override val imageLoader: ImageLoader
-        get() = imageLoaderDelegate ?: viewModel.imageLoader
+    override val previewDataProvider
+        get() = viewModel.previewDataProvider
 
     override val payloadToggleInteractor: PayloadToggleInteractor?
         get() = viewModel.payloadToggleInteractor
 
+    override fun init(
+        targetIntent: Intent,
+        chooserIntent: Intent,
+        additionalContentUri: Uri?,
+        focusedItemIdx: Int,
+        isPayloadTogglingEnabled: Boolean,
+    ) {
+        viewModel.init(
+            targetIntent,
+            chooserIntent,
+            additionalContentUri,
+            focusedItemIdx,
+            isPayloadTogglingEnabled
+        )
+    }
+
     companion object {
         fun wrap(
             factory: ViewModelProvider.Factory,