tests/tests/view/src/android/view/cts/TextureViewTest.java

/*
 * Copyright (C) 2016 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.view.cts;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import android.graphics.Bitmap;
import android.graphics.Color;
import android.graphics.Point;
import android.support.test.filters.MediumTest;
import android.support.test.rule.ActivityTestRule;
import android.support.test.runner.AndroidJUnit4;
import android.view.PixelCopy;
import android.view.View;
import android.view.Window;

import com.android.compatibility.common.util.SynchronousPixelCopy;
import com.android.compatibility.common.util.WidgetTestUtils;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;

import java.util.concurrent.TimeoutException;

@MediumTest
@RunWith(AndroidJUnit4.class)
public class TextureViewTest {
    private TextureViewCtsActivity mActivity;
    private SynchronousPixelCopy mPixelCopy;
    private Window mWindow;

    @Rule
    public ActivityTestRule<TextureViewCtsActivity> mActivityRule =
            new ActivityTestRule<>(TextureViewCtsActivity.class);

    @Before
    public void setup() {
        mActivity = mActivityRule.getActivity();
        mPixelCopy = new SynchronousPixelCopy();
        assertNotNull(mActivity);
    }

    @Test
    public void testFirstFrames() throws Throwable {
        mActivity.waitForEnterAnimationComplete();

        final Point center = new Point();
        mActivityRule.runOnUiThread(() -> {
            View content = mActivity.findViewById(android.R.id.content);
            int[] outLocation = new int[2];
            content.getLocationOnScreen(outLocation);
            center.x = outLocation[0] + (content.getWidth() / 2);
            center.y = outLocation[1] + (content.getHeight() / 2);
            mWindow = mActivity.getWindow();
        });
        assertTrue(center.x > 0);
        assertTrue(center.y > 0);
        waitForColor(center, Color.WHITE);
        mActivity.waitForSurface();
        mActivity.initGl();
        int updatedCount;
        updatedCount = mActivity.waitForSurfaceUpdateCount(0);
        assertEquals(0, updatedCount);
        mActivity.drawColor(Color.GREEN);
        updatedCount = mActivity.waitForSurfaceUpdateCount(1);
        assertEquals(1, updatedCount);
        assertEquals(Color.WHITE, getPixel(center));
        WidgetTestUtils.runOnMainAndDrawSync(mActivityRule,
                mActivity.findViewById(android.R.id.content), () -> mActivity.removeCover());

        int color = waitForChange(center, Color.WHITE);
        assertEquals(Color.GREEN, color);
        mActivity.drawColor(Color.BLUE);
        updatedCount = mActivity.waitForSurfaceUpdateCount(2);
        assertEquals(2, updatedCount);
        color = waitForChange(center, color);
        assertEquals(Color.BLUE, color);
    }

    private int getPixel(Point point) {
        Bitmap screenshot = Bitmap.createBitmap(mWindow.getDecorView().getWidth(),
                mWindow.getDecorView().getHeight(), Bitmap.Config.ARGB_8888);
        int result = mPixelCopy.request(mWindow, screenshot);
        assertEquals("Copy request failed", PixelCopy.SUCCESS, result);
        int pixel = screenshot.getPixel(point.x, point.y);
        screenshot.recycle();
        return pixel;
    }

    private void waitForColor(Point point, int color)
            throws InterruptedException, TimeoutException {
        for (int i = 0; i < 20; i++) {
            int pixel = getPixel(point);
            if (pixel == color) {
                return;
            }
            Thread.sleep(16);
        }
        throw new TimeoutException();
    }

    private int waitForChange(Point point, int color)
            throws InterruptedException, TimeoutException {
        for (int i = 0; i < 20; i++) {
            int pixel = getPixel(point);
            if (pixel != color) {
                return pixel;
            }
            Thread.sleep(16);
        }
        throw new TimeoutException();
    }
}