Vadim Tryshev
d17a270b31
Now when exiting the test, we finish Launcher activity and force the leak check. I've seen this helping one of the leaks to stay within the test that caused it instead of failing the next test. Bug: 187761685 Flag: N/A Test: presubmit, local Change-Id: I7eac26f749433c37a6880a2f035ab1ba9c5fef6e
54 lines
1.6 KiB
Java
54 lines
1.6 KiB
Java
/*
|
|
* Copyright (C) 2023 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.util.rule;
|
|
|
|
import androidx.annotation.NonNull;
|
|
|
|
import com.android.launcher3.ui.AbstractLauncherUiTest;
|
|
|
|
import org.junit.rules.TestRule;
|
|
import org.junit.runner.Description;
|
|
import org.junit.runners.model.Statement;
|
|
|
|
/**
|
|
* Isolates tests from some of the state created by the previous test.
|
|
*/
|
|
public class TestIsolationRule implements TestRule {
|
|
final AbstractLauncherUiTest mTest;
|
|
|
|
public TestIsolationRule(AbstractLauncherUiTest test) {
|
|
mTest = test;
|
|
}
|
|
|
|
@NonNull
|
|
@Override
|
|
public Statement apply(@NonNull Statement base, @NonNull Description description) {
|
|
return new Statement() {
|
|
@Override
|
|
public void evaluate() throws Throwable {
|
|
mTest.finishLauncherActivity();
|
|
mTest.checkDetectedLeaks();
|
|
try {
|
|
base.evaluate();
|
|
} finally {
|
|
mTest.finishLauncherActivity();
|
|
}
|
|
mTest.checkDetectedLeaks();
|
|
}
|
|
};
|
|
}
|
|
}
|