| /******************************************************************************* |
| * Copyright (c) 2016, 2017 IBM Corporation and others. |
| * |
| * This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License 2.0 |
| * which accompanies this distribution, and is available at |
| * https://www.eclipse.org/legal/epl-2.0/ |
| * |
| * SPDX-License-Identifier: EPL-2.0 |
| * |
| * Contributors: |
| * IBM Corporation - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.jdt.internal.junit5.runner; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| import org.junit.platform.engine.Filter; |
| import org.junit.platform.engine.discovery.ClassNameFilter; |
| import org.junit.platform.engine.discovery.DiscoverySelectors; |
| import org.junit.platform.launcher.Launcher; |
| import org.junit.platform.launcher.LauncherDiscoveryRequest; |
| import org.junit.platform.launcher.TagFilter; |
| import org.junit.platform.launcher.core.LauncherDiscoveryRequestBuilder; |
| import org.junit.platform.launcher.core.LauncherFactory; |
| |
| import org.eclipse.jdt.internal.junit.runner.ITestLoader; |
| import org.eclipse.jdt.internal.junit.runner.ITestReference; |
| import org.eclipse.jdt.internal.junit.runner.RemoteTestRunner; |
| |
| public class JUnit5TestLoader implements ITestLoader { |
| |
| private Launcher fLauncher= LauncherFactory.create(); |
| |
| private RemoteTestRunner fRemoteTestRunner; |
| |
| @Override |
| public ITestReference[] loadTests(Class[] testClasses, String testName, String[] failureNames, String[] packages, String[][] includeExcludeTags, String uniqueId, RemoteTestRunner listener) { |
| fRemoteTestRunner= listener; |
| ITestReference[] refs= new ITestReference[0]; |
| if (uniqueId != null && !uniqueId.trim().isEmpty()) { |
| refs= new ITestReference[1]; |
| refs[0]= createUniqueIdTest(uniqueId, includeExcludeTags); |
| } else if (packages != null) { |
| refs= new ITestReference[packages.length]; |
| for (int i= 0; i < packages.length; i++) { |
| refs[i]= createTest(packages[i], includeExcludeTags); |
| } |
| } else { |
| refs= new ITestReference[testClasses.length]; |
| for (int i= 0; i < testClasses.length; i++) { |
| refs[i]= createTest(testClasses[i], testName, includeExcludeTags); |
| } |
| } |
| return refs; |
| } |
| |
| private ITestReference createTest(Class<?> clazz, String testName, String[][] includeExcludeTags) { |
| if (clazz == null) { |
| return null; |
| } |
| if (testName != null) { |
| return createFilteredTest(clazz, testName, includeExcludeTags); |
| } |
| return createUnfilteredTest(clazz, includeExcludeTags); |
| } |
| |
| private ITestReference createFilteredTest(Class<?> clazz, String testName, String[][] includeExcludeTags) { |
| LauncherDiscoveryRequest request= LauncherDiscoveryRequestBuilder.request().selectors(DiscoverySelectors.selectMethod(clazz.getName() + "#" + testName)).filters(getTagFilters(includeExcludeTags)).build(); //$NON-NLS-1$ |
| return new JUnit5TestReference(request, fLauncher, fRemoteTestRunner); |
| } |
| |
| private ITestReference createUnfilteredTest(Class<?> clazz, String[][] includeExcludeTags) { |
| LauncherDiscoveryRequest request= LauncherDiscoveryRequestBuilder.request().selectors(DiscoverySelectors.selectClass(clazz)).filters(getTagFilters(includeExcludeTags)).build(); |
| return new JUnit5TestReference(request, fLauncher, fRemoteTestRunner); |
| } |
| |
| private ITestReference createTest(String pkg, String[][] includeExcludeTags) { |
| if (pkg == null) { |
| return null; |
| } |
| String pattern; |
| if (pkg.equals("<default>")) { //$NON-NLS-1$ |
| pkg= ""; //$NON-NLS-1$ |
| pattern= "^[^.]+$"; //$NON-NLS-1$ |
| } else { |
| pattern= "^" + pkg + "\\.[^.]+$"; //$NON-NLS-1$//$NON-NLS-2$ |
| } |
| LauncherDiscoveryRequest request= LauncherDiscoveryRequestBuilder.request() |
| .selectors(DiscoverySelectors.selectPackage(pkg)) |
| .filters(ClassNameFilter.includeClassNamePatterns(pattern)) |
| .filters(getTagFilters(includeExcludeTags)) |
| .build(); |
| |
| return new JUnit5TestReference(request, fLauncher, fRemoteTestRunner); |
| } |
| |
| private ITestReference createUniqueIdTest(String uniqueId, String[][] includeExcludeTags) { |
| LauncherDiscoveryRequest request= LauncherDiscoveryRequestBuilder.request().selectors(DiscoverySelectors.selectUniqueId(uniqueId)).filters(getTagFilters(includeExcludeTags)).build(); |
| return new JUnit5TestReference(request, fLauncher, fRemoteTestRunner); |
| } |
| |
| private Filter<?>[] getTagFilters(String[][] includeExcludeTags) { |
| String[] includeTags= includeExcludeTags[0]; |
| String[] excludeTags= includeExcludeTags[1]; |
| List<Filter<?>> tagFilters= new ArrayList<>(); |
| if (includeTags != null) { |
| tagFilters.add(TagFilter.includeTags(includeTags)); |
| } |
| if (excludeTags != null) { |
| tagFilters.add(TagFilter.excludeTags(excludeTags)); |
| } |
| return tagFilters.toArray(new Filter[tagFilters.size()]); |
| } |
| } |