Add paused download into SharedPreferences

chrome/android/javatests/src/org/chromium/chrome/browser/download/DownloadNotificationServiceTest.java

 // Copyright 2015 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.download;
 
 import android.content.Context;
 import android.content.Intent;
 import android.content.SharedPreferences;
 import android.os.Handler;
@@ skipping 125 matching lines @@
     @SmallTest
     @Feature({"Download"})
     public void testResumptionScheduledWithoutDownloadOperationIntent() throws Exception {
         MockDownloadResumptionScheduler scheduler = new MockDownloadResumptionScheduler(
                 getSystemContext().getApplicationContext());
         DownloadResumptionScheduler.setDownloadResumptionScheduler(scheduler);
         setupService();
         Set<String> notifications = new HashSet<String>();
         notifications.add(
                 new DownloadSharedPreferenceEntry(1, false, true, UUID.randomUUID().toString(),
-                        "test1", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
+                        "test1", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
                         .getSharedPreferenceString());
         SharedPreferences sharedPrefs = ContextUtils.getAppSharedPreferences();
         SharedPreferences.Editor editor = sharedPrefs.edit();
         editor.putStringSet(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
         editor.apply();
         startNotificationService();
         assertTrue(scheduler.mScheduled);
     }
 
@@ skipping 13 matching lines @@
             public void run() {
                 Intent intent = new Intent(getService(), MockDownloadNotificationService.class);
                 intent.setAction(DownloadNotificationService.ACTION_DOWNLOAD_RESUME_ALL);
                 startService(intent);
             }
         });
         assertFalse(scheduler.mScheduled);
     }
 
     /**
+     * Tests that download resumption task is not scheduled when there is no auto resumable
+     * download in SharedPreferences.
+     */
+    @SmallTest
+    @Feature({"Download"})
+    public void testResumptionNotScheduledWithoutAutoResumableDownload() throws Exception {
+        MockDownloadResumptionScheduler scheduler = new MockDownloadResumptionScheduler(
+                getSystemContext().getApplicationContext());
+        DownloadResumptionScheduler.setDownloadResumptionScheduler(scheduler);
+        setupService();
+        Set<String> notifications = new HashSet<String>();

[gone, 2016/11/29 23:58:37]

new HashSet<>()

[qinmin, 2016/11/30 00:44:48]

Done.

+        notifications.add(
+                new DownloadSharedPreferenceEntry(1, false, true, UUID.randomUUID().toString(),
+                        "test1", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, false)
+                        .getSharedPreferenceString());
+        SharedPreferences sharedPrefs = ContextUtils.getAppSharedPreferences();
+        SharedPreferences.Editor editor = sharedPrefs.edit();
+        editor.putStringSet(
+                DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
+        editor.apply();
+        startNotificationService();
+        assertTrue(scheduler.mScheduled);
+    }
+
+    /**
      * Tests that creating the service without launching chrome will pause all ongoing downloads.
      */
     @SmallTest
     @Feature({"Download"})
     public void testPausingWithOngoingDownloads() {
         setupService();
         Context mockContext = new AdvancedMockContext(getSystemContext());
         getService().setContext(mockContext);
         Set<String> notifications = new HashSet<String>();
         notifications.add(
                 new DownloadSharedPreferenceEntry(1, false, true, UUID.randomUUID().toString(),
-                        "test1", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
+                        "test1", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
                         .getSharedPreferenceString());
         notifications.add(
                 new DownloadSharedPreferenceEntry(2, false, true, UUID.randomUUID().toString(),
-                        "test2", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
+                        "test2", DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
                         .getSharedPreferenceString());
         SharedPreferences sharedPrefs =
                 ContextUtils.getAppSharedPreferences();
         SharedPreferences.Editor editor = sharedPrefs.edit();
         editor.putStringSet(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
         editor.apply();
         startNotificationService();
         assertTrue(getService().isPaused());
         assertEquals(2, getService().getNotificationIds().size());
         assertTrue(getService().getNotificationIds().contains(1));
         assertTrue(getService().getNotificationIds().contains(2));
         assertTrue(
                 sharedPrefs.contains(DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS));
     }
 
     /**
      * Tests adding and cancelling notifications.
      */
     @SmallTest
     @Feature({"Download"})
     public void testAddingAndCancelingNotifications() {
         setupService();
         Context mockContext = new AdvancedMockContext(getSystemContext());
         getService().setContext(mockContext);
         Set<String> notifications = new HashSet<String>();
         String guid1 = UUID.randomUUID().toString();
         notifications.add(new DownloadSharedPreferenceEntry(3, false, true, guid1, "success",
-                                  DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
-                                  .getSharedPreferenceString());
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
+                        .getSharedPreferenceString());
         String guid2 = UUID.randomUUID().toString();
         notifications.add(new DownloadSharedPreferenceEntry(4, false, true, guid2, "failed",
-                                  DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
-                                  .getSharedPreferenceString());
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
+                        .getSharedPreferenceString());
         SharedPreferences sharedPrefs = ContextUtils.getAppSharedPreferences();
         SharedPreferences.Editor editor = sharedPrefs.edit();
         editor.putStringSet(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
         editor.apply();
         startNotificationService();
         assertEquals(2, getService().getNotificationIds().size());
         assertTrue(getService().getNotificationIds().contains(3));
         assertTrue(getService().getNotificationIds().contains(4));
 
@@ skipping 30 matching lines @@
     public void testDownloadSuccessNotification() {
         setupService();
         startNotificationService();
         DownloadNotificationService service = bindNotificationService();
         String guid = UUID.randomUUID().toString();
         service.notifyDownloadSuccessful(guid, "/path/to/test", "test", 100L, false, false);
         assertEquals(1, getService().getNotificationIds().size());
     }
 
     /**
-     * Tests resume all pending downloads.
+     * Tests resume all pending downloads. Only auto resumable downloads can resume.
      */
     @SmallTest
     @Feature({"Download"})
     public void testResumeAllPendingDownloads() throws Exception {
         setupService();
         Context mockContext = new AdvancedMockContext(getSystemContext());
         getService().setContext(mockContext);
         Set<String> notifications = new HashSet<String>();
         String guid1 = UUID.randomUUID().toString();
         notifications.add(new DownloadSharedPreferenceEntry(3, false, false, guid1, "success",
-                                  DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
-                                  .getSharedPreferenceString());
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
+                        .getSharedPreferenceString());
         String guid2 = UUID.randomUUID().toString();
         notifications.add(new DownloadSharedPreferenceEntry(4, false, true, guid2, "failed",
-                                  DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD)
-                                  .getSharedPreferenceString());
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
+                        .getSharedPreferenceString());
+        String guid3 = UUID.randomUUID().toString();
+        notifications.add(new DownloadSharedPreferenceEntry(5, false, true, guid3, "nonresumable",
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, false)
+                        .getSharedPreferenceString());
         SharedPreferences sharedPrefs = ContextUtils.getAppSharedPreferences();
         SharedPreferences.Editor editor = sharedPrefs.edit();
         editor.putStringSet(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
         editor.apply();
         startNotificationService();
         DownloadNotificationService service = bindNotificationService();
         DownloadManagerService.disableNetworkListenerForTest();
 
         final MockDownloadManagerService manager =
@@ skipping 21 matching lines @@
      */
     @SmallTest
     @Feature({"Download"})
     public void testIncognitoDownloadCanceledOnBrowserKill() throws Exception {
         setupService();
         Context mockContext = new AdvancedMockContext(getSystemContext());
         getService().setContext(mockContext);
         Set<String> notifications = new HashSet<String>();
         String uuid = UUID.randomUUID().toString();
         notifications.add(new DownloadSharedPreferenceEntry(1, true, true, uuid, "test1",
-                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD).getSharedPreferenceString());
+                DownloadSharedPreferenceEntry.ITEM_TYPE_DOWNLOAD, true)
+                        .getSharedPreferenceString());
         SharedPreferences sharedPrefs =
                 ContextUtils.getAppSharedPreferences();
         SharedPreferences.Editor editor = sharedPrefs.edit();
         editor.putStringSet(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS, notifications);
         editor.apply();
         startNotificationService();
         assertTrue(getService().isPaused());
         assertFalse(sharedPrefs.contains(
                 DownloadNotificationService.PENDING_DOWNLOAD_NOTIFICATIONS));
@@ skipping 19 matching lines @@
         assertEquals("3 hours left", DownloadNotificationService.formatRemainingTime(context,
                 150 * DownloadNotificationService.SECONDS_PER_MINUTE * MILLIS_PER_SECOND));
         assertEquals("1 day left", DownloadNotificationService.formatRemainingTime(context,
                 DownloadNotificationService.SECONDS_PER_DAY * MILLIS_PER_SECOND));
         assertEquals("2 days left", DownloadNotificationService.formatRemainingTime(context,
                 59 * DownloadNotificationService.SECONDS_PER_HOUR * MILLIS_PER_SECOND));
         assertEquals("3 days left", DownloadNotificationService.formatRemainingTime(context,
                 60 * DownloadNotificationService.SECONDS_PER_HOUR * MILLIS_PER_SECOND));
     }
 }