path: root/tests/src/com/android/wallpaper/picker/individual/IndividualPickerActivityTest.java

/*
 * 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.picker.individual;

import static androidx.test.espresso.Espresso.onView;
import static androidx.test.espresso.action.ViewActions.click;
import static androidx.test.espresso.assertion.ViewAssertions.doesNotExist;
import static androidx.test.espresso.assertion.ViewAssertions.matches;
import static androidx.test.espresso.intent.Intents.intended;
import static androidx.test.espresso.intent.Intents.intending;
import static androidx.test.espresso.intent.matcher.ComponentNameMatchers.hasClassName;
import static androidx.test.espresso.intent.matcher.IntentMatchers.hasAction;
import static androidx.test.espresso.intent.matcher.IntentMatchers.hasComponent;
import static androidx.test.espresso.intent.matcher.IntentMatchers.hasExtra;
import static androidx.test.espresso.matcher.ViewMatchers.isChecked;
import static androidx.test.espresso.matcher.ViewMatchers.isDisplayed;
import static androidx.test.espresso.matcher.ViewMatchers.withId;
import static androidx.test.espresso.matcher.ViewMatchers.withText;

import static junit.framework.TestCase.assertFalse;

import static org.hamcrest.Matchers.allOf;
import static org.hamcrest.Matchers.equalTo;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import android.app.Activity;
import android.app.Instrumentation.ActivityResult;
import android.app.WallpaperManager;
import android.content.Context;
import android.content.Intent;
import android.graphics.drawable.ColorDrawable;
import android.widget.FrameLayout;

import androidx.recyclerview.widget.RecyclerView;
import androidx.test.espresso.contrib.RecyclerViewActions;
import androidx.test.espresso.intent.Intents;
import androidx.test.filters.MediumTest;
import androidx.test.internal.runner.junit4.AndroidJUnit4ClassRunner;
import androidx.test.platform.app.InstrumentationRegistry;
import androidx.test.rule.ActivityTestRule;

import com.android.wallpaper.R;
import com.android.wallpaper.config.Flags;
import com.android.wallpaper.model.Category;
import com.android.wallpaper.model.PickerIntentFactory;
import com.android.wallpaper.model.WallpaperInfo;
import com.android.wallpaper.model.WallpaperRotationInitializer;
import com.android.wallpaper.model.WallpaperRotationInitializer.RotationInitializationState;
import com.android.wallpaper.module.FormFactorChecker;
import com.android.wallpaper.module.Injector;
import com.android.wallpaper.module.InjectorProvider;
import com.android.wallpaper.testing.TestCategoryProvider;
import com.android.wallpaper.testing.TestFormFactorChecker;
import com.android.wallpaper.testing.TestInjector;
import com.android.wallpaper.testing.TestUserEventLogger;
import com.android.wallpaper.testing.TestWallpaperCategory;
import com.android.wallpaper.testing.TestWallpaperInfo;
import com.android.wallpaper.testing.TestWallpaperRotationInitializer;

import org.hamcrest.Matcher;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;

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

/**
 * Tests for {@link IndividualPickerActivity}.
 */
@RunWith(AndroidJUnit4ClassRunner.class)
@MediumTest
public class IndividualPickerActivityTest {

    private static final String EXTRA_WALLPAPER_INFO =
            "com.android.wallpaper.picker.wallpaper_info";
    private static final TestWallpaperInfo sWallpaperInfo1 = new TestWallpaperInfo(
            TestWallpaperInfo.COLOR_BLACK, "test-wallpaper-1");
    private static final TestWallpaperInfo sWallpaperInfo2 = new TestWallpaperInfo(
            TestWallpaperInfo.COLOR_BLACK, "test-wallpaper-2");
    private static final TestWallpaperInfo sWallpaperInfo3 = new TestWallpaperInfo(
            TestWallpaperInfo.COLOR_BLACK, "test-wallpaper-3");

    private TestCategoryProvider mTestCategoryProvider;

    private TestFormFactorChecker mTestFormFactorChecker;
    private Injector mInjector;

    private TestWallpaperCategory mTestCategory;

    @Rule
    public ActivityTestRule<IndividualPickerActivity> mActivityRule =
            new ActivityTestRule<>(IndividualPickerActivity.class, false, false);

