Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(90)

Unified Diff: components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java

Issue 2819703002: [Android] Implements OS upgrade check and rescheduling (Closed)
Patch Set: Calling upgrade task from DeferredStartupHandler Created 3 years, 8 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java
diff --git a/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java b/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java
index 1bb216c67622851ebad9ee91f8204646cb00c11d..5cb37e6521f01dc643ff64d03634aefee457356a 100644
--- a/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java
+++ b/components/background_task_scheduler/android/junit/src/org/chromium/components/background_task_scheduler/ShadowGcmNetworkManager.java
@@ -11,6 +11,9 @@ import com.google.android.gms.gcm.Task;
import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;
+import java.util.HashSet;
+import java.util.Set;
+
/**
* Custom shadow for the OS's GcmNetworkManager. We use this to hook the call to GcmNetworkManager
* to make sure it was invoked as we expect.
@@ -19,6 +22,11 @@ import org.robolectric.annotation.Implements;
public class ShadowGcmNetworkManager {
private Task mTask;
private Task mCanceledTask;
+ private Set<String> mCanceledTaskTags;
+
+ public ShadowGcmNetworkManager() {
+ mCanceledTaskTags = new HashSet<>();
+ }
@Implementation
public void schedule(Task task) {
@@ -33,6 +41,7 @@ public class ShadowGcmNetworkManager {
mCanceledTask = mTask;
mTask = null;
}
+ mCanceledTaskTags.add(tag);
}
public Task getScheduledTask() {
@@ -43,8 +52,13 @@ public class ShadowGcmNetworkManager {
return mCanceledTask;
}
+ public Set<String> getCanceledTaskTags() {
+ return mCanceledTaskTags;
+ }
+
public void clear() {
mTask = null;
mCanceledTask = null;
+ mCanceledTaskTags = new HashSet<>();
}
}

Powered by Google App Engine
This is Rietveld 408576698