- lint checks completed

app/src/main/java/com/appttude/h_mal/farmr/viewmodel/InfoViewModel.kt

@@ -35,6 +35,5 @@ class InfoViewModel(
             stringBuilder.append(" (+ ").append(shiftObject.breakMins).append(" minutes break)")
         }
         return stringBuilder.toString()
-
     }
 }

app/src/test/java/com/appttude/h_mal/farmr/viewmodel/InfoViewModelTest.kt

@@ -0,0 +1,93 @@
+package com.appttude.h_mal.farmr.viewmodel
+
+import android.os.Bundle
+import com.appttude.h_mal.farmr.data.legacydb.ShiftObject
+import com.appttude.h_mal.farmr.utils.ID
+import io.mockk.every
+import io.mockk.mockk
+import org.junit.Assert.assertEquals
+import org.junit.Test
+import org.mockito.ArgumentMatchers.anyLong
+import kotlin.test.assertIs
+
+class InfoViewModelTest : ShiftViewModelTest<InfoViewModel>() {
+
+    @Test
+    fun retrieveData_validBundleAndId_successfulRetrieval() {
+        // Arrange
+        val id = anyLong()
+        val shift = mockk<ShiftObject>()
+        val bundle = mockk<Bundle>()
+
+        // Act
+        every { repository.readSingleShiftFromDatabase(id) }.returns(shift)
+        every { bundle.getLong(ID) }.returns(id)
+        viewModel.retrieveData(bundle)
+
+        // Assert
+        assertIs<ShiftObject>(retrieveCurrentData())
+        assertEquals(
+            retrieveCurrentData(),
+            shift
+        )
+    }
+
+    @Test
+    fun retrieveData_noValidBundleAndId_unsuccessfulRetrieval() {
+        // Arrange
+        val id = anyLong()
+        val shift = mockk<ShiftObject>()
+        val bundle = mockk<Bundle>()
+
+        // Act
+        every { repository.readSingleShiftFromDatabase(id) }.returns(shift)
+        every { bundle.getLong(ID) }.returns(id)
+        viewModel.retrieveData(null)
+
+        // Assert
+        assertEquals(
+            retrieveCurrentError(),
+            "Failed to retrieve shift"
+        )
+    }
+
+    @Test
+    fun retrieveData_validBundleNoShift_successfulRetrieval() {
+        // Arrange
+        val id = anyLong()
+        val bundle = mockk<Bundle>()
+
+        // Act
+        every { repository.readSingleShiftFromDatabase(id) }.returns(null)
+        every { bundle.getLong(ID) }.returns(id)
+        viewModel.retrieveData(bundle)
+
+        // Assert
+        assertEquals(
+            retrieveCurrentError(),
+            "Failed to retrieve shift"
+        )
+    }
+
+    @Test
+    fun buildDurationSummary_validHourlyShift_successfulRetrieval() {
+        // Arrange
+        val shift = getShifts()[0]
+        val shiftWithBreak = getShifts()[3]
+
+        // Act
+        val summary = viewModel.buildDurationSummary(shift)
+        val summaryWithBreak = viewModel.buildDurationSummary(shiftWithBreak)
+
+        // Assert
+        assertEquals(
+            "1 Hours 0 Minutes ",
+            summary
+        )
+        assertEquals(
+            "1 Hours 0 Minutes  (+ 30 minutes break)",
+            summaryWithBreak
+        )
+    }
+
+}