Fix flakey test.

Signed-off-by: Lentumunai-Mark <[email protected]>
opensrp · Mar 4, 2024 · 209a900 · 209a900
1 parent 5e9453f
commit 209a900
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...ne/src/test/java/org/smartregister/fhircore/engine/util/extension/PatientExtensionTest.kt b/...ne/src/test/java/org/smartregister/fhircore/engine/util/extension/PatientExtensionTest.kt
@@ -256,11 +256,16 @@ class PatientExtensionTest : RobolectricTest() {
 
   @Test
   fun testExtractAgeShouldReturnCallGetAgeStringFromDaysWhenPatientHasBirthDate() {
+    val currentDate = LocalDate.now()
+
+    val oneYearAgo = currentDate.minusYears(1)
+
     val calendar =
-      Calendar.getInstance().apply { timeInMillis = (timeInMillis - (1L * 365 * 24 * 3600 * 1000)) }
+      Calendar.getInstance().apply {
+        timeInMillis = oneYearAgo.atStartOfDay(ZoneId.systemDefault()).toInstant().toEpochMilli()
+      }
 
     val patient = Patient().apply { birthDate = calendar.time }
-
     Assert.assertEquals("1y", patient.extractAge(context))
   }