Move DownloadSharedPreferenceEntry handling into another class

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

Index: chrome/android/javatests/src/org/chromium/chrome/browser/download/MockDownloadNotificationService.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/download/MockDownloadNotificationService.java b/chrome/android/javatests/src/org/chromium/chrome/browser/download/MockDownloadNotificationService.java
index a7952e6d258ed8cb309cc73f46a4961cdf8314fa..237010cf2a4defef457e114cea7934f92a86458a 100644
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/download/MockDownloadNotificationService.java
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/download/MockDownloadNotificationService.java
@@ -7,8 +7,11 @@ package org.chromium.chrome.browser.download;
 import android.app.Notification;
 import android.content.Context;
 
+import org.chromium.base.ThreadUtils;
+
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.Callable;
 
 /**
  * Mock class to DownloadNotificationService for testing purpose.
@@ -59,5 +62,56 @@ public class MockDownloadNotificationService extends DownloadNotificationService
     public Context getApplicationContext() {
         return mContext == null ? super.getApplicationContext() : mContext;
     }
+
+    @Override
+    public int notifyDownloadSuccessful(
+            final String downloadGuid, final String filePath, final String fileName,
+            final long systemDownloadId, final boolean isOfflinePage,
+            final boolean isSupportedMimeType) {
+        return ThreadUtils.runOnUiThreadBlockingNoException(new Callable<Integer>() {
+            @Override
+            public Integer call() throws Exception {
+                return MockDownloadNotificationService.super.notifyDownloadSuccessful(
+                        downloadGuid, filePath, fileName, systemDownloadId, isOfflinePage,
+                        isSupportedMimeType);
+            }
+        });
+    }
+
+    @Override
+    public void notifyDownloadProgress(final String downloadGuid, final String fileName,
+            final int percentage, final long timeRemainingInMillis, final long startTime,
+            final boolean isOffTheRecord, final boolean canDownloadWhileMetered,
+            final boolean isOfflinePage) {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                MockDownloadNotificationService.super.notifyDownloadProgress(
+                        downloadGuid, fileName, percentage, timeRemainingInMillis, startTime,
+                        isOffTheRecord, canDownloadWhileMetered, isOfflinePage);
+            }
+        });
+    }
+
+    @Override
+    public void notifyDownloadFailed(final String downloadGuid, final String fileName) {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                MockDownloadNotificationService.super.notifyDownloadFailed(
+                        downloadGuid, fileName);
+            }
+        });
+    }
+
+    @Override
+    public void notifyDownloadCanceled(final String downloadGuid) {
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                MockDownloadNotificationService.super.notifyDownloadCanceled(downloadGuid);
+            }
+        });
+    }
 }