weblayer/shell/android/javatests/src/org/chromium/weblayer/test/WebLayerShellActivityTestRule.java

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.weblayer.test;

import android.app.Activity;
import android.app.Instrumentation.ActivityMonitor;
import android.content.ComponentName;
import android.content.Intent;
import android.content.pm.ActivityInfo;
import android.content.res.Configuration;
import android.net.Uri;
import android.support.test.InstrumentationRegistry;
import android.support.test.rule.ActivityTestRule;

import org.chromium.content_public.browser.test.util.Criteria;
import org.chromium.content_public.browser.test.util.CriteriaHelper;
import org.chromium.content_public.browser.test.util.TestThreadUtils;
import org.chromium.weblayer.NavigationController;
import org.chromium.weblayer.shell.WebLayerShellActivity;

import java.lang.reflect.Field;

/**
 * ActivityTestRule for WebLayerShellActivity.
 *
 * Test can use this ActivityTestRule to launch or get WebLayerShellActivity.
 */
public class WebLayerShellActivityTestRule extends ActivityTestRule<WebLayerShellActivity> {
    private static final long WAIT_FOR_NAVIGATION_TIMEOUT = 10000L;

    public WebLayerShellActivityTestRule() {
        super(WebLayerShellActivity.class, false, false);
    }

    /**
     * Starts the WebLayer activity and loads the given URL.
     */
    public WebLayerShellActivity launchShellWithUrl(String url) {
        Intent intent = new Intent(Intent.ACTION_MAIN);
        intent.addCategory(Intent.CATEGORY_LAUNCHER);
        intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
        if (url != null) intent.setData(Uri.parse(url));
        intent.setComponent(
                new ComponentName(InstrumentationRegistry.getInstrumentation().getTargetContext(),
                        WebLayerShellActivity.class));
        return launchActivity(intent);
    }

    /**
     * Waits for the shell to navigate to the given URI.
     */
    public void waitForNavigation(String uri) {
        CriteriaHelper.pollUiThread(new Criteria() {
            @Override
            public boolean isSatisfied() {
                NavigationController navigationController =
                        getActivity().getBrowserController().getNavigationController();
                Uri currentUri = navigationController.getNavigationEntryDisplayUri(
                        navigationController.getNavigationListCurrentIndex());
                return currentUri.toString().equals(uri);
            }
        }, WAIT_FOR_NAVIGATION_TIMEOUT, CriteriaHelper.DEFAULT_POLLING_INTERVAL);
    }

    /**
     * Loads the given URL in the shell.
     */
    public void loadUrl(String url) {
        TestThreadUtils.runOnUiThreadBlocking(() -> { getActivity().loadUrl(url); });
    }

    /**
     * Rotates the Activity, blocking until the Activity is re-created.
     * After calling this, getActivity() returns the new Activity.
     */
    public void rotateActivity() {
        Activity activity = getActivity();

        ActivityMonitor monitor = new ActivityMonitor(WebLayerShellActivity.class.getName(),
                null, false);
        InstrumentationRegistry.getInstrumentation().addMonitor(monitor);

        int current = activity.getResources().getConfiguration().orientation;
        int requested = current == Configuration.ORIENTATION_LANDSCAPE ?
                ActivityInfo.SCREEN_ORIENTATION_PORTRAIT :
                ActivityInfo.SCREEN_ORIENTATION_LANDSCAPE;
        activity.setRequestedOrientation(requested);
        CriteriaHelper.pollUiThread(() ->
            monitor.getLastActivity() != null && monitor.getLastActivity() != activity
        );
        InstrumentationRegistry.getInstrumentation().removeMonitor(monitor);

        // There is no way to rotate the activity using ActivityTestRule or even notify it.
        // So we have to hack...
        try {
            Field field = ActivityTestRule.class.getDeclaredField("mActivity");
            field.setAccessible(true);
            field.set(this, monitor.getLastActivity());
        } catch (ReflectiveOperationException e) {
            throw new RuntimeException(e);
        }
    }
}