    @Before
    public void setUp() {
        Intents.init();

        mInjector = new TestInjector();
        InjectorProvider.setInjector(mInjector);

        Context context = InstrumentationRegistry.getInstrumentation().getTargetContext();
        mTestFormFactorChecker = (TestFormFactorChecker) mInjector.getFormFactorChecker(context);
        mTestCategoryProvider = (TestCategoryProvider) mInjector.getCategoryProvider(context);

        sWallpaperInfo1.setAttributions(Arrays.asList(
                "Attribution 0", "Attribution 1", "Attribution 2"));
    }

    @After
    public void tearDown() {
        Intents.release();
        mActivityRule.finishActivity();
    }

    private IndividualPickerActivity getActivity() {
        return mActivityRule.getActivity();
    }

    private void setUpFragmentForTesting() {
        IndividualPickerFragment fragment = (IndividualPickerFragment)
                getActivity().getSupportFragmentManager().findFragmentById(R.id.fragment_container);
        fragment.setTestingMode(true);
    }

    private void setActivityWithMockWallpapers(boolean isRotationEnabled,
            @RotationInitializationState int rotationState) {
        sWallpaperInfo1.setCollectionId("collection");

        ArrayList<WallpaperInfo> wallpapers = new ArrayList<>();
        wallpapers.add(sWallpaperInfo1);
        wallpapers.add(sWallpaperInfo2);
        wallpapers.add(sWallpaperInfo3);

        mTestCategory = new TestWallpaperCategory(
                "Test category", "collection", wallpapers, 0 /* priority */);
        mTestCategory.setIsRotationEnabled(isRotationEnabled);
        mTestCategory.setRotationInitializationState(rotationState);

        List<Category> testCategories = mTestCategoryProvider.getTestCategories();
        testCategories.set(0, mTestCategory);

        PickerIntentFactory intentFactory =
                new IndividualPickerActivity.IndividualPickerActivityIntentFactory();
        Intent intent = intentFactory.newIntent(
                InstrumentationRegistry.getInstrumentation().getTargetContext(),
                mTestCategory.getCollectionId());
        mActivityRule.launchActivity(intent);
    }

    @Test
    public void testDrawsTilesForProvidedWallpapers() {
        setActivityWithMockWallpapers(false /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        IndividualPickerActivity activity = getActivity();

        RecyclerView recyclerView = activity.findViewById(R.id.wallpaper_grid);

        // There are only three wallpapers in the category, so the grid should only have three
        // items.
        assertNotNull(recyclerView.findViewHolderForAdapterPosition(0));
        assertNotNull(recyclerView.findViewHolderForAdapterPosition(1));
        assertNotNull(recyclerView.findViewHolderForAdapterPosition(2));
        assertNull(recyclerView.findViewHolderForAdapterPosition(3));
    }

    @Test
    public void testClickTile_Mobile_LaunchesPreviewActivityWithWallpaper() {
        mTestFormFactorChecker.setFormFactor(FormFactorChecker.FORM_FACTOR_MOBILE);

        setActivityWithMockWallpapers(false /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        onView(withId(R.id.wallpaper_grid)).perform(
                RecyclerViewActions.actionOnItemAtPosition(0, click()));
        intended(allOf(
                hasComponent(hasClassName("com.android.wallpaper.picker.PreviewActivity")),
                hasExtra(equalTo(EXTRA_WALLPAPER_INFO), equalTo(sWallpaperInfo1))));

        Context context = InstrumentationRegistry.getInstrumentation().getTargetContext();
        TestUserEventLogger eventLogger = (TestUserEventLogger) mInjector.getUserEventLogger(
                context);
        assertEquals(1, eventLogger.getNumIndividualWallpaperSelectedEvents());
        assertEquals(sWallpaperInfo1.getCollectionId(context), eventLogger.getLastCollectionId());
    }

    /**
     * Tests that the static daily rotation tile (with flag dynamicStartRotationTileEnabled=false)
     * has a background of the light blue accent color and says "Tap to turn on".
     */
    @Test
    public void testRotationEnabled_StaticDailyRotationTile() {
        Flags.dynamicStartRotationTileEnabled = false;
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        onView(withText(R.string.daily_refresh_tile_title)).check(matches(isDisplayed()));
        onView(withText(R.string.daily_refresh_tile_subtitle)).check(matches(isDisplayed()));

        // Check that the background color of the "daily refresh" tile is the blue accent color.
        FrameLayout rotationTile = getActivity().findViewById(R.id.daily_refresh);
        int backgroundColor = ((ColorDrawable) rotationTile.getBackground()).getColor();
        assertEquals(getActivity().getResources().getColor(R.color.accent_color),
                backgroundColor);
    }

    /**
     * Tests that when rotation is enabled and the rotation for this category is already in effect
     * on both home & lock screens, then the "start rotation" tile reads "Home & Lock".
     */
    @Test
    public void testRotationEnabled_RotationInitializedHomeAndLock() {
        Flags.dynamicStartRotationTileEnabled = true;

        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_HOME_AND_LOCK);
        getActivity();

        onView(withText(R.string.daily_refresh_tile_title)).check(matches(isDisplayed()));
        onView(withText(R.string.home_and_lock_short_label)).check(matches(isDisplayed()));
    }

    /**
     * Tests that when rotation is enabled and the rotation is aleady for this category is already
     * in
     * effect on the home-screen only, then the "start rotation" tile reads "Home screen".
     */
    @Test
    public void testRotationEnabled_RotationInitializedHomeScreenOnly() {
        Flags.dynamicStartRotationTileEnabled = true;

        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_HOME_ONLY);
        getActivity();

        onView(withText(R.string.daily_refresh_tile_title)).check(matches(isDisplayed()));
        onView(withText(R.string.home_screen_message)).check(matches(isDisplayed()));
    }

