1 files changed, 423 insertions, 0 deletions
diff --git a/tests/unit/src/com/android/intentresolver/contentpreview/FilesPlusTextContentPreviewUiTest.kt b/tests/unit/src/com/android/intentresolver/contentpreview/FilesPlusTextContentPreviewUiTest.kt
new file mode 100644
index 0000000..7cc0b4b
--- /dev/null
+++ b/tests/unit/src/com/android/intentresolver/contentpreview/FilesPlusTextContentPreviewUiTest.kt
@@ -0,0 +1,423 @@
+/*
+ * Copyright (C) 2023 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.intentresolver.contentpreview
+
+import android.net.Uri
+import android.view.LayoutInflater
+import android.view.View
+import android.view.ViewGroup
+import android.widget.TextView
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import androidx.test.platform.app.InstrumentationRegistry.getInstrumentation
+import com.android.intentresolver.R
+import com.android.intentresolver.mock
+import com.android.intentresolver.whenever
+import com.android.intentresolver.widget.ActionRow
+import com.google.common.truth.Truth.assertThat
+import com.google.common.truth.Truth.assertWithMessage
+import java.util.function.Consumer
+import kotlin.coroutines.EmptyCoroutineContext
+import kotlinx.coroutines.test.TestScope
+import kotlinx.coroutines.test.UnconfinedTestDispatcher
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.Mockito.anyInt
+import org.mockito.Mockito.never
+import org.mockito.Mockito.times
+import org.mockito.Mockito.verify
+
+private const val HEADLINE_IMAGES = "Image Headline"
+private const val HEADLINE_VIDEOS = "Video Headline"
+private const val HEADLINE_FILES = "Files Headline"
+private const val SHARED_TEXT = "Some text to share"
+
+@RunWith(AndroidJUnit4::class)
+class FilesPlusTextContentPreviewUiTest {
+    private val testScope = TestScope(EmptyCoroutineContext + UnconfinedTestDispatcher())
+    private val actionFactory =
+        object : ChooserContentPreviewUi.ActionFactory {
+            override fun getEditButtonRunnable(): Runnable? = null
+            override fun getCopyButtonRunnable(): Runnable? = null
+            override fun createCustomActions(): List<ActionRow.Action> = emptyList()
+            override fun getModifyShareAction(): ActionRow.Action? = null
+            override fun getExcludeSharedTextAction(): Consumer<Boolean> = Consumer<Boolean> {}
+        }
+    private val imageLoader = mock<ImageLoader>()
+    private val headlineGenerator =
+        mock<HeadlineGenerator> {
+            whenever(getImagesHeadline(anyInt())).thenReturn(HEADLINE_IMAGES)
+            whenever(getVideosHeadline(anyInt())).thenReturn(HEADLINE_VIDEOS)
+            whenever(getFilesHeadline(anyInt())).thenReturn(HEADLINE_FILES)
+        }
+
+    private val context
+        get() = getInstrumentation().context
+
+    @Test
+    fun test_displayImagesPlusTextWithoutUriMetadata_showImagesHeadline() {
+        val sharedFileCount = 2
+        val previewView = testLoadingHeadline("image/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_IMAGES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayImagesPlusTextWithoutUriMetadataExternalHeader_showImagesHeadline() {
+        val sharedFileCount = 2
+        val (previewView, headerParent) = testLoadingExternalHeadline("image/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_IMAGES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayVideosPlusTextWithoutUriMetadata_showVideosHeadline() {
+        val sharedFileCount = 2
+        val previewView = testLoadingHeadline("video/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getVideosHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_VIDEOS)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayVideosPlusTextWithoutUriMetadataExternalHeader_showVideosHeadline() {
+        val sharedFileCount = 2
+        val (previewView, headerParent) = testLoadingExternalHeadline("video/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getVideosHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_VIDEOS)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayDocsPlusTextWithoutUriMetadata_showFilesHeadline() {
+        val sharedFileCount = 2
+        val previewView = testLoadingHeadline("application/pdf", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayDocsPlusTextWithoutUriMetadataExternalHeader_showFilesHeadline() {
+        val sharedFileCount = 2
+        val (previewView, headerParent) =
+            testLoadingExternalHeadline("application/pdf", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayMixedContentPlusTextWithoutUriMetadata_showFilesHeadline() {
+        val sharedFileCount = 2
+        val previewView = testLoadingHeadline("*/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayMixedContentPlusTextWithoutUriMetadataExternalHeader_showFilesHeadline() {
+        val sharedFileCount = 2
+        val (previewView, headerParent) = testLoadingExternalHeadline("*/*", sharedFileCount)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayImagesPlusTextWithUriMetadataSet_showImagesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("image/png", "image/jpeg")
+        val sharedFileCount = loadedFileMetadata.size
+        val previewView = testLoadingHeadline("image/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_IMAGES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayImagesPlusTextWithUriMetadataSetExternalHeader_showImagesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("image/png", "image/jpeg")
+        val sharedFileCount = loadedFileMetadata.size
+        val (previewView, headerParent) =
+            testLoadingExternalHeadline("image/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_IMAGES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayVideosPlusTextWithUriMetadataSet_showVideosHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("video/mp4", "video/mp4")
+        val sharedFileCount = loadedFileMetadata.size
+        val previewView = testLoadingHeadline("video/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getVideosHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_VIDEOS)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayVideosPlusTextWithUriMetadataSetExternalHeader_showVideosHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("video/mp4", "video/mp4")
+        val sharedFileCount = loadedFileMetadata.size
+        val (previewView, headerParent) =
+            testLoadingExternalHeadline("video/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getVideosHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_VIDEOS)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayImagesAndVideosPlusTextWithUriMetadataSet_showFilesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("image/png", "video/mp4")
+        val sharedFileCount = loadedFileMetadata.size
+        val previewView = testLoadingHeadline("*/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayImagesAndVideosPlusTextWithUriMetadataSetExternalHeader_showFilesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("image/png", "video/mp4")
+        val sharedFileCount = loadedFileMetadata.size
+        val (previewView, headerParent) =
+            testLoadingExternalHeadline("*/*", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayDocsPlusTextWithUriMetadataSet_showFilesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("application/pdf", "application/pdf")
+        val sharedFileCount = loadedFileMetadata.size
+        val previewView =
+            testLoadingHeadline("application/pdf", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_displayDocsPlusTextWithUriMetadataSetExternalHeader_showFilesHeadline() {
+        val loadedFileMetadata = createFileInfosWithMimeTypes("application/pdf", "application/pdf")
+        val sharedFileCount = loadedFileMetadata.size
+        val (previewView, headerParent) =
+            testLoadingExternalHeadline("application/pdf", sharedFileCount, loadedFileMetadata)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(headerParent, HEADLINE_FILES)
+        verifySharedText(previewView)
+    }
+
+    @Test
+    fun test_uriMetadataIsMoreSpecificThanIntentMimeType_headlineGetsUpdated() {
+        val sharedFileCount = 2
+        val testSubject =
+            FilesPlusTextContentPreviewUi(
+                testScope,
+                /*isSingleImage=*/ false,
+                sharedFileCount,
+                SHARED_TEXT,
+                /*intentMimeType=*/ "*/*",
+                actionFactory,
+                imageLoader,
+                DefaultMimeTypeClassifier,
+                headlineGenerator
+            )
+        val layoutInflater = LayoutInflater.from(context)
+        val gridLayout = layoutInflater.inflate(R.layout.chooser_grid, null, false) as ViewGroup
+
+        val previewView =
+            testSubject.display(context.resources, LayoutInflater.from(context), gridLayout, null)
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verify(headlineGenerator, never()).getImagesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_FILES)
+
+        testSubject.updatePreviewMetadata(createFileInfosWithMimeTypes("image/png", "image/jpg"))
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyPreviewHeadline(previewView, HEADLINE_IMAGES)
+    }
+
+    @Test
+    fun test_uriMetadataIsMoreSpecificThanIntentMimeTypeExternalHeader_headlineGetsUpdated() {
+        val sharedFileCount = 2
+        val testSubject =
+            FilesPlusTextContentPreviewUi(
+                testScope,
+                /*isSingleImage=*/ false,
+                sharedFileCount,
+                SHARED_TEXT,
+                /*intentMimeType=*/ "*/*",
+                actionFactory,
+                imageLoader,
+                DefaultMimeTypeClassifier,
+                headlineGenerator
+            )
+        val layoutInflater = LayoutInflater.from(context)
+        val gridLayout =
+            layoutInflater.inflate(R.layout.chooser_grid_scrollable_preview, null, false)
+                as ViewGroup
+        val externalHeaderView =
+            gridLayout.requireViewById<View>(R.id.chooser_headline_row_container)
+
+        assertWithMessage("External headline should not be inflated by default")
+            .that(externalHeaderView.findViewById<View>(R.id.headline))
+            .isNull()
+
+        val previewView =
+            testSubject.display(
+                context.resources,
+                LayoutInflater.from(context),
+                gridLayout,
+                externalHeaderView
+            )
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verify(headlineGenerator, never()).getImagesHeadline(sharedFileCount)
+        verifyInternalHeadlineAbsence(previewView)
+        verifyPreviewHeadline(externalHeaderView, HEADLINE_FILES)
+
+        testSubject.updatePreviewMetadata(createFileInfosWithMimeTypes("image/png", "image/jpg"))
+
+        verify(headlineGenerator, times(1)).getFilesHeadline(sharedFileCount)
+        verify(headlineGenerator, times(1)).getImagesHeadline(sharedFileCount)
+        verifyPreviewHeadline(externalHeaderView, HEADLINE_IMAGES)
+    }
+
+    private fun testLoadingHeadline(
+        intentMimeType: String,
+        sharedFileCount: Int,
+        loadedFileMetadata: List<FileInfo>? = null,
+    ): ViewGroup? {
+        val testSubject =
+            FilesPlusTextContentPreviewUi(
+                testScope,
+                /*isSingleImage=*/ false,
+                sharedFileCount,
+                SHARED_TEXT,
+                intentMimeType,
+                actionFactory,
+                imageLoader,
+                DefaultMimeTypeClassifier,
+                headlineGenerator
+            )
+        val layoutInflater = LayoutInflater.from(context)
+        val gridLayout = layoutInflater.inflate(R.layout.chooser_grid, null, false) as ViewGroup
+
+        loadedFileMetadata?.let(testSubject::updatePreviewMetadata)
+        return testSubject.display(
+            context.resources,
+            LayoutInflater.from(context),
+            gridLayout,
+            /*headlineViewParent=*/ null
+        )
+    }
+
+    private fun testLoadingExternalHeadline(
+        intentMimeType: String,
+        sharedFileCount: Int,
+        loadedFileMetadata: List<FileInfo>? = null,
+    ): Pair<ViewGroup?, View> {
+        val testSubject =
+            FilesPlusTextContentPreviewUi(
+                testScope,
+                /*isSingleImage=*/ false,
+                sharedFileCount,
+                SHARED_TEXT,
+                intentMimeType,
+                actionFactory,
+                imageLoader,
+                DefaultMimeTypeClassifier,
+                headlineGenerator
+            )
+        val layoutInflater = LayoutInflater.from(context)
+        val gridLayout =
+            layoutInflater.inflate(R.layout.chooser_grid_scrollable_preview, null, false)
+                as ViewGroup
+        val externalHeaderView =
+            gridLayout.requireViewById<View>(R.id.chooser_headline_row_container)
+
+        assertWithMessage("External headline should not be inflated by default")
+            .that(externalHeaderView.findViewById<View>(R.id.headline))
+            .isNull()
+
+        loadedFileMetadata?.let(testSubject::updatePreviewMetadata)
+        return testSubject.display(
+            context.resources,
+            LayoutInflater.from(context),
+            gridLayout,
+            externalHeaderView
+        ) to externalHeaderView
+    }
+
+    private fun createFileInfosWithMimeTypes(vararg mimeTypes: String): List<FileInfo> {
+        val uri = Uri.parse("content://pkg.app/file")
+        return mimeTypes.map { mimeType -> FileInfo.Builder(uri).withMimeType(mimeType).build() }
+    }
+
+    private fun verifyPreviewHeadline(headerViewParent: View?, expectedText: String) {
+        assertThat(headerViewParent).isNotNull()
+        val headlineView = headerViewParent?.findViewById<TextView>(R.id.headline)
+        assertThat(headlineView).isNotNull()
+        assertThat(headlineView?.text).isEqualTo(expectedText)
+    }
+
+    private fun verifySharedText(previewView: ViewGroup?) {
+        assertThat(previewView).isNotNull()
+        val textContentView = previewView?.findViewById<TextView>(R.id.content_preview_text)
+        assertThat(textContentView).isNotNull()
+        assertThat(textContentView?.text).isEqualTo(SHARED_TEXT)
+    }
+
+    private fun verifyInternalHeadlineAbsence(previewView: ViewGroup?) {
+        assertWithMessage("Preview parent should not be null").that(previewView).isNotNull()
+        assertWithMessage(
+                "Preview headline should not be inflated when an external headline is used"
+            )
+            .that(previewView?.findViewById<View>(R.id.headline))
+            .isNull()
+    }
+}