1 files changed, 7 insertions, 8 deletions
diff --git a/tests/src/com/android/providers/downloads/FakeSystemFacade.java b/tests/src/com/android/providers/downloads/FakeSystemFacade.java
index c48f815d..4ff313ab 100644
--- a/tests/src/com/android/providers/downloads/FakeSystemFacade.java
+++ b/tests/src/com/android/providers/downloads/FakeSystemFacade.java
@@ -2,12 +2,11 @@ package com.android.providers.downloads;
 
 import android.net.ConnectivityManager;
 
-import java.util.BitSet;
-
 public class FakeSystemFacade implements SystemFacade {
     long mTimeMillis = 0;
     Integer mActiveNetworkType = ConnectivityManager.TYPE_WIFI;
     boolean mIsRoaming = false;
+    Integer mMaxBytesOverMobile = null;
 
     void incrementTimeMillis(long delta) {
         mTimeMillis += delta;
@@ -17,15 +16,15 @@ public class FakeSystemFacade implements SystemFacade {
         return mTimeMillis;
     }
 
-    public BitSet getConnectedNetworkTypes() {
-        BitSet connectedTypes = new BitSet();
-        if (mActiveNetworkType != null) {
-            connectedTypes.set(mActiveNetworkType);
-        }
-        return connectedTypes;
+    public Integer getActiveNetworkType() {
+        return mActiveNetworkType;
     }
 
     public boolean isNetworkRoaming() {
         return mIsRoaming;
     }
+
+    public Integer getMaxBytesOverMobile() {
+        return mMaxBytesOverMobile ;
+    }
 }