From 8912a9d73c78285425e04bfd54c601d8016852ca Mon Sep 17 00:00:00 2001 From: eduebernal Date: Wed, 6 Jul 2022 10:48:58 -0700 Subject: [PATCH] add test for getWeekdaysInMonth --- .../uhabits/core/utils/DateUtilsTest.kt | 25 +++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt b/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt index 75280ab9b..7f505ee13 100644 --- a/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt +++ b/uhabits-core/src/jvmTest/java/org/isoron/uhabits/core/utils/DateUtilsTest.kt @@ -118,6 +118,31 @@ class DateUtilsTest : BaseUnitTest() { assertThat(arrayOf("Sat", "Sun", "Mon", "Tue", "Wed", "Thu", "Fri"), equalTo(longWeekdayNames)) } + @Test + fun getWeekdaysInMonth() { + val february = GregorianCalendar(2018, Calendar.FEBRUARY, 1) + val leapFebruary = GregorianCalendar(2020, Calendar.FEBRUARY, 1) + val month = GregorianCalendar(2020, Calendar.APRIL, 1) + val longMonth = GregorianCalendar(2020, Calendar.AUGUST, 1) + + assertThat( + DateUtils.getWeekdaysInMonth(Timestamp(february)), + equalTo(arrayOf(4, 4, 4, 4, 4, 4, 4)) + ) + assertThat( + DateUtils.getWeekdaysInMonth(Timestamp(leapFebruary)), + equalTo(arrayOf(4, 4, 4, 4, 4, 4, 5)) + ) + assertThat( + DateUtils.getWeekdaysInMonth(Timestamp(month)), + equalTo(arrayOf(4, 4, 4, 5, 5, 4, 4)) + ) + assertThat( + DateUtils.getWeekdaysInMonth(Timestamp(longMonth)), + equalTo(arrayOf(5, 5, 4, 4, 4, 4, 5)) + ) + } + @Test fun testGetToday() { setFixedLocalTime(FIXED_LOCAL_TIME)