public void testRunningJUnit() throws ExecutionException {
PsiClass testA = findTestA(getModule1());
JUnitConfiguration configuration = createConfiguration(testA);
JavaParameters parameters = checkCanRun(configuration);
assertEmpty(parameters.getVMParametersList().getList());
final SegmentedOutputStream notifications = new SegmentedOutputStream(System.out);
assertTrue(
JUnitStarter.checkVersion(
parameters.getProgramParametersList().getArray(), new PrintStream(notifications)));
assertTrue(parameters.getProgramParametersList().getList().contains(testA.getQualifiedName()));
assertEquals(JUnitStarter.class.getName(), parameters.getMainClass());
assertEquals(myJdk.getHomeDirectory().getPresentableUrl(), parameters.getJdkPath());
}
protected void initialize() throws ExecutionException {
JavaParametersUtil.configureConfiguration(myJavaParameters, myConfiguration);
myJavaParameters.setMainClass(JUnitConfiguration.JUNIT_START_CLASS);
final Module module = myConfiguration.getConfigurationModule().getModule();
if (myJavaParameters.getJdk() == null) {
myJavaParameters.setJdk(
module != null
? ModuleRootManager.getInstance(module).getSdk()
: ProjectRootManager.getInstance(myProject).getProjectSdk());
}
myJavaParameters.getClassPath().add(JavaSdkUtil.getIdeaRtJarPath());
myJavaParameters.getClassPath().add(PathUtil.getJarPathForClass(JUnitStarter.class));
myJavaParameters
.getProgramParametersList()
.add(JUnitStarter.IDE_VERSION + JUnitStarter.VERSION);
for (RunConfigurationExtension ext :
Extensions.getExtensions(RunConfigurationExtension.EP_NAME)) {
ext.updateJavaParameters(myConfiguration, myJavaParameters, getRunnerSettings());
}
final Object[] listeners = Extensions.getExtensions(IDEAJUnitListener.EP_NAME);
final StringBuilder buf = new StringBuilder();
for (final Object listener : listeners) {
boolean enabled = true;
for (RunConfigurationExtension ext :
Extensions.getExtensions(RunConfigurationExtension.EP_NAME)) {
if (ext.isListenerDisabled(myConfiguration, listener, getRunnerSettings())) {
enabled = false;
break;
}
}
if (enabled) {
final Class classListener = listener.getClass();
buf.append(classListener.getName()).append("\n");
myJavaParameters.getClassPath().add(PathUtil.getJarPathForClass(classListener));
}
}
if (buf.length() > 0) {
try {
myListenersFile = FileUtil.createTempFile("junit_listeners_", "");
myListenersFile.deleteOnExit();
myJavaParameters.getProgramParametersList().add("@@" + myListenersFile.getPath());
FileUtil.writeToFile(myListenersFile, buf.toString().getBytes());
} catch (IOException e) {
LOG.error(e);
}
}
}
protected void createServerSocket(JavaParameters javaParameters) {
try {
myServerSocket = new ServerSocket(0, 0, InetAddress.getByName("127.0.0.1"));
javaParameters.getProgramParametersList().add("-socket" + myServerSocket.getLocalPort());
} catch (IOException e) {
LOG.error(e);
}
}
protected void createTempFiles(JavaParameters javaParameters) {
try {
myTempFile = FileUtil.createTempFile("idea_" + getFrameworkId(), ".tmp");
myTempFile.deleteOnExit();
passTempFile(javaParameters.getProgramParametersList(), myTempFile.getAbsolutePath());
} catch (IOException e) {
LOG.error(e);
}
}
private static List<String> extractAllInPackageTests(
JavaParameters parameters, PsiPackage psiPackage) throws IOException {
String filePath =
ContainerUtil.find(
parameters.getProgramParametersList().getArray(),
value ->
StringUtil.startsWithChar(value, '@') && !StringUtil.startsWith(value, "@w@"))
.substring(1);
List<String> lines = readLinesFrom(new File(filePath));
assertEquals(psiPackage.getQualifiedName(), lines.get(0));
// lines.remove(0);
lines.remove(0);
return lines;
}
private void appendForkInfo(Executor executor) throws ExecutionException {
final String forkMode = myConfiguration.getForkMode();
if (Comparing.strEqual(forkMode, "none")) {
return;
}
if (getRunnerSettings().getData() != null) {
final String actionName = executor.getActionName();
throw new CantRunException(
actionName
+ " is disabled in fork mode.<br/>Please change fork mode to <none> to "
+ actionName.toLowerCase()
+ ".");
}
final JavaParameters javaParameters = getJavaParameters();
final Sdk jdk = javaParameters.getJdk();
if (jdk == null) {
throw new ExecutionException(
ExecutionBundle.message("run.configuration.error.no.jdk.specified"));
}
try {
final File tempFile = FileUtil.createTempFile("command.line", "", true);
final PrintWriter writer = new PrintWriter(tempFile, "UTF-8");
try {
writer.println(((JavaSdkType) jdk.getSdkType()).getVMExecutablePath(jdk));
for (String vmParameter : javaParameters.getVMParametersList().getList()) {
writer.println(vmParameter);
}
writer.println("-classpath");
writer.println(javaParameters.getClassPath().getPathsString());
} finally {
writer.close();
}
myJavaParameters
.getProgramParametersList()
.add("@@@" + forkMode + ',' + tempFile.getAbsolutePath());
} catch (Exception e) {
LOG.error(e);
}
}
@Override
protected JavaParameters createJavaParameters() throws ExecutionException {
final JavaParameters javaParameters = new JavaParameters();
final Module module = getConfiguration().getConfigurationModule().getModule();
Project project = getConfiguration().getProject();
Sdk jdk =
module == null
? ProjectRootManager.getInstance(project).getProjectSdk()
: ModuleRootManager.getInstance(module).getSdk();
javaParameters.setJdk(jdk);
final Object[] patchers = Extensions.getExtensions(ExtensionPoints.JUNIT_PATCHER);
for (Object patcher : patchers) {
((JUnitPatcher) patcher).patchJavaParameters(module, javaParameters);
}
// Append coverage parameters if appropriate
for (RunConfigurationExtension ext :
Extensions.getExtensions(RunConfigurationExtension.EP_NAME)) {
ext.updateJavaParameters(getConfiguration(), javaParameters, getRunnerSettings());
}
final String parameters = getConfiguration().getProgramParameters();
getConfiguration().setProgramParameters(null);
try {
JavaParametersUtil.configureConfiguration(javaParameters, getConfiguration());
} finally {
getConfiguration().setProgramParameters(parameters);
}
javaParameters.getClassPath().addFirst(JavaSdkUtil.getIdeaRtJarPath());
configureClasspath(javaParameters);
if (!StringUtil.isEmptyOrSpaces(parameters)) {
javaParameters.getProgramParametersList().add("@name" + parameters);
}
return javaParameters;
}
protected <T> void addClassesListToJavaParameters(
Collection<? extends T> elements,
Function<T, String> nameFunction,
String packageName,
boolean createTempFile,
boolean junit4) {
try {
if (createTempFile) {
myTempFile = FileUtil.createTempFile("idea_junit", ".tmp");
myTempFile.deleteOnExit();
myJavaParameters.getProgramParametersList().add("@" + myTempFile.getAbsolutePath());
}
final PrintWriter writer = new PrintWriter(myTempFile, "UTF-8");
try {
writer.println(packageName);
final List<String> testNames = new ArrayList<String>();
for (final T element : elements) {
final String name = nameFunction.fun(element);
if (name == null) {
LOG.error("invalid element " + element);
return;
}
testNames.add(name);
}
Collections.sort(testNames); // sort tests in FQN order
for (String testName : testNames) {
writer.println(testName);
}
} finally {
writer.close();
}
} catch (IOException e) {
LOG.error(e);
}
}
protected JavaParameters createJavaParameters() throws ExecutionException {
log.debug("createJavaParameters");
SubstepsRunConfiguration runConfig =
(SubstepsRunConfiguration) this.getEnvironment().getRunProfile();
SubstepsRunnerConfigurationModel model = runConfig.getModel();
JavaParameters params = model.getJavaParameters();
params.setWorkingDirectory(model.getWorkingDir());
if (params.getJdk() == null) {
log.debug("params on the jdk is null");
params.setJdk(
JavaSdkImpl.getInstance().createJdk(model.getVersionString(), model.getHomePath()));
params.getClassPath().add(model.getClassPathString());
}
params.getProgramParametersList().add("prog-args-env", "prg-localhost");
ParametersList vmParametersList = params.getVMParametersList();
vmParametersList.addParametersString("-Dfile.encoding=UTF-8");
vmParametersList.addParametersString("-Dcom.sun.management.jmxremote.port=" + jmxPort);
vmParametersList.addParametersString("-Dcom.sun.management.jmxremote.authenticate=false");
vmParametersList.addParametersString("-Dcom.sun.management.jmxremote.ssl=false");
vmParametersList.addParametersString("-Djava.rmi.server.hostname=localhost");
// String rmiClasspathString = "\"file://" +
// model.getClassPathString().replaceAll(File.pathSeparator, " file://") + "\"";
// log.debug("rmi classpath: " + rmiClasspathString);
// vmParametersList.addParametersString("-Djava.rmi.server.codebase=" +
// rmiClasspathString);
vmParametersList.addParametersString("-Dsun.io.serialization.extendedDebugInfo=true");
// System.setProperty("java.rmi.server.codebase", rmiClasspathString);
// createServerSocket(params);
//
//
// if (myServerSocket != null) {
// params.getProgramParametersList().add(ForkedDebuggerHelper.DEBUG_SOCKET +
// myServerSocket.getLocalPort());
// }
//
log.debug(
"launching substeps runner with classpath: "
+ params.getClassPath().getPathsString()
+ "\njvm info: "
+ model.getHomePath()
+ " version: "
+ model.getVersionString());
return params;
}
