DateUtilsTest: Simplify getCalendarUTC

2 years ago · 7f6248123c
parent 2024277ebe
commit 7f6248123c
1 changed files with 4 additions and 5 deletions
--- 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
@ -120,11 +120,10 @@ class DateUtilsTest : BaseUnitTest() {

    @Test
    fun getWeekdaysInMonth() {
-        fun getCalendarUTC(year: Int, month: Int, dayOfMonth: Int): GregorianCalendar {
-            val gregCalendar = GregorianCalendar(year, month, dayOfMonth)
-            gregCalendar.timeZone = TimeZone.getTimeZone("UTC")
-            return gregCalendar
-        }
+        fun getCalendarUTC(year: Int, month: Int, dayOfMonth: Int) =
+            GregorianCalendar(year, month, dayOfMonth).apply {
+                timeZone = TimeZone.getTimeZone("UTC")
+            }

        val february = getCalendarUTC(2018, Calendar.FEBRUARY, 1)
        val leapFebruary = getCalendarUTC(2020, Calendar.FEBRUARY, 1)