Merge pull request #12535 from dreis2211:intellij-modified-classpath-runner

* pr/12535:
  Fix ModifiedClassPathRunner tests if run via IDEA

Author: snicoll

spring-boot-tools/spring-boot-test-support/src/main/java/org/springframework/boot/junit/runner/classpath/ModifiedClassPathRunner.java

@@ -27,6 +27,7 @@
 import java.util.List;
 import java.util.jar.Attributes;
 import java.util.jar.JarFile;
+import java.util.regex.Pattern;
 
 import org.apache.maven.repository.internal.MavenRepositorySystemUtils;
 import org.eclipse.aether.DefaultRepositorySystemSession;
@@ -64,6 +65,9 @@
  */
 public class ModifiedClassPathRunner extends BlockJUnit4ClassRunner {
 
+	private static final Pattern INTELLIJ_CLASSPATH_JAR_PATTERN = Pattern.compile(
+			".*classpath(\\d+)?.jar");
+
 	public ModifiedClassPathRunner(Class<?> testClass) throws InitializationError {
 		super(testClass);
 	}
@@ -102,7 +106,7 @@ private URLClassLoader createTestClassLoader(Class<?> testClass) throws Exceptio
 	private URL[] extractUrls(URLClassLoader classLoader) throws Exception {
 		List<URL> extractedUrls = new ArrayList<URL>();
 		for (URL url : classLoader.getURLs()) {
-			if (isSurefireBooterJar(url)) {
+			if (isManifestOnlyJar(url)) {
 				extractedUrls.addAll(extractUrlsFromManifestClassPath(url));
 			}
 			else {
@@ -112,10 +116,30 @@ private URL[] extractUrls(URLClassLoader classLoader) throws Exception {
 		return extractedUrls.toArray(new URL[extractedUrls.size()]);
 	}
 
+	private boolean isManifestOnlyJar(URL url) {
+		return isSurefireBooterJar(url) || isShortenedIntelliJJar(url);
+	}
+
 	private boolean isSurefireBooterJar(URL url) {
 		return url.getPath().contains("surefirebooter");
 	}
 
+	private boolean isShortenedIntelliJJar(URL url) {
+		String urlPath = url.getPath();
+		boolean isCandidate = INTELLIJ_CLASSPATH_JAR_PATTERN.matcher(urlPath).matches();
+		if (isCandidate) {
+			try {
+				Attributes attributes = getManifestMainAttributesFromUrl(url);
+				String createdBy = attributes.getValue("Created-By");
+				return createdBy != null && createdBy.contains("IntelliJ");
+			}
+			catch (Exception ex) {
+				return false;
+			}
+		}
+		return false;
+	}
+
 	private List<URL> extractUrlsFromManifestClassPath(URL booterJar) throws Exception {
 		List<URL> urls = new ArrayList<URL>();
 		for (String entry : getClassPath(booterJar)) {
@@ -125,10 +149,15 @@ private List<URL> extractUrlsFromManifestClassPath(URL booterJar) throws Excepti
 	}
 
 	private String[] getClassPath(URL booterJar) throws Exception {
-		JarFile jarFile = new JarFile(new File(booterJar.toURI()));
+		Attributes attributes = getManifestMainAttributesFromUrl(booterJar);
+		return StringUtils.delimitedListToStringArray(attributes
+				.getValue(Attributes.Name.CLASS_PATH), " ");
+	}
+
+	private Attributes getManifestMainAttributesFromUrl(URL url) throws Exception {
+		JarFile jarFile = new JarFile(new File(url.toURI()));
 		try {
-			return StringUtils.delimitedListToStringArray(jarFile.getManifest()
-					.getMainAttributes().getValue(Attributes.Name.CLASS_PATH), " ");
+			return jarFile.getManifest().getMainAttributes();
 		}
 		finally {
 			jarFile.close();