test

dhis2 · Apr 29, 2024 · 0599920 · 0599920
1 parent e22f516
commit 0599920
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/app/src/test/java/org/dhis2/usescases/teiDashboard/DashboardViewModelTest.kt b/app/src/test/java/org/dhis2/usescases/teiDashboard/DashboardViewModelTest.kt
@@ -120,6 +120,7 @@ class DashboardViewModelTest {
             whenever(repository.updateEnrollmentStatus(any(), any())) doReturn Observable.just(
                 StatusChangeResultCode.CHANGED,
             )
+            whenever(mockedEnrollmentModel.currentEnrollment) doReturn mockedCompletedEnrollment
             updateEnrollmentStatus(EnrollmentStatus.COMPLETED)
             testingDispatcher.scheduler.advanceUntilIdle()
             verify(repository).updateEnrollmentStatus("enrollmentUid", EnrollmentStatus.COMPLETED)
@@ -186,4 +187,10 @@ class DashboardViewModelTest {
         on { aggregatedSyncState() } doReturn State.SYNCED
         on { status() } doReturn EnrollmentStatus.ACTIVE
     }
+
+    private val mockedCompletedEnrollment: Enrollment = mock {
+        on { uid() } doReturn "enrollmentUid"
+        on { aggregatedSyncState() } doReturn State.TO_UPDATE
+        on { status() } doReturn EnrollmentStatus.COMPLETED
+    }
 }