1 files changed, 133 insertions, 0 deletions
diff --git a/tests/src/com/android/wallpaper/testing/TestWallpaperRotationInitializer.java b/tests/src/com/android/wallpaper/testing/TestWallpaperRotationInitializer.java
new file mode 100644
index 0000000..9494d19
--- /dev/null
+++ b/tests/src/com/android/wallpaper/testing/TestWallpaperRotationInitializer.java
@@ -0,0 +1,133 @@
+/*
+ * Copyright (C) 2019 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.wallpaper.testing;
+
+import android.content.Context;
+import android.os.Parcel;
+import android.os.Parcelable;
+
+import com.android.wallpaper.model.WallpaperRotationInitializer;
+import com.android.wallpaper.module.InjectorProvider;
+import com.android.wallpaper.module.WallpaperPreferences;
+
+/**
+ * Test implementation of {@link WallpaperRotationInitializer}.
+ */
+public class TestWallpaperRotationInitializer implements WallpaperRotationInitializer {
+
+    private boolean mIsRotationInitialized;
+    @NetworkPreference
+    private int mNetworkPreference;
+    private Listener mListener;
+    @RotationInitializationState
+    private int mRotationInitializationState;
+
+    TestWallpaperRotationInitializer() {
+        mIsRotationInitialized = false;
+        mNetworkPreference = NETWORK_PREFERENCE_WIFI_ONLY;
+        mRotationInitializationState = WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED;
+    }
+
+    TestWallpaperRotationInitializer(@RotationInitializationState int rotationState) {
+        mIsRotationInitialized = false;
+        mNetworkPreference = NETWORK_PREFERENCE_WIFI_ONLY;
+        mRotationInitializationState = rotationState;
+    }
+
+    private TestWallpaperRotationInitializer(Parcel unused) {
+        mIsRotationInitialized = false;
+        mNetworkPreference = NETWORK_PREFERENCE_WIFI_ONLY;
+    }
+
+    @Override
+    public void setFirstWallpaperInRotation(Context context,
+            @NetworkPreference int networkPreference,
+            Listener listener) {
+        mListener = listener;
+        mNetworkPreference = networkPreference;
+    }
+
+    @Override
+    public boolean startRotation(Context appContext) {
+        WallpaperPreferences wallpaperPreferences =
+                InjectorProvider.getInjector().getPreferences(appContext);
+        wallpaperPreferences.setWallpaperPresentationMode(
+                WallpaperPreferences.PRESENTATION_MODE_ROTATING);
+        return true;
+    }
+
+    @Override
+    public void fetchRotationInitializationState(Context context, RotationStateListener listener) {
+        listener.onRotationStateReceived(mRotationInitializationState);
+    }
+
+    @Override
+    public int getRotationInitializationStateDirty(Context context) {
+        return mRotationInitializationState;
+    }
+
+    /** Sets the mocked rotation initialization state. */
+    public void setRotationInitializationState(@RotationInitializationState int rotationState) {
+        mRotationInitializationState = rotationState;
+    }
+
+    /**
+     * Simulates completion of the asynchronous task to download the first wallpaper in a rotation..
+     *
+     * @param isSuccessful Whether the first wallpaper downloaded successfully.
+     */
+    public void finishDownloadingFirstWallpaper(boolean isSuccessful) {
+        if (isSuccessful) {
+            mIsRotationInitialized = true;
+            mListener.onFirstWallpaperInRotationSet();
+        } else {
+            mIsRotationInitialized = false;
+            mListener.onError();
+        }
+    }
+
+    /** Returns whether a wallpaper rotation is initialized. */
+    public boolean isRotationInitialized() {
+        return mIsRotationInitialized;
+    }
+
+    /** Returns whether a wallpaper rotation is enabled for WiFi-only. */
+    public boolean isWifiOnly() {
+        return mNetworkPreference == NETWORK_PREFERENCE_WIFI_ONLY;
+    }
+
+    @Override
+    public void writeToParcel(Parcel dest, int flags) {
+    }
+
+    @Override
+    public int describeContents() {
+        return 0;
+    }
+
+    public static final Parcelable.Creator<TestWallpaperRotationInitializer> CREATOR =
+            new Parcelable.Creator<TestWallpaperRotationInitializer>() {
+                @Override
+                public TestWallpaperRotationInitializer createFromParcel(Parcel in) {
+                    return new TestWallpaperRotationInitializer(in);
+                }
+
+                @Override
+                public TestWallpaperRotationInitializer[] newArray(int size) {
+                    return new TestWallpaperRotationInitializer[size];
+                }
+            };
+}