Home | History | Annotate | Download | only in greedy 
      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 
     17 package androidx.work.impl.background.greedy;
     18 
     19 import static org.mockito.Mockito.mock;
     20 import static org.mockito.Mockito.never;
     21 import static org.mockito.Mockito.verify;
     22 import static org.mockito.Mockito.when;
     23 
     24 import android.support.test.filters.SmallTest;
     25 import android.support.test.runner.AndroidJUnit4;
     26 
     27 import androidx.work.OneTimeWorkRequest;
     28 import androidx.work.PeriodicWorkRequest;
     29 import androidx.work.WorkManagerTest;
     30 import androidx.work.impl.Processor;
     31 import androidx.work.impl.WorkManagerImpl;
     32 import androidx.work.impl.constraints.WorkConstraintsTracker;
     33 import androidx.work.impl.model.WorkSpec;
     34 import androidx.work.worker.TestWorker;
     35 
     36 import org.junit.Before;
     37 import org.junit.Test;
     38 import org.junit.runner.RunWith;
     39 import org.mockito.ArgumentMatchers;
     40 
     41 import java.util.Collections;
     42 import java.util.concurrent.TimeUnit;
     43 
     44 
     45 @RunWith(AndroidJUnit4.class)
     46 public class GreedySchedulerTest extends WorkManagerTest {
     47 
     48     private static final String TEST_ID = "test";
     49 
     50     private WorkManagerImpl mWorkManagerImpl;
     51     private WorkConstraintsTracker mMockWorkConstraintsTracker;
     52     private GreedyScheduler mGreedyScheduler;
     53 
     54     @Before
     55     public void setUp() {
     56         mWorkManagerImpl = mock(WorkManagerImpl.class);
     57         Processor processor = mock(Processor.class);
     58         mMockWorkConstraintsTracker = mock(WorkConstraintsTracker.class);
     59         when(mWorkManagerImpl.getProcessor()).thenReturn(processor);
     60         mGreedyScheduler = new GreedyScheduler(mWorkManagerImpl, mMockWorkConstraintsTracker);
     61     }
     62 
     63     @Test
     64     @SmallTest
     65     public void testGreedyScheduler_startsUnconstrainedWork() {
     66         OneTimeWorkRequest work = new OneTimeWorkRequest.Builder(TestWorker.class).build();
     67         WorkSpec workSpec = getWorkSpec(work);
     68         mGreedyScheduler.schedule(workSpec);
     69         verify(mWorkManagerImpl).startWork(workSpec.id);
     70     }
     71 
     72     @Test
     73     @SmallTest
     74     public void testGreedyScheduler_ignoresPeriodicWork() {
     75         PeriodicWorkRequest periodicWork =
     76                 new PeriodicWorkRequest.Builder(TestWorker.class, 0L, TimeUnit.MILLISECONDS)
     77                         .build();
     78         mGreedyScheduler.schedule(getWorkSpec(periodicWork));
     79         verify(mMockWorkConstraintsTracker, never()).replace(ArgumentMatchers.<WorkSpec>anyList());
     80     }
     81 
     82     @Test
     83     @SmallTest
     84     public void testGreedyScheduler_ignoresInitialDelayWork() {
     85         OneTimeWorkRequest work = new OneTimeWorkRequest.Builder(TestWorker.class)
     86                 .setInitialDelay(1000L, TimeUnit.MILLISECONDS)
     87                 .build();
     88         mGreedyScheduler.schedule(getWorkSpec(work));
     89         verify(mMockWorkConstraintsTracker, never()).replace(ArgumentMatchers.<WorkSpec>anyList());
     90     }
     91 
     92     @Test
     93     @SmallTest
     94     public void testGreedyScheduler_startsWorkWhenConstraintsMet() {
     95         mGreedyScheduler.onAllConstraintsMet(Collections.singletonList(TEST_ID));
     96         verify(mWorkManagerImpl).startWork(TEST_ID);
     97     }
     98 
     99     @Test
    100     @SmallTest
    101     public void testGreedyScheduler_stopsWorkWhenConstraintsNotMet() {
    102         mGreedyScheduler.onAllConstraintsNotMet(Collections.singletonList(TEST_ID));
    103         verify(mWorkManagerImpl).stopWork(TEST_ID);
    104     }
    105 }
    106