From 05999207b50e544a3d67565528f0a9b96c417308 Mon Sep 17 00:00:00 2001
From: manu <mmunoz@quadram.mobi>
Date: Mon, 29 Apr 2024 13:29:54 +0200
Subject: [PATCH] test

---
 .../dhis2/usescases/teiDashboard/DashboardViewModelTest.kt | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/app/src/test/java/org/dhis2/usescases/teiDashboard/DashboardViewModelTest.kt b/app/src/test/java/org/dhis2/usescases/teiDashboard/DashboardViewModelTest.kt
index a10a2c35316..1e0c57fe87a 100644
--- 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
+    }
 }