1/*
2 * Copyright 2018 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package androidx.work
18
19import android.support.test.filters.SdkSuppress
20import android.support.test.filters.SmallTest
21import android.support.test.runner.AndroidJUnit4
22import androidx.work.workers.TestWorker
23import org.junit.Assert.assertEquals
24import org.junit.Test
25import org.junit.runner.RunWith
26import java.time.Duration
27import java.util.concurrent.TimeUnit
28
29@RunWith(AndroidJUnit4::class)
30@SmallTest
31class PeriodicWorkRequestTest {
32    @Test
33    fun testPeriodicWorkRequestBuilder() {
34        val builder = PeriodicWorkRequestBuilder<TestWorker>(
35                repeatInterval = 15L,
36                repeatIntervalTimeUnit = TimeUnit.MINUTES)
37        val workRequest = builder.build()
38        assertEquals(workRequest.workSpec.workerClassName, TestWorker::class.java.name)
39        assertEquals(workRequest.workSpec.isPeriodic, true)
40        assertEquals(workRequest.workSpec.intervalDuration, TimeUnit.MINUTES.toMillis(15))
41        assertEquals(workRequest.workSpec.flexDuration, TimeUnit.MINUTES.toMillis(15))
42    }
43
44    @Test
45    @SdkSuppress(minSdkVersion = 26)
46    fun testPeriodicWorkRequestBuilder_withDuration() {
47        val repeatInterval = Duration.ofDays(2).plusHours(3)
48        val builder = PeriodicWorkRequestBuilder<TestWorker>(repeatInterval)
49        val workRequest = builder.build()
50        assertEquals(workRequest.workSpec.workerClassName, TestWorker::class.java.name)
51        assertEquals(workRequest.workSpec.isPeriodic, true)
52        assertEquals(workRequest.workSpec.intervalDuration, repeatInterval.toMillis())
53        assertEquals(workRequest.workSpec.flexDuration, repeatInterval.toMillis())
54    }
55
56    @Test
57    fun testPeriodicWorkRequestBuilder_withFlexTime() {
58        val builder = PeriodicWorkRequestBuilder<TestWorker>(
59                repeatInterval = 15L,
60                repeatIntervalTimeUnit = TimeUnit.MINUTES,
61                flexTimeInterval = 10L,
62                flexTimeIntervalUnit = TimeUnit.MINUTES)
63        val workRequest = builder.build()
64        assertEquals(workRequest.workSpec.workerClassName, TestWorker::class.java.name)
65        assertEquals(workRequest.workSpec.isPeriodic, true)
66        assertEquals(workRequest.workSpec.flexDuration, TimeUnit.MINUTES.toMillis(10))
67    }
68
69    @Test
70    @SdkSuppress(minSdkVersion = 26)
71    fun testPeriodicWorkRequestBuilder_withFlexTimeandDuration() {
72        val repeatInterval = Duration.ofHours(3).plusMinutes(25)
73        val flexInterval = repeatInterval.minusMinutes(15)
74        val builder = PeriodicWorkRequestBuilder<TestWorker>(
75                repeatInterval = repeatInterval,
76                flexTimeInterval = flexInterval)
77        val workRequest = builder.build()
78        assertEquals(workRequest.workSpec.workerClassName, TestWorker::class.java.name)
79        assertEquals(workRequest.workSpec.isPeriodic, true)
80        assertEquals(workRequest.workSpec.intervalDuration, repeatInterval.toMillis())
81        assertEquals(workRequest.workSpec.flexDuration, flexInterval.toMillis())
82    }
83}
84