From fb81ae4c3caacdd3ffc06200513a44dda0ab3745 Mon Sep 17 00:00:00 2001 From: Joshua Trask Date: Tue, 25 Apr 2023 19:05:22 +0000 Subject: Null-check mime types in content preview handling Test: atest CtsSharesheetDeviceTest Bug: 279447957 Change-Id: I44bb350ddf2c9eb79ad51b485cc06d9b8532edf5 --- .../contentpreview/ContentPreviewUiTest.kt | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'java/tests') diff --git a/java/tests/src/com/android/intentresolver/contentpreview/ContentPreviewUiTest.kt b/java/tests/src/com/android/intentresolver/contentpreview/ContentPreviewUiTest.kt index c6a47515..6c30fc9e 100644 --- a/java/tests/src/com/android/intentresolver/contentpreview/ContentPreviewUiTest.kt +++ b/java/tests/src/com/android/intentresolver/contentpreview/ContentPreviewUiTest.kt @@ -20,6 +20,7 @@ import android.content.res.Resources import android.view.LayoutInflater import android.view.ViewGroup import com.android.intentresolver.widget.ActionRow +import com.android.intentresolver.widget.ScrollableImagePreviewView.PreviewType import com.google.common.truth.Truth.assertThat import org.junit.Test @@ -44,6 +45,23 @@ class ContentPreviewUiTest { } } + @Test + fun testPreviewTypes() { + val typeClassifier = object : MimeTypeClassifier { + override fun isImageType(type: String?) = (type == "image") + override fun isVideoType(type: String?) = (type == "video") + } + + assertThat(ContentPreviewUi.getPreviewType(typeClassifier, "image")) + .isEqualTo(PreviewType.Image) + assertThat(ContentPreviewUi.getPreviewType(typeClassifier, "video")) + .isEqualTo(PreviewType.Video) + assertThat(ContentPreviewUi.getPreviewType(typeClassifier, "other")) + .isEqualTo(PreviewType.File) + assertThat(ContentPreviewUi.getPreviewType(typeClassifier, null)) + .isEqualTo(PreviewType.File) + } + @Test fun testCreateActions() { val preview = TestablePreview() -- cgit v1.2.3-59-g8ed1b