From 9488737f3e2ff92289239af485751e9a2c468c9b Mon Sep 17 00:00:00 2001
From: Aditya-Srivastav <54016427+4D17Y4@users.noreply.github.com>
Date: Wed, 30 Jun 2021 11:53:04 +0530
Subject: [PATCH] Image Helper test (#4485)

---
 .../customselector/helper/ImageHelperTest.kt  | 55 +++++++++++++++++++
 .../nrw/commons/filepicker/FilePickerTest.kt  | 10 ++++
 2 files changed, 65 insertions(+)
 create mode 100644 app/src/test/kotlin/fr/free/nrw/commons/customselector/helper/ImageHelperTest.kt

diff --git a/app/src/test/kotlin/fr/free/nrw/commons/customselector/helper/ImageHelperTest.kt b/app/src/test/kotlin/fr/free/nrw/commons/customselector/helper/ImageHelperTest.kt
new file mode 100644
index 000000000..2fe382368
--- /dev/null
+++ b/app/src/test/kotlin/fr/free/nrw/commons/customselector/helper/ImageHelperTest.kt
@@ -0,0 +1,55 @@
+package fr.free.nrw.commons.customselector.helper
+
+import android.net.Uri
+import fr.free.nrw.commons.customselector.model.Folder
+import fr.free.nrw.commons.customselector.model.Image
+import org.junit.jupiter.api.Assertions.*
+
+import org.junit.jupiter.api.Test
+import org.mockito.Mockito.mock
+
+/**
+ * Custom Selector Image Helper Test
+ */
+internal class ImageHelperTest {
+
+    var uri: Uri = mock(Uri::class.java)
+    private val folderImage1 = Image(1, "image1", uri, "abc/abc", 1, "bucket1")
+    private val folderImage2 = Image(2, "image1", uri, "xyz/xyz", 2, "bucket2")
+    private val mockImageList = ArrayList<Image>(listOf(folderImage1, folderImage2))
+    private val folderImageList1 = ArrayList<Image>(listOf(folderImage1))
+    private val folderImageList2 = ArrayList<Image>(listOf(folderImage2))
+
+    /**
+     * Test folder list from images.
+     */
+    @Test
+    fun folderListFromImages() {
+        val folderList = ArrayList<Folder>(listOf(Folder(1, "bucket1", folderImageList1), Folder(2, "bucket2", folderImageList2)))
+        assertEquals(folderList, ImageHelper.folderListFromImages(mockImageList))
+    }
+
+    /**
+     * Test filter images.
+     */
+    @Test
+    fun filterImages() {
+        assertEquals(folderImageList1, ImageHelper.filterImages(mockImageList, 1))
+    }
+
+    /**
+     * Test get index from image list.
+     */
+    @Test
+    fun getIndex() {
+        assertEquals(1,ImageHelper.getIndex(mockImageList, folderImage2))
+    }
+
+    /**
+     * Test get index list.
+     */
+    @Test
+    fun getIndexList() {
+        assertEquals(ArrayList<Int>(listOf(0)), ImageHelper.getIndexList(mockImageList, folderImageList2))
+    }
+}
\ No newline at end of file
diff --git a/app/src/test/kotlin/fr/free/nrw/commons/filepicker/FilePickerTest.kt b/app/src/test/kotlin/fr/free/nrw/commons/filepicker/FilePickerTest.kt
index b9712df04..ae841cd1c 100644
--- a/app/src/test/kotlin/fr/free/nrw/commons/filepicker/FilePickerTest.kt
+++ b/app/src/test/kotlin/fr/free/nrw/commons/filepicker/FilePickerTest.kt
@@ -54,4 +54,14 @@ class FilePickerTest {
         verify(activity).startActivityForResult(ArgumentMatchers.anyObject(), requestCodeCaptor?.capture()?.toInt()!!)
         assertEquals(requestCodeCaptor?.value, RequestCodes.TAKE_PICTURE)
     }
+
+    @Test
+    fun testOpenCustomSelectorRequestCode() {
+        `when`(PreferenceManager.getDefaultSharedPreferences(activity)).thenReturn(sharedPref)
+        `when`(sharedPref.edit()).thenReturn(sharedPreferencesEditor)
+        `when`(sharedPref.edit().putInt("type", 0)).thenReturn(sharedPreferencesEditor)
+        FilePicker.openCustomSelector(activity, 0)
+        verify(activity).startActivityForResult(ArgumentMatchers.anyObject(), requestCodeCaptor?.capture()?.toInt()!!)
+        assertEquals(requestCodeCaptor?.value, RequestCodes.PICK_PICTURE_FROM_CUSTOM_SELECTOR)
+    }
 }
\ No newline at end of file