Image Helper test (#4485)

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))
+    }
+}

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)
+    }
 }