Index: components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskGcmTaskServiceTest.java |
diff --git a/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskGcmTaskServiceTest.java b/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskGcmTaskServiceTest.java |
index 2497c7e2e6143eb4e88c479916ff1d6067d3f09a..36018c4f6c303632fcc75618e7e5a47a7231800c 100644 |
--- a/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskGcmTaskServiceTest.java |
+++ b/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskGcmTaskServiceTest.java |
@@ -8,6 +8,7 @@ import static org.junit.Assert.assertEquals; |
import static org.junit.Assert.assertNotNull; |
import android.content.Context; |
+import android.os.Build; |
import android.os.Bundle; |
import com.google.android.gms.gcm.GcmNetworkManager; |
@@ -20,6 +21,7 @@ import org.mockito.Mock; |
import org.mockito.MockitoAnnotations; |
import org.robolectric.RuntimeEnvironment; |
import org.robolectric.annotation.Config; |
+import org.robolectric.util.ReflectionHelpers; |
import org.chromium.base.ContextUtils; |
import org.chromium.base.test.util.Feature; |
@@ -121,18 +123,33 @@ public class BackgroundTaskGcmTaskServiceTest { |
@Test |
@Feature({"BackgroundTaskScheduler"}) |
- public void testOnInitializeTasks() { |
+ public void testOnInitializeTasksOnPreM() { |
+ ReflectionHelpers.setStaticField( |
+ Build.VERSION.class, "SDK_INT", Build.VERSION_CODES.LOLLIPOP); |
TaskInfo task = TaskInfo.createOneOffTask(TaskIds.TEST, TestBackgroundTask.class, |
TimeUnit.DAYS.toMillis(1)) |
.build(); |
BackgroundTaskSchedulerPrefs.addScheduledTask(task); |
assertEquals(0, TestBackgroundTask.getRescheduleCalls()); |
- BackgroundTaskGcmTaskService taskService = new BackgroundTaskGcmTaskService(); |
- taskService.onInitializeTasks(); |
+ new BackgroundTaskGcmTaskService().onInitializeTasks(); |
assertEquals(1, TestBackgroundTask.getRescheduleCalls()); |
} |
+ @Test |
+ @Feature({"BackgroundTaskScheduler"}) |
+ public void testOnInitializeTasksOnMPlus() { |
+ ReflectionHelpers.setStaticField(Build.VERSION.class, "SDK_INT", Build.VERSION_CODES.M); |
+ TaskInfo task = TaskInfo.createOneOffTask(TaskIds.TEST, TestBackgroundTask.class, |
+ TimeUnit.DAYS.toMillis(1)) |
+ .build(); |
+ BackgroundTaskSchedulerPrefs.addScheduledTask(task); |
+ assertEquals(0, TestBackgroundTask.getRescheduleCalls()); |
+ |
+ new BackgroundTaskGcmTaskService().onInitializeTasks(); |
+ assertEquals(0, TestBackgroundTask.getRescheduleCalls()); |
+ } |
+ |
private TaskParams buildTaskParams(Class clazz, Bundle taskExtras) { |
Bundle extras = new Bundle(); |
extras.putBundle( |