diff options
Diffstat (limited to 'src/vogar/target')
-rw-r--r-- | src/vogar/target/CaliperRunnerFactory.java | 3 | ||||
-rw-r--r-- | src/vogar/target/MainRunnerFactory.java | 3 | ||||
-rw-r--r-- | src/vogar/target/RunnerFactory.java | 2 | ||||
-rw-r--r-- | src/vogar/target/TestRunner.java | 16 | ||||
-rw-r--r-- | src/vogar/target/junit/ExcludeFilter.java | 49 | ||||
-rw-r--r-- | src/vogar/target/junit/JUnitRunnerFactory.java | 5 | ||||
-rw-r--r-- | src/vogar/target/junit/JUnitTargetRunner.java | 12 | ||||
-rw-r--r-- | src/vogar/target/testng/TestNgRunnerFactory.java | 2 | ||||
-rw-r--r-- | src/vogar/target/testng/TestNgTargetRunner.java | 1 |
9 files changed, 84 insertions, 9 deletions
diff --git a/src/vogar/target/CaliperRunnerFactory.java b/src/vogar/target/CaliperRunnerFactory.java index f1dab97..2955c74 100644 --- a/src/vogar/target/CaliperRunnerFactory.java +++ b/src/vogar/target/CaliperRunnerFactory.java @@ -51,7 +51,8 @@ public class CaliperRunnerFactory implements RunnerFactory { @Override @Nullable public TargetRunner newRunner(TargetMonitor monitor, String qualification, Class<?> klass, AtomicReference<String> skipPastReference, - TestEnvironment testEnvironment, int timeoutSeconds, String[] args) { + TestEnvironment testEnvironment, int timeoutSeconds, String[] excludeFilters, + String[] args) { if (benchmarkClassChecker != null && benchmarkClassChecker.isBenchmark(klass)) { return new CaliperTargetRunner(monitor, klass, args); } else { diff --git a/src/vogar/target/MainRunnerFactory.java b/src/vogar/target/MainRunnerFactory.java index cca463f..41280c3 100644 --- a/src/vogar/target/MainRunnerFactory.java +++ b/src/vogar/target/MainRunnerFactory.java @@ -28,7 +28,8 @@ public class MainRunnerFactory implements RunnerFactory { @Override @Nullable public TargetRunner newRunner(TargetMonitor monitor, String qualification, Class<?> klass, AtomicReference<String> skipPastReference, - TestEnvironment testEnvironment, int timeoutSeconds, String[] args) { + TestEnvironment testEnvironment, int timeoutSeconds, String[] excludeFilters, + String[] args) { if (new ClassAnalyzer(klass).hasMethod(true, void.class, "main", String[].class)) { return new MainTargetRunner(monitor, klass, args); } else { diff --git a/src/vogar/target/RunnerFactory.java b/src/vogar/target/RunnerFactory.java index 993acd4..e90c070 100644 --- a/src/vogar/target/RunnerFactory.java +++ b/src/vogar/target/RunnerFactory.java @@ -32,5 +32,5 @@ public interface RunnerFactory { @Nullable TargetRunner newRunner(TargetMonitor monitor, String qualification, Class<?> klass, AtomicReference<String> skipPastReference, TestEnvironment testEnvironment, - int timeoutSeconds, String[] args); + int timeoutSeconds, String[] excludeFilters, String[] args); } diff --git a/src/vogar/target/TestRunner.java b/src/vogar/target/TestRunner.java index 7edc5b5..5b6f539 100644 --- a/src/vogar/target/TestRunner.java +++ b/src/vogar/target/TestRunner.java @@ -52,6 +52,7 @@ public final class TestRunner { private final RunnerFactory runnerFactory; private final String[] args; + private final String[] excludeFilters; private boolean useSocketMonitor; public TestRunner(Properties properties, List<String> argsList) { @@ -60,6 +61,8 @@ public final class TestRunner { int monitorPort = Integer.parseInt(properties.getProperty(TestProperties.MONITOR_PORT)); String skipPast = null; + List<String> excludeFilters = new ArrayList<>(); + for (Iterator<String> i = argsList.iterator(); i.hasNext(); ) { String arg = i.next(); @@ -73,6 +76,11 @@ public final class TestRunner { skipPast = i.next(); i.remove(); } + if (arg.equals("--exclude-filter")) { + i.remove(); + excludeFilters.add(i.next()); + i.remove(); + } } // Select the RunnerFactory instances to use based on the selected runner type. @@ -95,6 +103,7 @@ public final class TestRunner { this.monitorPort = monitorPort; this.skipPastReference = new AtomicReference<>(skipPast); + this.excludeFilters = excludeFilters.toArray(String[]::new); this.args = argsList.toArray(new String[argsList.size()]); } @@ -201,7 +210,7 @@ public final class TestRunner { TargetRunner targetRunner; try { targetRunner = runnerFactory.newRunner(monitor, qualification, klass, - skipPastReference, testEnvironment, timeoutSeconds, args); + skipPastReference, testEnvironment, timeoutSeconds, excludeFilters, args); } catch (RuntimeException e) { monitor.outcomeStarted(klass.getName()); e.printStackTrace(); @@ -246,10 +255,11 @@ public final class TestRunner { @Override @Nullable public TargetRunner newRunner(TargetMonitor monitor, String qualification, Class<?> klass, AtomicReference<String> skipPastReference, - TestEnvironment testEnvironment, int timeoutSeconds, String[] args) { + TestEnvironment testEnvironment, int timeoutSeconds, String[] excludeFilters, + String[] args) { for (RunnerFactory runnerFactory : runnerFactories) { TargetRunner targetRunner = runnerFactory.newRunner(monitor, qualification, klass, - skipPastReference, testEnvironment, timeoutSeconds, args); + skipPastReference, testEnvironment, timeoutSeconds, excludeFilters, args); if (targetRunner != null) { return targetRunner; } diff --git a/src/vogar/target/junit/ExcludeFilter.java b/src/vogar/target/junit/ExcludeFilter.java new file mode 100644 index 0000000..1e98f3d --- /dev/null +++ b/src/vogar/target/junit/ExcludeFilter.java @@ -0,0 +1,49 @@ +/* + * 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 vogar.target.junit; + +import org.junit.runner.Description; +import org.junit.runner.manipulation.Filter; + +import java.util.HashSet; +import java.util.List; +import java.util.Set; + +public class ExcludeFilter extends Filter { + + private final Set<String> excludeFilters; + + public ExcludeFilter(String[] excludeFilters) { + Set<String> set = new HashSet<>(); + if (excludeFilters != null) { + set.addAll(List.of(excludeFilters)); + } + this.excludeFilters = set; + } + + @Override + public boolean shouldRun(Description description) { + return !description.getAnnotations().stream() + .anyMatch(annotation -> + excludeFilters.contains(annotation.annotationType().getName())); + } + + @Override + public String describe() { + return "ExcludeFilter[" + String.join(",", excludeFilters) + "]"; + } +} diff --git a/src/vogar/target/junit/JUnitRunnerFactory.java b/src/vogar/target/junit/JUnitRunnerFactory.java index 2719356..3a83576 100644 --- a/src/vogar/target/junit/JUnitRunnerFactory.java +++ b/src/vogar/target/junit/JUnitRunnerFactory.java @@ -30,10 +30,11 @@ public class JUnitRunnerFactory implements RunnerFactory { @Override @Nullable public TargetRunner newRunner(TargetMonitor monitor, String qualification, Class<?> klass, AtomicReference<String> skipPastReference, - TestEnvironment testEnvironment, int timeoutSeconds, String[] args) { + TestEnvironment testEnvironment, int timeoutSeconds, String[] excludeFilters, + String[] args) { if (supports(klass)) { return new JUnitTargetRunner(monitor, skipPastReference, testEnvironment, - timeoutSeconds, klass, qualification, args); + timeoutSeconds, klass, qualification, excludeFilters, args); } else { return null; } diff --git a/src/vogar/target/junit/JUnitTargetRunner.java b/src/vogar/target/junit/JUnitTargetRunner.java index 3fd2232..906115a 100644 --- a/src/vogar/target/junit/JUnitTargetRunner.java +++ b/src/vogar/target/junit/JUnitTargetRunner.java @@ -44,19 +44,23 @@ public final class JUnitTargetRunner implements TargetRunner { private final Class<?> testClass; private final RunnerParams runnerParams; + private final String[] excludeFilters; + public JUnitTargetRunner(TargetMonitor monitor, AtomicReference<String> skipPastReference, TestEnvironment testEnvironment, int timeoutSeconds, Class<?> testClass, - String qualification, String[] args) { + String qualification, String[] excludeFilters, String[] args) { this.monitor = monitor; this.skipPastReference = skipPastReference; this.testEnvironment = testEnvironment; this.testClass = testClass; + this.excludeFilters = excludeFilters; TimeoutAndAbortRunRule timeoutRule = new TimeoutAndAbortRunRule(timeoutSeconds); runnerParams = new RunnerParams(qualification, args, timeoutRule); } + @Override public boolean run() { // Use JUnit infrastructure to run the tests. RunnerBuilder builder = new VogarRunnerBuilder(runnerParams); @@ -79,6 +83,12 @@ public final class JUnitTargetRunner implements TargetRunner { } try { + new ExcludeFilter(excludeFilters).apply(runner); + } catch (NoTestsRemainException e) { + return true; + } + + try { JUnitCore core = new JUnitCore(); // The TestEnvironmentRunListener resets static state between tests. core.addListener(new TestEnvironmentRunListener(testEnvironment)); diff --git a/src/vogar/target/testng/TestNgRunnerFactory.java b/src/vogar/target/testng/TestNgRunnerFactory.java index 2e214fd..f6cd279 100644 --- a/src/vogar/target/testng/TestNgRunnerFactory.java +++ b/src/vogar/target/testng/TestNgRunnerFactory.java @@ -35,6 +35,7 @@ public class TestNgRunnerFactory implements RunnerFactory { AtomicReference<String> skipPastReference, TestEnvironment testEnvironment, int timeoutSeconds, + String[] excludeFilters, String[] args) { if (supports(klass)) { return new TestNgTargetRunner( @@ -44,6 +45,7 @@ public class TestNgRunnerFactory implements RunnerFactory { timeoutSeconds, klass, qualification, + excludeFilters, args); } else { return null; diff --git a/src/vogar/target/testng/TestNgTargetRunner.java b/src/vogar/target/testng/TestNgTargetRunner.java index 6b10509..40824c7 100644 --- a/src/vogar/target/testng/TestNgTargetRunner.java +++ b/src/vogar/target/testng/TestNgTargetRunner.java @@ -39,6 +39,7 @@ public class TestNgTargetRunner implements TargetRunner { int timeoutSeconds, Class<?> testClass, String qualification, + String[] excludeFilters, String[] args) { this.monitor = monitor; this.testClass = testClass; |