Fix an issue that notification will disappear when pausing a download on M and L

on M and L, detaching a notification from foreground service is not supported.
As a result, the current implementation just put the notification in a
sharedPreference and persist it across service restarts.
However, the recent introduction of stopSelf() will remove the notification,
and service will not restart.
The logic for M and L really complicates the implementation, and the only side
effect it tries to save is the notification flicker when pausing a download.
This CL removes the notification persisting logic for L and M. When stopping
the foreground service, we simply kill the existing notification,
and launch a new one.

BUG=919026

Change-Id: Ie9df5d17714d7ff9c54eed8f9b3ca1ddfaaf3199
Reviewed-on: https://chromium-review.googlesource.com/c/1432875Reviewed-by: David Trainor <dtrainor@chromium.org>
Reviewed-by: Xing Liu <xingliu@chromium.org>
Commit-Queue: Min Qin <qinmin@chromium.org>
Cr-Commit-Position: refs/heads/master@{#625803}

chrome/android/java/src/org/chromium/chrome/browser/download/DownloadForegroundService.java

@@ -11,7 +11,6 @@ import android.app.NotificationManager;
 import android.app.Service;
 import android.content.Context;
 import android.content.Intent;
-import android.content.SharedPreferences;
 import android.os.Binder;
 import android.os.Build;
 import android.os.IBinder;
@@ -32,18 +31,17 @@ import java.lang.annotation.RetentionPolicy;
  */
 public class DownloadForegroundService extends Service {
     private static final String TAG = "DownloadFg";
+    // Deprecated, remove this after M75.
     private static final String KEY_PERSISTED_NOTIFICATION_ID = "PersistedNotificationId";
     private final IBinder mBinder = new LocalBinder();
 
     private NotificationManager mNotificationManager;
 
-    @IntDef({StopForegroundNotification.KILL, StopForegroundNotification.DETACH_OR_PERSIST,
-            StopForegroundNotification.DETACH_OR_ADJUST})
+    @IntDef({StopForegroundNotification.KILL, StopForegroundNotification.DETACH})
     @Retention(RetentionPolicy.SOURCE)
     public @interface StopForegroundNotification {
         int KILL = 0; // Kill notification regardless of ability to detach.
-        int DETACH_OR_PERSIST = 1; // Try to detach, otherwise persist info.
-        int DETACH_OR_ADJUST = 2; // Try detach, otherwise kill and relaunch.
+        int DETACH = 1; // Try to detach, otherwise kill and relaunch.
     }
 
     @Override
@@ -53,6 +51,7 @@ public class DownloadForegroundService extends Service {
         mNotificationManager =
                 (NotificationManager) ContextUtils.getApplicationContext().getSystemService(
                         Context.NOTIFICATION_SERVICE);
+        clearPersistedNotificationId();
     }
 
     /**
@@ -123,11 +122,8 @@ public class DownloadForegroundService extends Service {
      *                                      and would need adjustment.
      * @param pinnedNotification            The actual notification that is pinned to the foreground
      *                                      and would need adjustment.
-     * @return                              Whether the notification was handled properly (ie. if it
-     *                                      was detached or killed as intended). If this is false,
-     *                                      the notification will be persisted and handled later.
      */
-    public boolean stopDownloadForegroundService(
+    public void stopDownloadForegroundService(
             @StopForegroundNotification int stopForegroundNotification, int pinnedNotificationId,
             Notification pinnedNotification) {
         Log.w(TAG,
@@ -140,7 +136,6 @@ public class DownloadForegroundService extends Service {
                 DownloadNotificationUmaHelper.ServiceStopped.STOPPED, true /* withForeground */);
 
         // Handle notifications and stop foreground.
-        boolean notificationHandledProperly = true;
         if (stopForegroundNotification == StopForegroundNotification.KILL) {
             // Regardless of the SDK level, stop foreground and kill if so indicated.
             stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_REMOVE);
@@ -152,40 +147,19 @@ public class DownloadForegroundService extends Service {
             } else if (getCurrentSdk() >= 23) {
                 // Android M+ can't detach the notification but doesn't have other caveats. Kill the
                 // notification and relaunch if detach was desired.
-                if (stopForegroundNotification == StopForegroundNotification.DETACH_OR_ADJUST) {
-                    stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_REMOVE);
-                    relaunchOldNotification(pinnedNotificationId, pinnedNotification);
-                } else {
-                    updatePersistedNotificationId(pinnedNotificationId);
-                    stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_DETACH);
-                    notificationHandledProperly = false;
-                }
-
-            } else if (getCurrentSdk() >= 21) {
+                stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_REMOVE);
+                relaunchOldNotification(pinnedNotificationId, pinnedNotification);
+            } else {
                 // In phones that are Lollipop and older (API < 23), the service gets killed with
                 // the task, which might result in the notification being unable to be relaunched
-                // where it needs to be. Relaunch the old notification before stopping the service.
-                if (stopForegroundNotification == StopForegroundNotification.DETACH_OR_ADJUST) {
-                    // Clear pinned id in case the service stops before we can do this properly.
-                    relaunchOldNotification(
-                            getNewNotificationIdFor(pinnedNotificationId), pinnedNotification);
-                    stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_REMOVE);
-                } else {
-                    updatePersistedNotificationId(pinnedNotificationId);
-                    stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_DETACH);
-                    notificationHandledProperly = false;
-                }
-
-            } else {
-                // For pre-Lollipop phones (API < 21), we need to kill all notifications because
-                // otherwise the notification gets stuck in the ongoing state.
+                // where it needs to be. kill and relaunch the old notification before stopping the
+                // service.
                 relaunchOldNotification(
                         getNewNotificationIdFor(pinnedNotificationId), pinnedNotification);
                 stopForegroundInternal(ServiceCompat.STOP_FOREGROUND_REMOVE);
             }
         }
         stopSelf();
