[Android] Adding scheduling through GcmNetworkManager

components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/BackgroundTaskSchedulerPrefsTest.java

+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package org.chromium.components.background_task_scheduler;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.RuntimeEnvironment;
+import org.robolectric.annotation.Config;
+
+import org.chromium.base.ContextUtils;
+import org.chromium.base.test.util.Feature;
+import org.chromium.testing.local.LocalRobolectricTestRunner;
+
+import java.util.Set;
+import java.util.concurrent.TimeUnit;
+
+/** Unit tests for {@link BackgroundTaskSchedulerPrefs}. */
+@RunWith(LocalRobolectricTestRunner.class)
+@Config(manifest = Config.NONE)
+public class BackgroundTaskSchedulerPrefsTest {

[nyquist, 2017/04/18 21:58:28]

Given the format of how this is stored, could you also add a test for same
BackgroundTask.class, but multiple TaskIds?

[fgorski, 2017/04/20 05:54:16]

Done.

+    private static final TaskInfo TASK_1 =
+            TaskInfo.createOneOffTask(
+                            TaskIds.TEST, TestBackgroundTask.class, TimeUnit.DAYS.toMillis(1))
+                    .build();
+    private static final TaskInfo TASK_2 =
+            TaskInfo.createOneOffTask(TaskIds.OFFLINE_PAGES_BACKGROUND_JOB_ID,
+                            TestBackgroundTask2.class, TimeUnit.DAYS.toMillis(1))
+                    .build();
+
+    /**
+     * Dummy extension of the class above to ensure scheduled tasks returned from shared
+     * preferences are not missing something.
+     */
+    static class TestBackgroundTask2 extends TestBackgroundTask {}
+
+    @Before
+    public void setUp() {
+        ContextUtils.initApplicationContextForTests(RuntimeEnvironment.application);
+    }
+
+    @Test
+    @Feature({"BackgroundTaskScheduler"})
+    public void testAddScheduledTask() {
+        BackgroundTaskSchedulerPrefs.addScheduledTask(TASK_1);
+        assertEquals("We are expecting a single entry.", 1,
+                BackgroundTaskSchedulerPrefs.getScheduledTasks().size());
+
+        BackgroundTaskSchedulerPrefs.addScheduledTask(TASK_1);
+        assertEquals("Still there should be only one entry, as duplicate was added.", 1,
+                BackgroundTaskSchedulerPrefs.getScheduledTasks().size());
+
+        BackgroundTaskSchedulerPrefs.addScheduledTask(TASK_2);
+        assertEquals("There should be 2 tasks in shared prefs.", 2,
+                BackgroundTaskSchedulerPrefs.getScheduledTasks().size());
+
+        Set<String> scheduledTasks = BackgroundTaskSchedulerPrefs.getScheduledTasks();
+        assertTrue("TASK_1 class name in scheduled tasks.",
+                scheduledTasks.contains(TASK_1.getBackgroundTaskClass().getName()));
+        assertTrue("TASK_2 class name in scheduled tasks.",
+                scheduledTasks.contains(TASK_2.getBackgroundTaskClass().getName()));
+    }
+
+    @Test
+    @Feature("BackgroundTaskScheduler")
+    public void testRemoveScheduledTask() {
+        BackgroundTaskSchedulerPrefs.addScheduledTask(TASK_1);
+        BackgroundTaskSchedulerPrefs.addScheduledTask(TASK_2);
+        BackgroundTaskSchedulerPrefs.removeScheduledTask(TASK_1.getTaskId());
+        assertEquals("We are expecting a single entry.", 1,
+                BackgroundTaskSchedulerPrefs.getScheduledTasks().size());
+
+        BackgroundTaskSchedulerPrefs.removeScheduledTask(TASK_1.getTaskId());
+        assertEquals("Removing a task which is not there does not affect the task set.", 1,
+                BackgroundTaskSchedulerPrefs.getScheduledTasks().size());
+
+        Set<String> scheduledTasks = BackgroundTaskSchedulerPrefs.getScheduledTasks();
+        assertFalse("TASK_1 class name in scheduled tasks.",
+                scheduledTasks.contains(TASK_1.getBackgroundTaskClass().getName()));
+        assertTrue("TASK_2 class name in scheduled tasks.",
+                scheduledTasks.contains(TASK_2.getBackgroundTaskClass().getName()));
+    }
+}