tests/app/src/android/app/cts/DisplayTest.java

/*
 * Copyright (C) 2017 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 android.app.cts;

import static com.android.compatibility.common.util.PackageUtil.supportsRotation;

import static com.google.common.truth.Truth.assertWithMessage;

import android.app.stubs.DisplayTestActivity;
import android.app.stubs.OrientationTestUtils;
import android.graphics.Point;
import android.server.wm.IgnoreOrientationRequestSession;
import android.test.ActivityInstrumentationTestCase2;
import android.view.Display;

/**
 * Tests to verify functionality of {@link Display}.
 */
public class DisplayTest extends ActivityInstrumentationTestCase2<DisplayTestActivity> {
    private DisplayTestActivity mActivity;

    public DisplayTest() {
        super("android.app.stubs", DisplayTestActivity.class);
    }

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        mActivity = getActivity();
    }

    /**
     * Tests that the underlying {@link android.view.DisplayAdjustments} in {@link Display} updates.
     * The method {@link DisplayTestActivity#getDisplay()} fetches the Display directly from the
     * {@link android.view.WindowManager}. A display fetched before the rotation should have the
     * updated adjustments after a rotation.
     */
    public void testRotation() throws Throwable {
        if (!supportsRotation()) {
            // Skip rotation test if device doesn't support it.
            return;
        }

        try (IgnoreOrientationRequestSession session =
                new IgnoreOrientationRequestSession(false /* enable */)) {

            // Get a {@link Display} instance before rotation.
            final Display origDisplay = mActivity.getDisplay();

            // Capture the originally reported width and heights
            final Point origSize = new Point();
            origDisplay.getRealSize(origSize);

                         // Change orientation
            mActivity.configurationChangeObserver.startObserving();
            OrientationTestUtils.switchOrientation(mActivity);

            final boolean closeToSquareDisplay =
                    OrientationTestUtils.isCloseToSquareDisplay(mActivity);

             // Don't wait for the configuration to change if the
             // the display is square. In many cases it won't.
            if (!closeToSquareDisplay) {
                mActivity.configurationChangeObserver.await();
            }

            final Point newOrigSize = new Point();
            origDisplay.getRealSize(newOrigSize);

            // Get a {@link Display} instance after rotation.
            final Display updatedDisplay = mActivity.getDisplay();
            final Point updatedSize = new Point();
            updatedDisplay.getRealSize(updatedSize);

             // For square screens the following assertions do not make sense and will always fail.
            if (!closeToSquareDisplay) {
                // Ensure that the width and height of the original instance no longer are the same.
                // Note that this will be false if the device width and height are identical.
                // Note there are cases where width and height may not all be updated, such as on
                // docked devices where the app is letterboxed. However, at least one dimension
                // needs to be updated.
                assertWithMessage("size from original display instance should have changed")
                        .that(origSize).isNotEqualTo(newOrigSize);
            }

            // Ensure that the width and height of the original instance have been updated to match
            // the values that would be found in a new instance.
            assertWithMessage("size from original display instance should match current")
                    .that(newOrigSize).isEqualTo(updatedSize);
        }
    }
}