[Offline Prefetch] Backoff support for PrefetchBackgroundTask

chrome/android/junit/src/org/chromium/chrome/browser/offlinepages/prefetch/PrefetchBackgroundTaskTest.java

Index: chrome/android/junit/src/org/chromium/chrome/browser/offlinepages/prefetch/PrefetchBackgroundTaskTest.java
diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/offlinepages/prefetch/PrefetchBackgroundTaskTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/offlinepages/prefetch/PrefetchBackgroundTaskTest.java
deleted file mode 100644
index 14523810f4446b8737db18c429128b12e0e96945..0000000000000000000000000000000000000000
--- a/chrome/android/junit/src/org/chromium/chrome/browser/offlinepages/prefetch/PrefetchBackgroundTaskTest.java
+++ /dev/null
@@ -1,146 +0,0 @@
-// 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.chrome.browser.offlinepages.prefetch;
-
-import android.content.Context;
-
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertNull;
-import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.Mockito.doAnswer;
-import static org.mockito.Mockito.doNothing;
-import static org.mockito.Mockito.doReturn;
-import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.when;
-
-import org.junit.After;
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
-import org.mockito.MockitoAnnotations;
-import org.mockito.Spy;
-import org.mockito.invocation.InvocationOnMock;
-import org.mockito.stubbing.Answer;
-import org.robolectric.annotation.Config;
-import org.robolectric.annotation.Implementation;
-import org.robolectric.annotation.Implements;
-import org.robolectric.internal.ShadowExtractor;
-import org.robolectric.shadows.multidex.ShadowMultiDex;
-
-import org.chromium.components.background_task_scheduler.BackgroundTask.TaskFinishedCallback;
-import org.chromium.components.background_task_scheduler.BackgroundTaskScheduler;
-import org.chromium.components.background_task_scheduler.BackgroundTaskSchedulerFactory;
-import org.chromium.components.background_task_scheduler.TaskIds;
-import org.chromium.components.background_task_scheduler.TaskInfo;
-import org.chromium.components.background_task_scheduler.TaskParameters;
-import org.chromium.testing.local.LocalRobolectricTestRunner;
-
-import java.util.ArrayList;
-import java.util.HashMap;
-
-/** Unit tests for {@link PrefetchBackgroundTask}. */
-@RunWith(LocalRobolectricTestRunner.class)
-@Config(manifest = Config.NONE,
-        shadows = {PrefetchBackgroundTaskTest.ShadowBackgroundTaskScheduler.class,
-                ShadowMultiDex.class})
-public class PrefetchBackgroundTaskTest {
-    /**
-     * Shadow of BackgroundTaskScheduler system service.
-     */
-    @Implements(BackgroundTaskScheduler.class)
-    public static class ShadowBackgroundTaskScheduler {
-        private HashMap<Integer, TaskInfo> mTaskInfos = new HashMap<>();
-
-        @Implementation
-        public boolean schedule(Context context, TaskInfo taskInfo) {
-            mTaskInfos.put(taskInfo.getTaskId(), taskInfo);
-            return true;
-        }
-
-        @Implementation
-        public void cancel(Context context, int taskId) {
-            mTaskInfos.remove(taskId);
-        }
-
-        public TaskInfo getTaskInfo(int taskId) {
-            return mTaskInfos.get(taskId);
-        }
-
-        public void clear() {
-            mTaskInfos = new HashMap<>();
-        }
-    }
-
-    @Spy
-    private PrefetchBackgroundTask mPrefetchBackgroundTask = new PrefetchBackgroundTask(null);
-    private ShadowBackgroundTaskScheduler mShadowTaskScheduler;
-
-    @Before
-    public void setUp() {
-        MockitoAnnotations.initMocks(this);
-        doAnswer(new Answer() {
-            public Object answer(InvocationOnMock invocation) {
-                mPrefetchBackgroundTask.setNativeTask(1);
-                return Boolean.TRUE;
-            }
-        })
-                .when(mPrefetchBackgroundTask)
-                .nativeStartPrefetchTask(any());
-        doReturn(true).when(mPrefetchBackgroundTask).nativeOnStopTask(1);
-        doNothing().when(mPrefetchBackgroundTask).launchBrowserIfNecessary(any());
-
-        mShadowTaskScheduler = (ShadowBackgroundTaskScheduler) ShadowExtractor.extract(
-                BackgroundTaskSchedulerFactory.getScheduler());
-    }
-
-    @After
-    public void tearDown() {
-        mShadowTaskScheduler.clear();
-    }
-
-    @Test
-    public void scheduleTask() {
-        PrefetchBackgroundTask.scheduleTask();
-        TaskInfo scheduledTask =
-                mShadowTaskScheduler.getTaskInfo(TaskIds.OFFLINE_PAGES_PREFETCH_JOB_ID);
-        assertNotNull(scheduledTask);
-        assertEquals(true, scheduledTask.isPersisted());
-        assertEquals(TaskInfo.NETWORK_TYPE_UNMETERED, scheduledTask.getRequiredNetworkType());
-    }
-
-    @Test
-    public void cancelTask() {
-        TaskInfo scheduledTask =
-                mShadowTaskScheduler.getTaskInfo(TaskIds.OFFLINE_PAGES_PREFETCH_JOB_ID);
-        assertNull(scheduledTask);
-
-        PrefetchBackgroundTask.scheduleTask();
-        scheduledTask = mShadowTaskScheduler.getTaskInfo(TaskIds.OFFLINE_PAGES_PREFETCH_JOB_ID);
-        assertNotNull(scheduledTask);
-
-        PrefetchBackgroundTask.cancelTask();
-        scheduledTask = mShadowTaskScheduler.getTaskInfo(TaskIds.OFFLINE_PAGES_PREFETCH_JOB_ID);
-        assertNull(scheduledTask);
-    }
-
-    @Test
-    public void createNativeTask() {
-        final ArrayList<Boolean> reschedules = new ArrayList<>();
-        TaskParameters params = mock(TaskParameters.class);
-        when(params.getTaskId()).thenReturn(TaskIds.OFFLINE_PAGES_PREFETCH_JOB_ID);
-
-        mPrefetchBackgroundTask.onStartTask(null, params, new TaskFinishedCallback() {
-            @Override
-            public void taskFinished(boolean needsReschedule) {
-                reschedules.add(needsReschedule);
-            }
-        });
-        mPrefetchBackgroundTask.doneProcessing(false);
-
-        assertEquals(1, reschedules.size());
-        assertEquals(false, reschedules.get(0));
-    }
-}