-        return notificationHandledProperly;
     }
 
     @VisibleForTesting
@@ -202,15 +176,6 @@ public class DownloadForegroundService extends Service {
             DownloadNotificationUmaHelper.recordServiceStoppedHistogram(
                     DownloadNotificationUmaHelper.ServiceStopped.START_STICKY, true);
 
-            // Alert observers that the service restarted with null intent.
-            // Pass along the id of the notification that was pinned to the service when it died so
-            // that the observers can do any corrections (ie. relaunch notification) if needed.
-            int persistedNotificationId = getPersistedNotificationId();
-            Log.w(TAG, "onStartCommand intent: " + null + ", id: " + persistedNotificationId);
-            DownloadForegroundServiceObservers.alertObserversServiceRestarted(
-                    persistedNotificationId);
-            clearPersistedNotificationId();
-
             // Allow observers to restart service on their own, if needed.
             stopSelf();
         }
@@ -257,28 +222,6 @@ public class DownloadForegroundService extends Service {
         }
     }
 
-    /**
-     * Get stored value for the id of the notification pinned to the service.
-     * This has to be persisted in the case that the service dies and the notification dies with it.
-     * @return Id of the notification pinned to the service.
-     */
-    @VisibleForTesting
-    static int getPersistedNotificationId() {
-        SharedPreferences sharedPrefs = ContextUtils.getAppSharedPreferences();
-        return sharedPrefs.getInt(KEY_PERSISTED_NOTIFICATION_ID, INVALID_NOTIFICATION_ID);
-    }
-
-    /**
-     * Set stored value for the id of the notification pinned to the service.
-     * This has to be persisted in the case that the service dies and the notification dies with it.
-     * @param pinnedNotificationId Id of the notification pinned to the service.
-     */
-    private static void updatePersistedNotificationId(int pinnedNotificationId) {
-        ContextUtils.getAppSharedPreferences()
-                .edit()
-                .putInt(KEY_PERSISTED_NOTIFICATION_ID, pinnedNotificationId)
-                .apply();
-    }
 
     /**
      * Clear stored value for the id of the notification pinned to the service.

chrome/android/java/src/org/chromium/chrome/browser/download/DownloadForegroundServiceManager.java

@@ -274,41 +274,33 @@ public class DownloadForegroundServiceManager {
         int stopForegroundNotification;
         if (downloadStatus == DownloadNotificationService.DownloadStatus.CANCELLED) {
             stopForegroundNotification = DownloadForegroundService.StopForegroundNotification.KILL;
-        } else if (downloadStatus == DownloadNotificationService.DownloadStatus.PAUSED) {
-            stopForegroundNotification =
-                    DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST;
         } else {
             stopForegroundNotification =
-                    DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST;
+                    DownloadForegroundService.StopForegroundNotification.DETACH;
         }
 
         DownloadUpdate downloadUpdate = mDownloadUpdateQueue.get(mPinnedNotificationId);
         Notification oldNotification =
                 (downloadUpdate == null) ? null : downloadUpdate.mNotification;
 
-        boolean notificationHandledProperly = stopAndUnbindServiceInternal(
+        stopAndUnbindServiceInternal(
                 stopForegroundNotification, mPinnedNotificationId, oldNotification);
 
         mBoundService = null;
 
-        // Only if the notification was handled properly (ie. detached or killed), reset stored ID.
-        if (notificationHandledProperly) mPinnedNotificationId = INVALID_NOTIFICATION_ID;
+        mPinnedNotificationId = INVALID_NOTIFICATION_ID;
     }
 
     @VisibleForTesting
-    boolean stopAndUnbindServiceInternal(
+    void stopAndUnbindServiceInternal(
             @DownloadForegroundService.StopForegroundNotification int stopForegroundStatus,
             int pinnedNotificationId, Notification pinnedNotification) {
-        boolean notificationHandledProperly = mBoundService.stopDownloadForegroundService(
+        mBoundService.stopDownloadForegroundService(
                 stopForegroundStatus, pinnedNotificationId, pinnedNotification);
         ContextUtils.getApplicationContext().unbindService(mConnection);
 
-        if (notificationHandledProperly) {
-            DownloadForegroundServiceObservers.removeObserver(
-                    DownloadNotificationServiceObserver.class);
-        }
-
-        return notificationHandledProperly;
+        DownloadForegroundServiceObservers.removeObserver(
+                DownloadNotificationServiceObserver.class);
     }
 
     /** Helper code for testing. */

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

