Moving some tests off TAPL
Bug: 377772352 Test: Presubmit Flag: EXEMPT test refactor Change-Id: I390fcfd1ad698ba8858891b72e61e3b6792cf718
This commit is contained in:
Sunny Goyal
@@ -0,0 +1,203 @@
|
||||
/*
|
||||
* 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.ui.workspace;
|
||||
|
||||
import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;
|
||||
|
||||
import static com.android.launcher3.AbstractFloatingView.TYPE_ACTION_POPUP;
|
||||
import static com.android.launcher3.util.Executors.MAIN_EXECUTOR;
|
||||
import static com.android.launcher3.util.TestConstants.AppNames.TEST_APP_NAME;
|
||||
|
||||
import static org.junit.Assert.assertEquals;
|
||||
import static org.junit.Assert.assertNotNull;
|
||||
import static org.junit.Assert.assertTrue;
|
||||
|
||||
import android.content.ContentProviderClient;
|
||||
import android.content.ContentResolver;
|
||||
import android.content.ContentValues;
|
||||
import android.net.Uri;
|
||||
import android.view.View;
|
||||
import android.view.ViewGroup;
|
||||
|
||||
import androidx.test.filters.LargeTest;
|
||||
import androidx.test.uiautomator.UiDevice;
|
||||
|
||||
import com.android.launcher3.AbstractFloatingView;
|
||||
import com.android.launcher3.BubbleTextView;
|
||||
import com.android.launcher3.Launcher;
|
||||
import com.android.launcher3.LauncherState;
|
||||
import com.android.launcher3.allapps.AllAppsRecyclerView;
|
||||
import com.android.launcher3.celllayout.FavoriteItemsTransaction;
|
||||
import com.android.launcher3.icons.ThemedIconDrawable;
|
||||
import com.android.launcher3.popup.ArrowPopup;
|
||||
import com.android.launcher3.util.BaseLauncherActivityTest;
|
||||
import com.android.launcher3.util.Executors;
|
||||
import com.android.launcher3.util.TestUtil;
|
||||
|
||||
import org.junit.Test;
|
||||
|
||||
import java.util.ArrayDeque;
|
||||
import java.util.Queue;
|
||||
|
||||
/**
|
||||
* Tests for theme icon support in Launcher
|
||||
*
|
||||
* Note running these tests will clear the workspace on the device.
|
||||
*/
|
||||
@LargeTest
|
||||
public class ThemeIconsTest extends BaseLauncherActivityTest<Launcher> {
|
||||
|
||||
private static final String APP_NAME = "IconThemedActivity";
|
||||
private static final String SHORTCUT_NAME = "Shortcut 1";
|
||||
|
||||
@Test
|
||||
public void testIconWithoutTheme() throws Exception {
|
||||
setThemeEnabled(false);
|
||||
new FavoriteItemsTransaction(targetContext()).commit();
|
||||
loadLauncherSync();
|
||||
goToState(LauncherState.ALL_APPS);
|
||||
freezeAllApps();
|
||||
|
||||
scrollToAppIcon(APP_NAME);
|
||||
BubbleTextView btv = getFromLauncher(
|
||||
l -> verifyIconTheme(APP_NAME, l.getAppsView(), false));
|
||||
addToWorkspace(btv);
|
||||
executeOnLauncher(l -> verifyIconTheme(APP_NAME, l.getWorkspace(), false));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShortcutIconWithoutTheme() throws Exception {
|
||||
setThemeEnabled(false);
|
||||
new FavoriteItemsTransaction(targetContext()).commit();
|
||||
loadLauncherSync();
|
||||
goToState(LauncherState.ALL_APPS);
|
||||
freezeAllApps();
|
||||
|
||||
scrollToAppIcon(TEST_APP_NAME);
|
||||
BubbleTextView btv = getFromLauncher(l -> findBtv(TEST_APP_NAME, l.getAppsView()));
|
||||
TestUtil.runOnExecutorSync(MAIN_EXECUTOR, btv::performLongClick);
|
||||
|
||||
BubbleTextView menuItem = getOnceNotNull("Popup menu not open", l ->
|
||||
(AbstractFloatingView.getOpenView(l, TYPE_ACTION_POPUP) instanceof ArrowPopup ap)
|
||||
? findBtv(SHORTCUT_NAME, ap) : null);
|
||||
addToWorkspace(menuItem);
|
||||
executeOnLauncher(l -> verifyIconTheme(SHORTCUT_NAME, l.getWorkspace(), false));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIconWithTheme() throws Exception {
|
||||
setThemeEnabled(true);
|
||||
new FavoriteItemsTransaction(targetContext()).commit();
|
||||
loadLauncherSync();
|
||||
goToState(LauncherState.ALL_APPS);
|
||||
freezeAllApps();
|
||||
|
||||
scrollToAppIcon(APP_NAME);
|
||||
BubbleTextView btv = getFromLauncher(l ->
|
||||
verifyIconTheme(APP_NAME, l.getAppsView(), false));
|
||||
addToWorkspace(btv);
|
||||
executeOnLauncher(l -> verifyIconTheme(APP_NAME, l.getWorkspace(), true));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testShortcutIconWithTheme() throws Exception {
|
||||
setThemeEnabled(true);
|
||||
loadLauncherSync();
|
||||
goToState(LauncherState.ALL_APPS);
|
||||
freezeAllApps();
|
||||
|
||||
scrollToAppIcon(TEST_APP_NAME);
|
||||
BubbleTextView btv = getFromLauncher(l -> findBtv(TEST_APP_NAME, l.getAppsView()));
|
||||
TestUtil.runOnExecutorSync(MAIN_EXECUTOR, btv::performLongClick);
|
||||
|
||||
BubbleTextView menuItem = getOnceNotNull("Popup menu not open", l ->
|
||||
(AbstractFloatingView.getOpenView(l, TYPE_ACTION_POPUP) instanceof ArrowPopup ap)
|
||||
? findBtv(SHORTCUT_NAME, ap) : null);
|
||||
addToWorkspace(menuItem);
|
||||
executeOnLauncher(l -> verifyIconTheme(SHORTCUT_NAME, l.getWorkspace(), true));
|
||||
}
|
||||
|
||||
private BubbleTextView findBtv(String title, ViewGroup parent) {
|
||||
// Wait for Launcher model to be completed
|
||||
try {
|
||||
Executors.MODEL_EXECUTOR.submit(() -> { }).get();
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
|
||||
// Find the app icon
|
||||
Queue<View> viewQueue = new ArrayDeque<>();
|
||||
viewQueue.add(parent);
|
||||
BubbleTextView icon = null;
|
||||
while (!viewQueue.isEmpty()) {
|
||||
View view = viewQueue.poll();
|
||||
if (view instanceof ViewGroup) {
|
||||
parent = (ViewGroup) view;
|
||||
for (int i = parent.getChildCount() - 1; i >= 0; i--) {
|
||||
viewQueue.add(parent.getChildAt(i));
|
||||
}
|
||||
} else if (view instanceof BubbleTextView btv) {
|
||||
if (btv.getContentDescription() != null
|
||||
&& title.equals(btv.getContentDescription().toString())) {
|
||||
icon = btv;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
return icon;
|
||||
}
|
||||
|
||||
private BubbleTextView verifyIconTheme(String title, ViewGroup parent, boolean isThemed) {
|
||||
BubbleTextView icon = findBtv(title, parent);
|
||||
assertNotNull(icon.getIcon());
|
||||
assertEquals(isThemed, icon.getIcon() instanceof ThemedIconDrawable);
|
||||
return icon;
|
||||
}
|
||||
|
||||
private void setThemeEnabled(boolean isEnabled) throws Exception {
|
||||
Uri uri = new Uri.Builder()
|
||||
.scheme(ContentResolver.SCHEME_CONTENT)
|
||||
.authority(targetContext().getPackageName() + ".grid_control")
|
||||
.appendPath("set_icon_themed")
|
||||
.build();
|
||||
ContentValues values = new ContentValues();
|
||||
values.put("boolean_value", isEnabled);
|
||||
try (ContentProviderClient client = targetContext().getContentResolver()
|
||||
.acquireContentProviderClient(uri)) {
|
||||
int result = client.update(uri, values, null);
|
||||
assertTrue(result > 0);
|
||||
}
|
||||
}
|
||||
|
||||
private void scrollToAppIcon(String appName) {
|
||||
executeOnLauncher(l -> {
|
||||
l.hideKeyboard();
|
||||
AllAppsRecyclerView rv = l.getAppsView().getActiveRecyclerView();
|
||||
int pos = rv.getApps().getAdapterItems().indexOf(rv.getApps().getAdapterItems().stream()
|
||||
.filter(i -> i.itemInfo != null && appName.equals(i.itemInfo.title.toString()))
|
||||
.findFirst()
|
||||
.get());
|
||||
rv.getLayoutManager().scrollToPosition(pos);
|
||||
});
|
||||
}
|
||||
|
||||
private void addToWorkspace(View btv) {
|
||||
TestUtil.runOnExecutorSync(MAIN_EXECUTOR, () ->
|
||||
btv.getAccessibilityDelegate().performAccessibilityAction(
|
||||
btv, com.android.launcher3.R.id.action_add_to_workspace, null));
|
||||
UiDevice.getInstance(getInstrumentation()).waitForIdle();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user