Files
Lawnchair/tests/src/com/android/launcher3/deviceemulator/DisplayEmulator.java
T
Sebastian Franco 00aff95ac0 Give the tests the ability to emulate other devices screens
This code contains utility clases that can change the display
of a device and make it look like another device.
The function DisplayEmulator#emulate receives a DeviceEmulationData
a certain grid to emulate and a callback, everyting that happens
inside of the callback will happen when the device is being emulated
and can be used by other tests.

Example test:
package com.android.launcher3.deviceemulator;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.filters.MediumTest;
import com.android.launcher3.deviceemulator.models.DeviceEmulationData;
import com.android.launcher3.ui.AbstractLauncherUiTest;
import org.junit.Test;
import org.junit.runner.RunWith;
import java.util.concurrent.TimeUnit;

@MediumTest
@RunWith(AndroidJUnit4.class)
public class TestTest extends AbstractLauncherUiTest {
    @Test
    public void testEmulation() throws Exception {
        String deviceCode = "pixel6pro";
        DeviceEmulationData deviceData = DeviceEmulationData.getDevice(deviceCode);
        String grid = "normal";
        DisplayEmulator displayEmulator = new DisplayEmulator(mTargetContext);

        displayEmulator.emulate(deviceData, grid, () ->{
            TimeUnit.SECONDS.sleep(10);
            return true;
        });
    }
}

Test: You could use the test above to make your device look like a
Pixel6 pro for 10 secons.
Fix: 229028257

Change-Id: Icd79be405a2e14dda0bc5f555b0e46149e16f912
2022-04-25 11:57:58 -05:00

96 lines
3.9 KiB
Java

/*
* Copyright (C) 2022 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.launcher3.deviceemulator;
import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;
import android.content.ContentResolver;
import android.content.ContentValues;
import android.content.Context;
import android.net.Uri;
import android.os.UserHandle;
import android.view.Display;
import android.view.IWindowManager;
import android.view.WindowManagerGlobal;
import androidx.test.uiautomator.UiDevice;
import com.android.launcher3.deviceemulator.models.DeviceEmulationData;
import com.android.launcher3.tapl.LauncherInstrumentation;
import com.android.launcher3.util.window.WindowManagerProxy;
import java.util.concurrent.Callable;
public class DisplayEmulator {
Context mContext;
LauncherInstrumentation mLauncher;
DisplayEmulator(Context context, LauncherInstrumentation launcher) {
mContext = context;
mLauncher = launcher;
}
/**
* By changing the WindowManagerProxy we can override the window insets information
**/
private IWindowManager changeWindowManagerInstance(DeviceEmulationData deviceData) {
WindowManagerProxy.INSTANCE.initializeForTesting(
new TestWindowManagerProxy(mContext, deviceData));
return WindowManagerGlobal.getWindowManagerService();
}
public <T> T emulate(DeviceEmulationData device, String grid, Callable<T> runInEmulation)
throws Exception {
WindowManagerProxy original = WindowManagerProxy.INSTANCE.get(mContext);
// Set up emulation
final int userId = UserHandle.myUserId();
WindowManagerProxy.INSTANCE.initializeForTesting(
new TestWindowManagerProxy(mContext, device));
IWindowManager wm = changeWindowManagerInstance(device);
// Change density twice to force display controller to reset its state
wm.setForcedDisplayDensityForUser(Display.DEFAULT_DISPLAY, device.density / 2, userId);
wm.setForcedDisplayDensityForUser(Display.DEFAULT_DISPLAY, device.density, userId);
wm.setForcedDisplaySize(Display.DEFAULT_DISPLAY, device.width, device.height);
wm.setForcedDisplayScalingMode(Display.DEFAULT_DISPLAY, 1);
// Set up grid
setGrid(grid);
try {
return runInEmulation.call();
} finally {
// Clear emulation
WindowManagerProxy.INSTANCE.initializeForTesting(original);
UiDevice.getInstance(getInstrumentation()).executeShellCommand("cmd window reset");
}
}
private void setGrid(String gridType) {
// When the grid changes, the desktop arrangement get stored in SQL and we need to wait to
// make sure there is no SQL operations running and get SQL_BUSY error, that's why we need
// to call mLauncher.waitForLauncherInitialized();
mLauncher.waitForLauncherInitialized();
String testProviderAuthority = mContext.getPackageName() + ".grid_control";
Uri gridUri = new Uri.Builder()
.scheme(ContentResolver.SCHEME_CONTENT)
.authority(testProviderAuthority)
.appendPath("default_grid")
.build();
ContentValues values = new ContentValues();
values.put("name", gridType);
mContext.getContentResolver().update(gridUri, values, null, null);
}
}