@@ -71,11 +71,10 @@ public final class DownloadForegroundServiceManagerTest {
         }
 
         @Override
-        boolean stopAndUnbindServiceInternal(@DownloadForegroundService.StopForegroundNotification
-                                             int stopForegroundNotification,
+        void stopAndUnbindServiceInternal(@DownloadForegroundService.StopForegroundNotification
+                                          int stopForegroundNotification,
                 int pinnedNotificationId, Notification pinnedNotification) {
             mStopForegroundNotificationFlag = stopForegroundNotification;
-            return true;
         }
 
         @Override
@@ -231,7 +230,7 @@ public final class DownloadForegroundServiceManagerTest {
         mDownloadServiceManager.updateDownloadStatus(mContext,
                 DownloadNotificationService.DownloadStatus.PAUSED, FAKE_DOWNLOAD_1, mNotification);
         assertFalse(mDownloadServiceManager.mIsServiceBound);
-        assertEquals(DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST,
+        assertEquals(DownloadForegroundService.StopForegroundNotification.DETACH,
                 mDownloadServiceManager.mStopForegroundNotificationFlag);
 
         // Service restarts and then is cancelled, so notification is killed.
@@ -259,7 +258,7 @@ public final class DownloadForegroundServiceManagerTest {
                 DownloadNotificationService.DownloadStatus.COMPLETED, FAKE_DOWNLOAD_2,
                 mNotification);
         assertFalse(mDownloadServiceManager.mIsServiceBound);
-        assertEquals(DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST,
+        assertEquals(DownloadForegroundService.StopForegroundNotification.DETACH,
                 mDownloadServiceManager.mStopForegroundNotificationFlag);
     }
 

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

@@ -8,11 +8,8 @@ import static android.app.Service.STOP_FOREGROUND_DETACH;
 import static android.app.Service.STOP_FOREGROUND_REMOVE;
 
 import static junit.framework.Assert.assertEquals;
-import static junit.framework.Assert.assertFalse;
-import static junit.framework.Assert.assertTrue;
 
 import static org.chromium.chrome.browser.download.DownloadForegroundService.clearPersistedNotificationId;
-import static org.chromium.chrome.browser.download.DownloadForegroundService.getPersistedNotificationId;
 import static org.chromium.chrome.browser.download.DownloadSnackbarController.INVALID_NOTIFICATION_ID;
 
 import android.app.Notification;
@@ -223,36 +220,33 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        boolean isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST,
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH,
                 INVALID_NOTIFICATION_ID, null);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(STOP_FOREGROUND_DETACH, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
 
         // When the service gets stopped with request to detach and kill (complete/failed).
         mForegroundService.startOrUpdateForegroundService(
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST,
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH,
                 INVALID_NOTIFICATION_ID, null);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(STOP_FOREGROUND_DETACH, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
 
         // When the service gets stopped with request to not detach but to kill (cancel).
         mForegroundService.startOrUpdateForegroundService(
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
+        mForegroundService.stopDownloadForegroundService(
                 DownloadForegroundService.StopForegroundNotification.KILL, INVALID_NOTIFICATION_ID,
                 null);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
     }
 
     /**
@@ -272,15 +266,14 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        boolean isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         List<Integer> expectedMethodCalls =
-                Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
+                Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS,
+                        MockDownloadForegroundService.MethodID.RELAUNCH_NOTIFICATION);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
-        assertEquals(ServiceCompat.STOP_FOREGROUND_DETACH, mForegroundService.mStopForegroundFlags);
-        assertFalse(isNotificationHandledProperly);
-        assertEquals(FAKE_DOWNLOAD_ID1, getPersistedNotificationId());
+        assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
 
         mForegroundService.clearStoredState();
 
@@ -289,15 +282,14 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS,
                         MockDownloadForegroundService.MethodID.RELAUNCH_NOTIFICATION);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
         assertEquals(FAKE_DOWNLOAD_ID1, mForegroundService.mRelaunchedNotificationId);
 
         // When the service gets stopped with request to not detach but to kill (cancel).
@@ -305,14 +297,13 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
+        mForegroundService.stopDownloadForegroundService(
                 DownloadForegroundService.StopForegroundNotification.KILL, FAKE_DOWNLOAD_ID1,
                 mNotification);
         expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
     }
 
     /**
@@ -331,15 +322,14 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        boolean isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         List<Integer> expectedMethodCalls =
-                Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
+                Arrays.asList(MockDownloadForegroundService.MethodID.RELAUNCH_NOTIFICATION,
+                        MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
-        assertEquals(ServiceCompat.STOP_FOREGROUND_DETACH, mForegroundService.mStopForegroundFlags);
-        assertFalse(isNotificationHandledProperly);
-        assertEquals(FAKE_DOWNLOAD_ID1, getPersistedNotificationId());
+        assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
 
         // When the service gets stopped with request to detach and kill (complete/failed).
         mForegroundService.startOrUpdateForegroundService(
@@ -347,15 +337,14 @@ public class DownloadForegroundServiceTest {
         mForegroundService.clearStoredState();
 
         mForegroundService.mNextNotificationId = FAKE_DOWNLOAD_ID2;
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.RELAUNCH_NOTIFICATION,
                         MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
         assertEquals(mForegroundService.mNextNotificationId,
                 mForegroundService.mRelaunchedNotificationId);
 
@@ -364,14 +353,13 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
+        mForegroundService.stopDownloadForegroundService(
                 DownloadForegroundService.StopForegroundNotification.KILL, FAKE_DOWNLOAD_ID1,
                 mNotification);
         expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
     }
 
     /**
@@ -389,15 +377,14 @@ public class DownloadForegroundServiceTest {
         mForegroundService.clearStoredState();
 
         mForegroundService.mNextNotificationId = FAKE_DOWNLOAD_ID2;
-        boolean isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_PERSIST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         List<Integer> expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.RELAUNCH_NOTIFICATION,
                         MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
         assertEquals(mForegroundService.mNextNotificationId,
                 mForegroundService.mRelaunchedNotificationId);
 
@@ -407,12 +394,11 @@ public class DownloadForegroundServiceTest {
         mForegroundService.clearStoredState();
 
         mForegroundService.mNextNotificationId = FAKE_DOWNLOAD_ID2;
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
-                DownloadForegroundService.StopForegroundNotification.DETACH_OR_ADJUST,
-                FAKE_DOWNLOAD_ID1, mNotification);
+        mForegroundService.stopDownloadForegroundService(
+                DownloadForegroundService.StopForegroundNotification.DETACH, FAKE_DOWNLOAD_ID1,
+                mNotification);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
         assertEquals(mForegroundService.mNextNotificationId,
                 mForegroundService.mRelaunchedNotificationId);
 
@@ -421,13 +407,12 @@ public class DownloadForegroundServiceTest {
                 FAKE_DOWNLOAD_ID1, mNotification, INVALID_NOTIFICATION_ID, null, false);
         mForegroundService.clearStoredState();
 
-        isNotificationHandledProperly = mForegroundService.stopDownloadForegroundService(
+        mForegroundService.stopDownloadForegroundService(
                 DownloadForegroundService.StopForegroundNotification.KILL, FAKE_DOWNLOAD_ID1,
                 mNotification);
         expectedMethodCalls =
                 Arrays.asList(MockDownloadForegroundService.MethodID.STOP_FOREGROUND_FLAGS);
         assertEquals(expectedMethodCalls, mForegroundService.mMethodCalls);
         assertEquals(ServiceCompat.STOP_FOREGROUND_REMOVE, mForegroundService.mStopForegroundFlags);
-        assertTrue(isNotificationHandledProperly);
     }
 }