Move SharedPreference for download into DownloadNotificationService

chrome/android/javatests/src/org/chromium/chrome/browser/download/SystemDownloadNotifierTest.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.test.InstrumentationTestCase;
 import android.test.suitebuilder.annotation.SmallTest;
 
 import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.AdvancedMockContext;
 import org.chromium.base.test.util.Feature;
 import org.chromium.content.browser.DownloadInfo;
 
 /**
  * Tests of {@link SystemDownloadNotifier}.
  */
 public class SystemDownloadNotifierTest extends InstrumentationTestCase {
     private MockSystemDownloadNotifier mDownloadNotifier;
     private MockDownloadNotificationService mService;
 
@@ skipping 22 matching lines @@
     }
 
     /**
      * Helper method to simulate that the DownloadNotificationService is connected.
      */
     private void onServiceConnected() {
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 mService = new MockDownloadNotificationService();
-                mService.setContext(
-                        getInstrumentation().getTargetContext().getApplicationContext());
+                mService.setContext(new AdvancedMockContext(
+                        getInstrumentation().getTargetContext().getApplicationContext()));
                 mService.onCreate();
             }
         });
         mDownloadNotifier.setDownloadNotificationService(mService);
         mDownloadNotifier.handlePendingNotifications();
     }
 
     /**
      * Tests that pending notifications will be handled after service is connected.
      */
@@ skipping 21 matching lines @@
         assertTrue(mDownloadNotifier.mStarted);
         DownloadInfo info2 = new DownloadInfo.Builder().setDownloadId(2).build();
         mDownloadNotifier.notifyDownloadProgress(info2, 1L);
 
         mDownloadNotifier.notifyDownloadFailed(info);
         assertTrue(mDownloadNotifier.mStarted);
         mDownloadNotifier.notifyDownloadSuccessful(info2, null);
         assertFalse(mDownloadNotifier.mStarted);
     }
 }