Update robolectric tetst for BackupEligibilityRules refactor

Bug: 161241479
Test: atest RunBackupFrameworksServicesRoboTests
Change-Id: I7705d441947b1e2143f5da3c298294f5a67377f0

3 files changed, 6 insertions, 83 deletions

diff --git a/services/robotests/src/com/android/server/testing/shadows/ShadowAppBackupUtils.java b/services/robotests/src/com/android/server/testing/shadows/ShadowAppBackupUtils.java
deleted file mode 100644
index aa1c6688b01d..000000000000
--- a/services/robotests/src/com/android/server/testing/shadows/ShadowAppBackupUtils.java
+++ /dev/null
@@ -1,81 +0,0 @@
-/*
- * Copyright (C) 2018 The Android Open Source Project
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- *      http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License
- */
-
-package com.android.server.testing.shadows;
-
-import android.annotation.Nullable;
-import android.content.pm.ApplicationInfo;
-import android.content.pm.PackageInfo;
-import android.content.pm.PackageManager;
-
-import com.android.server.backup.transport.TransportClient;
-import com.android.server.backup.utils.AppBackupUtils;
-
-import org.robolectric.annotation.Implementation;
-import org.robolectric.annotation.Implements;
-import org.robolectric.annotation.Resetter;
-
-import java.util.HashSet;
-import java.util.Set;
-
-@Implements(AppBackupUtils.class)
-public class ShadowAppBackupUtils {
-    private static final Set<String> sAppsRunningAndEligibleForBackupWithTransport =
-            new HashSet<>();
-    private static final Set<String> sAppsEligibleForBackup = new HashSet<>();
-    private static final Set<String> sAppsGetFullBackup = new HashSet<>();
-
-    public static void setAppRunningAndEligibleForBackupWithTransport(String packageName) {
-        sAppsEligibleForBackup.add(packageName);
-        sAppsRunningAndEligibleForBackupWithTransport.add(packageName);
-    }
-
-    public static void setAppEligibleForBackup(String packageName) {
-        sAppsEligibleForBackup.add(packageName);
-    }
-
-    /** By default the app will be key-value. */
-    public static void setAppGetsFullBackup(String packageName) {
-        sAppsGetFullBackup.add(packageName);
-    }
-
-    @Implementation
-    protected static boolean appIsRunningAndEligibleForBackupWithTransport(
-            @Nullable TransportClient transportClient,
-            String packageName,
-            PackageManager pm,
-            int userId) {
-        return sAppsRunningAndEligibleForBackupWithTransport.contains(packageName);
-    }
-
-    @Implementation
-    protected static boolean appIsEligibleForBackup(ApplicationInfo app, int userId,
-            int operationType) {
-        return sAppsEligibleForBackup.contains(app.packageName);
-    }
-
-    @Implementation
-    protected static boolean appGetsFullBackup(PackageInfo packageInfo, int operationType) {
-        return sAppsGetFullBackup.contains(packageInfo.packageName);
-    }
-
-    @Resetter
-    public static void reset() {
-        sAppsRunningAndEligibleForBackupWithTransport.clear();
-        sAppsEligibleForBackup.clear();
-        sAppsGetFullBackup.clear();
-    }
-}
diff --git a/services/robotests/src/com/android/server/testing/shadows/ShadowKeyValueBackupTask.java b/services/robotests/src/com/android/server/testing/shadows/ShadowKeyValueBackupTask.java
index ac5d2da9c3cf..fd51df7ab1f9 100644
--- a/services/robotests/src/com/android/server/testing/shadows/ShadowKeyValueBackupTask.java
+++ b/services/robotests/src/com/android/server/testing/shadows/ShadowKeyValueBackupTask.java
@@ -24,6 +24,7 @@ import com.android.server.backup.internal.OnTaskFinishedListener;
 import com.android.server.backup.keyvalue.KeyValueBackupReporter;
 import com.android.server.backup.keyvalue.KeyValueBackupTask;
 import com.android.server.backup.transport.TransportClient;
+import com.android.server.backup.utils.BackupEligibilityRules;
 
 import org.robolectric.annotation.Implementation;
 import org.robolectric.annotation.Implements;
@@ -63,7 +64,8 @@ public class ShadowKeyValueBackupTask {
             OnTaskFinishedListener listener,
             List<String> pendingFullBackups,
             boolean userInitiated,
-            boolean nonIncremental) {
+            boolean nonIncremental,
+            BackupEligibilityRules backupEligibilityRules) {
         mListener = listener;
         mQueue = queue;
         mPendingFullBackups = pendingFullBackups;
diff --git a/services/robotests/src/com/android/server/testing/shadows/ShadowPerformUnifiedRestoreTask.java b/services/robotests/src/com/android/server/testing/shadows/ShadowPerformUnifiedRestoreTask.java
index 8daef5fad032..5161070398d7 100644
--- a/services/robotests/src/com/android/server/testing/shadows/ShadowPerformUnifiedRestoreTask.java
+++ b/services/robotests/src/com/android/server/testing/shadows/ShadowPerformUnifiedRestoreTask.java
@@ -25,6 +25,7 @@ import com.android.server.backup.UserBackupManagerService;
 import com.android.server.backup.internal.OnTaskFinishedListener;
 import com.android.server.backup.restore.PerformUnifiedRestoreTask;
 import com.android.server.backup.transport.TransportClient;
+import com.android.server.backup.utils.BackupEligibilityRules;
 
 import org.robolectric.annotation.Implementation;
 import org.robolectric.annotation.Implements;
@@ -67,7 +68,8 @@ public class ShadowPerformUnifiedRestoreTask {
             int pmToken,
             boolean isFullSystemRestore,
             @Nullable String[] filterSet,
-            OnTaskFinishedListener listener) {
+            OnTaskFinishedListener listener,
+            BackupEligibilityRules backupEligibilityRules) {
         mBackupManagerService = backupManagerService;
         mPackage = targetPackage;
         mIsFullSystemRestore = isFullSystemRestore;