aixan/lawnchair
1
0
Fork 0
mirror of https://github.com/LawnchairLauncher/lawnchair.git synced 2026-02-20 03:08:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge "Verifying success of setting default Launcher in tests" into sc-dev

Browse Source
This commit is contained in:
Vadim Tryshev
2021-06-30 20:43:36 +00:00
committed by Android (Google) Code Review
parent e99ed7b9d1 a26516c575
commit ba302e8a6a
1 changed files with 48 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
tests/src/com/android/launcher3/util/rule/ShellCommandRule.java
View File

@@ -15,21 +15,27 @@
*/
package com.android.launcher3.util.rule;
import static com.android.launcher3.tapl.TestHelpers.getLauncherInMyProcess;
import static androidx.test.InstrumentationRegistry.getInstrumentation;
import static com.android.launcher3.tapl.TestHelpers.getLauncherInMyProcess;
import android.content.ComponentName;
import android.content.pm.ActivityInfo;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
import android.util.Log;
import androidx.annotation.Nullable;
import androidx.test.InstrumentationRegistry;
import androidx.test.uiautomator.UiDevice;
import com.android.systemui.shared.system.PackageManagerWrapper;
import org.junit.Assert;
import org.junit.rules.TestRule;
import org.junit.runner.Description;
import org.junit.runners.model.Statement;
import java.util.ArrayList;
/**
* Test rule which executes a shell command at the start of the test.
*/
@@ -37,10 +43,19 @@ public class ShellCommandRule implements TestRule {
private final String mCmd;
private final String mRevertCommand;
private final boolean mCheckSuccess;
private final Runnable mAdditionalChecks;
public ShellCommandRule(String cmd, @Nullable String revertCommand) {
public ShellCommandRule(String cmd, @Nullable String revertCommand, boolean checkSuccess,
Runnable additionalChecks) {
mCmd = cmd;
mRevertCommand = revertCommand;
mCheckSuccess = checkSuccess;
mAdditionalChecks = additionalChecks;
}
public ShellCommandRule(String cmd, @Nullable String revertCommand) {
this(cmd, revertCommand, false, null);
}
@Override
@@ -48,12 +63,27 @@ public class ShellCommandRule implements TestRule {
return new Statement() {
@Override
public void evaluate() throws Throwable {
UiDevice.getInstance(getInstrumentation()).executeShellCommand(mCmd);
final String result =
UiDevice.getInstance(getInstrumentation()).executeShellCommand(mCmd);
if (mCheckSuccess) {
Assert.assertTrue(
"Failed command: " + mCmd + ", result: " + result,
"Success".equals(result.replaceAll("\\s", "")));
}
if (mAdditionalChecks != null) mAdditionalChecks.run();
try {
base.evaluate();
} finally {
if (mRevertCommand != null) {
UiDevice.getInstance(getInstrumentation()).executeShellCommand(mRevertCommand);
final String revertResult = UiDevice.getInstance(
getInstrumentation()).executeShellCommand(
mRevertCommand);
if (mCheckSuccess) {
Assert.assertTrue(
"Failed command: " + mRevertCommand
+ ", result: " + revertResult,
"Success".equals(result.replaceAll("\\s", "")));
}
}
}
}
@@ -72,7 +102,15 @@ public class ShellCommandRule implements TestRule {
* Sets the target launcher as default launcher.
*/
public static ShellCommandRule setDefaultLauncher() {
return new ShellCommandRule(getLauncherCommand(getLauncherInMyProcess()), null);
final ActivityInfo launcher = getLauncherInMyProcess();
Log.d("b/187080582", "Launcher: " + new ComponentName(launcher.packageName, launcher.name)
.flattenToString());
return new ShellCommandRule(getLauncherCommand(launcher), null, true, () ->
Assert.assertEquals("Setting default launcher failed",
new ComponentName(launcher.packageName, launcher.name)
.flattenToString(),
PackageManagerWrapper.getInstance().getHomeActivities(new ArrayList<>())
.flattenToString()));
}
public static String getLauncherCommand(ActivityInfo launcher) {