From 0bd19aaefb35d7c5b9e219cf76cb9bd8e86b4cb4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sylvain=20N=C3=A9risson?= <sylvain.nerisson@epfl.ch>
Date: Fri, 13 Dec 2024 06:14:50 +0100
Subject: [PATCH] test: add unit testing for StorageDashboardViewModel

---
 .../home/StorageDashboardViewModelTests.kt    | 117 ++++++++++++++++++
 1 file changed, 117 insertions(+)

diff --git a/app/src/test/java/com/github/se/travelpouch/model/home/StorageDashboardViewModelTests.kt b/app/src/test/java/com/github/se/travelpouch/model/home/StorageDashboardViewModelTests.kt
index 146a3ce9..bc0fcf04 100644
--- a/app/src/test/java/com/github/se/travelpouch/model/home/StorageDashboardViewModelTests.kt
+++ b/app/src/test/java/com/github/se/travelpouch/model/home/StorageDashboardViewModelTests.kt
@@ -1,4 +1,121 @@
 package com.github.se.travelpouch.model.home
 
+import android.net.Uri
+import android.util.Log
+import androidx.documentfile.provider.DocumentFile
+import com.github.se.travelpouch.helper.FileDownloader
+import com.github.se.travelpouch.model.travels.Location
+import com.github.se.travelpouch.model.travels.Participant
+import com.github.se.travelpouch.model.travels.Role
+import com.github.se.travelpouch.model.travels.TravelContainer
+import com.google.firebase.Timestamp
+import com.google.firebase.firestore.DocumentReference
+import java.io.ByteArrayInputStream
+import java.io.InputStream
+import kotlin.random.Random
+import kotlinx.coroutines.ExperimentalCoroutinesApi
+import org.hamcrest.CoreMatchers.`is`
+import org.hamcrest.MatcherAssert.assertThat
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.mockito.MockedStatic
+import org.mockito.Mockito.any
+import org.mockito.Mockito.anyString
+import org.mockito.Mockito.doAnswer
+import org.mockito.Mockito.mock
+import org.mockito.Mockito.mockStatic
+import org.mockito.Mockito.spy
+import org.mockito.Mockito.verify
+import org.mockito.Mockito.`when`
+import org.mockito.kotlin.anyOrNull
+import org.mockito.kotlin.whenever
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.shadows.ShadowLog
+
+@RunWith(RobolectricTestRunner::class)
+@ExperimentalCoroutinesApi
 class StorageDashboardViewModelTests {
+
+  @Before
+  fun setup() {
+
+  }
+
+  @Test
+  fun formatStorageUnitAllUnits() {
+    assertThat(formatStorageUnit(0), `is`("0.0 B"))
+    assertThat(formatStorageUnit(1), `is`("1.0 B"))
+    assertThat(formatStorageUnit(1023), `is`("1023.0 B"))
+    assertThat(formatStorageUnit(1024), `is`("1.0 KiB"))
+    assertThat(formatStorageUnit(1024 * 1024 - 1), `is`("1024.0 KiB"))
+    assertThat(formatStorageUnit(1024 * 1024), `is`("1.0 MiB"))
+    assertThat(formatStorageUnit(1024 * 1024 * 1024 - 1), `is`("1024.0 MiB"))
+    assertThat(formatStorageUnit(1024 * 1024 * 1024), `is`("1.0 GiB"))
+    assertThat(formatStorageUnit(1024L * 1024 * 1024 * 1024 - 1), `is`("1024.0 GiB"))
+    assertThat(formatStorageUnit(1024L * 1024 * 1024 * 1024), `is`("1024.0 GiB"))
+  }
+
+  @Test
+  fun storageUsedRatio() {
+    val stats = StorageDashboardStats(1024, 512, 0)
+    assertThat(stats.storageUsedRatio(), `is`(0.5f))
+  }
+
+  @Test
+  fun storageReclaimableRatio() {
+    val stats = StorageDashboardStats(1024, 0, 512)
+    assertThat(stats.storageReclaimableRatio(), `is`(0.5f))
+  }
+
+  @Test
+  fun storageLimitToString() {
+    val stats = StorageDashboardStats(1024, 0, 0)
+    assertThat(stats.storageLimitToString(), `is`("1.0 KiB"))
+  }
+
+  @Test
+  fun storageAvailable() {
+    val stats = StorageDashboardStats(1024, 512, 0)
+    assertThat(stats.storageAvailable(), `is`(512))
+  }
+
+  @Test
+  fun storageAvailableUnlimited() {
+    val stats = StorageDashboardStats(null, 512, 0)
+    assertThat(stats.storageAvailable(), `is`(-1))
+  }
+
+  @Test
+  fun storageAvailableNegative() {
+    val stats = StorageDashboardStats(1024, 1025, 0)
+    assertThat(stats.storageAvailable(), `is`(0))
+  }
+
+  @Test
+  fun storageAvailableUnlimitedNegative() {
+    val stats = StorageDashboardStats(null, 1025, 0)
+    assertThat(stats.storageAvailable(), `is`(-1))
+  }
+
+  @Test
+  fun storageAvailableZero() {
+    val stats = StorageDashboardStats(1024, 1025, 0)
+    assertThat(stats.storageAvailable(), `is`(0))
+  }
+
+  @Test
+  fun storageDashboardViewModelSetStorageStats() {
+    val stats = StorageDashboardStats(1024, 512, 0)
+    val viewModel = StorageDashboardViewModel()
+    viewModel.setStorageStats(stats)
+    assertThat(viewModel.storageStats.value, `is`(stats))
+  }
+
+  @Test
+  fun storageDashboardViewModelUpdateStorageStats() {
+    val viewModel = StorageDashboardViewModel()
+    viewModel.updateStorageStats()
+    assertThat(viewModel.isLoading.value, `is`(false))
+  }
 }
\ No newline at end of file