    /**
     * Tests that after the IndividualPickerActivity loads, if the state of the current category's
     * rotation changes while the activity is restarted, then the UI for the "start rotation" tile
     * changes to reflect that new state.
     */
    @Test
    public void testActivityRestarted_RotationStateChanges_StartRotationTileUpdates()
            throws Throwable {
        Flags.dynamicStartRotationTileEnabled = true;

        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_HOME_ONLY);

        onView(withText(R.string.daily_refresh_tile_title)).check(matches(isDisplayed()));
        onView(withText(R.string.home_screen_message)).check(matches(isDisplayed()));

        // Now change the rotation initialization state such that the tile should say
        // "Tap to turn on" after the activity resumes.
        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        testWPRotationInitializer.setRotationInitializationState(
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);

        // Restart the activity.
        IndividualPickerActivity activity = getActivity();
        mActivityRule.runOnUiThread(activity::recreate);

        onView(withText(R.string.daily_refresh_tile_title)).check(matches(isDisplayed()));
        onView(withText(R.string.daily_refresh_tile_subtitle)).check(matches(isDisplayed()));
    }

    @Test
    public void testRotationDisabled_DoesNotRenderDailyRefreshTile() {
        setActivityWithMockWallpapers(false /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        onView(withText(R.string.daily_refresh_tile_title)).check(doesNotExist());
        onView(withText(R.string.daily_refresh_tile_subtitle)).check(doesNotExist());
    }

    @Test
    public void testClickDailyRefreshTile_ShowsStartRotationDialog() {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        onView(withId(R.id.daily_refresh)).perform(click());

        onView(withId(R.id.start_rotation_wifi_only_checkbox))
                .check(matches(isDisplayed()));
        // WiFi-only option should be checked by default.
        onView(withId(R.id.start_rotation_wifi_only_checkbox))
                .check(matches(isChecked()));
    }

    @Test
    public void testShowStartRotationDialog_WifiOnly_ClickOK_StartsRotation() throws Throwable {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        // Mock out the intent and response for the live wallpaper preview.
        Matcher<Intent> expectedIntent = hasAction(WallpaperManager.ACTION_CHANGE_LIVE_WALLPAPER);
        intending(expectedIntent).respondWith(new ActivityResult(Activity.RESULT_OK, null));

        onView(withId(R.id.daily_refresh)).perform(click());

        onView(withText(android.R.string.ok)).perform(click());
        mActivityRule.runOnUiThread(() -> {
            testWPRotationInitializer.finishDownloadingFirstWallpaper(true /* isSuccessful */);
            assertTrue(testWPRotationInitializer.isRotationInitialized());
            assertTrue(testWPRotationInitializer.isWifiOnly());

            // The activity should finish if starting a rotation was successful.
            assertTrue(getActivity().isFinishing());
        });
    }

    @Test
    public void testShowStartRotationDialog_WifiOnly_ClickOK_Fails_ShowsErrorDialog() {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        onView(withId(R.id.daily_refresh)).perform(click());
        onView(withText(android.R.string.ok)).perform(click());

        testWPRotationInitializer.finishDownloadingFirstWallpaper(false /* isSuccessful */);
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        // Error dialog should be shown with retry option.
        onView(withText(R.string.start_rotation_error_message)).check(matches(isDisplayed()));
        onView(withText(R.string.try_again)).check(matches(isDisplayed()));
    }

    @Test
    public void testStartRotation_WifiOnly_FailOnce_Retry() throws Throwable {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        onView(withId(R.id.daily_refresh)).perform(click());
        onView(withText(android.R.string.ok)).perform(click());

        testWPRotationInitializer.finishDownloadingFirstWallpaper(false /* isSuccessful */);
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        // Click try again to retry.
        onView(withText(R.string.try_again)).perform(click());

        mActivityRule.runOnUiThread(() -> {
            testWPRotationInitializer.finishDownloadingFirstWallpaper(true /* isSuccessful */);
            assertTrue(testWPRotationInitializer.isRotationInitialized());
            assertTrue(testWPRotationInitializer.isWifiOnly());
        });
    }

    @Test
    public void testShowStartRotationDialog_TurnOffWifiOnly_ClickOK_StartsRotation()
            throws Throwable {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        onView(withId(R.id.daily_refresh)).perform(click());
        // Click on WiFi-only option to toggle it off.
        onView(withId(R.id.start_rotation_wifi_only_checkbox)).perform(click());
        onView(withText(android.R.string.ok)).perform(click());

        mActivityRule.runOnUiThread(() -> {
            testWPRotationInitializer.finishDownloadingFirstWallpaper(true /* isSuccessful */);
            assertTrue(testWPRotationInitializer.isRotationInitialized());
            assertFalse(testWPRotationInitializer.isWifiOnly());
        });
    }

    @Test
    public void testStartRotation_WifiOnly_FailOnce_Retry_ShouldStillHaveWifiTurnedOff()
            throws Throwable {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        onView(withId(R.id.daily_refresh)).perform(click());
        // Click on WiFi-only option to toggle it off.
        onView(withId(R.id.start_rotation_wifi_only_checkbox)).perform(click());
        onView(withText(android.R.string.ok)).perform(click());

        testWPRotationInitializer.finishDownloadingFirstWallpaper(false /* isSuccessful */);
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        // Click try again to retry.
        onView(withText(R.string.try_again)).perform(click());

        mActivityRule.runOnUiThread(() -> {
            testWPRotationInitializer.finishDownloadingFirstWallpaper(true /* isSuccessful */);
            assertTrue(testWPRotationInitializer.isRotationInitialized());
            assertFalse(testWPRotationInitializer.isWifiOnly());
        });
    }

    @Test
    public void testShowStartRotationDialog_ClickCancel_DismissesDialog() {
        setActivityWithMockWallpapers(true /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        getActivity();

        setUpFragmentForTesting();
        TestWallpaperRotationInitializer
                testWPRotationInitializer = (TestWallpaperRotationInitializer)
                mTestCategory.getWallpaperRotationInitializer();
        assertFalse(testWPRotationInitializer.isRotationInitialized());

        onView(withId(R.id.daily_refresh)).perform(click());
        onView(withId(R.id.start_rotation_wifi_only_checkbox)).check(matches(isDisplayed()));
        // WiFi-only option should be checked by default.
        onView(withId(R.id.start_rotation_wifi_only_checkbox)).check(matches(isChecked()));

        // Click "Cancel" to dismiss the dialog.
        onView(withText(android.R.string.cancel)).perform(click());

        // Rotation was not initialized and dialog is no longer visible.
        assertFalse(testWPRotationInitializer.isRotationInitialized());
        onView(withId(R.id.start_rotation_wifi_only_checkbox)).check(doesNotExist());
    }

    /**
     * Tests that when no title is present, a wallpaper tile's content description attribute is
     * set to the first attribution.
     */
    @Test
    public void testWallpaperHasContentDescriptionFromAttribution() {
        setActivityWithMockWallpapers(false /* isRotationEnabled */,
                WallpaperRotationInitializer.ROTATION_NOT_INITIALIZED);
        IndividualPickerActivity activity = getActivity();

        RecyclerView recyclerView = activity.findViewById(R.id.wallpaper_grid);

        RecyclerView.ViewHolder holder = recyclerView.findViewHolderForAdapterPosition(0);
        assertEquals("Attribution 0", holder.itemView.findViewById(R.id.tile)
                .getContentDescription());
    }
}