services/robotests/src/com/android/server/testing/shadows/ShadowPerformBackupTask.java

/*
 * 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.app.backup.IBackupManagerMonitor;
import android.app.backup.IBackupObserver;

import com.android.server.backup.BackupManagerService;
import com.android.server.backup.DataChangedJournal;
import com.android.server.backup.internal.BackupRequest;
import com.android.server.backup.internal.OnTaskFinishedListener;
import com.android.server.backup.internal.PerformBackupTask;
import com.android.server.backup.transport.TransportClient;

import org.robolectric.annotation.Implementation;
import org.robolectric.annotation.Implements;

import java.util.ArrayList;
import java.util.List;

@Implements(PerformBackupTask.class)
public class ShadowPerformBackupTask {
    @Nullable private static ShadowPerformBackupTask sLastShadow;

    /**
     * Retrieves the shadow for the last {@link PerformBackupTask} object created.
     *
     * @return The shadow or {@code null} if no object created since last {@link #reset()}.
     */
    @Nullable
    public static ShadowPerformBackupTask getLastCreated() {
        return sLastShadow;
    }

    public static void reset() {
        sLastShadow = null;
    }

    private OnTaskFinishedListener mListener;
    private ArrayList<BackupRequest> mQueue;
    private List<String> mPendingFullBackups;

    @Implementation
    public void __constructor__(
            BackupManagerService backupManagerService,
            TransportClient transportClient,
            String dirName,
            ArrayList<BackupRequest> queue,
            @Nullable DataChangedJournal journal,
            IBackupObserver observer,
            IBackupManagerMonitor monitor,
            @Nullable OnTaskFinishedListener listener,
            List<String> pendingFullBackups,
            boolean userInitiated,
            boolean nonIncremental) {
        mListener = listener;
        mQueue = queue;
        mPendingFullBackups = pendingFullBackups;
        sLastShadow = this;
    }

    @Implementation
    public void execute() {
        mListener.onFinished("ShadowPerformBackupTask.execute()");
    }

    public List<BackupRequest> getQueue() {
        return mQueue;
    }

    public List<String> getPendingFullBackups() {
        return mPendingFullBackups;
    }
}