Java JavaTestConfigurationBase类代码示例

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
public void attachToProcess(@NotNull final RunConfigurationBase configuration, @NotNull final ProcessHandler handler, @Nullable RunnerSettings runnerSettings) {
    super.attachToProcess(configuration, handler, runnerSettings);
    if (!(runnerSettings instanceof DynatraceRunnerSettings)) {
        return;
    }
    if (!(handler instanceof OSProcessHandler)) {
        return;
    }

    DynatraceConfigurableStorage executionSettings = DynatraceConfigurableStorage.getOrCreateStorage(configuration);
    handler.putCopyableUserData(PROFILE_KEY, executionSettings.getSystemProfile());

    OSProcessHandler procHandler = (OSProcessHandler) handler;
    handler.addProcessListener(new DynatraceProcessListener(executionSettings.getSystemProfile(), configuration.getProject()));

    if(!(configuration instanceof JavaTestConfigurationBase)) {
        return;
    }

    Matcher matcher = TRID_EXTRACTOR.matcher(procHandler.getCommandLine());
    if (!matcher.find()) {
        return;
    }
    handler.putCopyableUserData(TRID_KEY, matcher.group(1));
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
private static void processAvailableTraces(RunConfigurationBase configuration)
{
	final String tracesDirectory = getTracesDirectory(configuration);
	final TestDiscoveryIndex coverageIndex = TestDiscoveryIndex.getInstance(configuration.getProject());
	synchronized(ourTracesLock)
	{
		final File tracesDirectoryFile = new File(tracesDirectory);
		final File[] testMethodTraces = tracesDirectoryFile.listFiles((dir, name) -> name.endsWith(".tr"));
		if(testMethodTraces != null)
		{
			for(File testMethodTrace : testMethodTraces)
			{
				try
				{
					coverageIndex.updateFromTestTrace(testMethodTrace, ((JavaTestConfigurationBase) configuration).getConfigurationModule().getModuleName(), ((JavaTestConfigurationBase)
							configuration).getFrameworkPrefix());
					FileUtil.delete(testMethodTrace);
				}
				catch(IOException e)
				{
					LOG.error("Can not load " + testMethodTrace, e);
				}
			}

			final String[] filesInTracedDirectories = tracesDirectoryFile.list();
			if(filesInTracedDirectories == null || filesInTracedDirectories.length == 0)
			{
				FileUtil.delete(tracesDirectoryFile);
			}
		}
	}
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
protected void attachToProcess(@NotNull final RunConfigurationBase configuration,
                               @NotNull final ProcessHandler handler,
                               @Nullable RunnerSettings runnerSettings) {
  if (runnerSettings == null && isApplicableFor(configuration)) {
    final Alarm processTracesAlarm = new Alarm(Alarm.ThreadToUse.POOLED_THREAD, null);
    final MessageBusConnection connection = configuration.getProject().getMessageBus().connect();
    connection.subscribe(SMTRunnerEventsListener.TEST_STATUS, new SMTRunnerEventsAdapter() {
      private List<String> myCompletedMethodNames = new ArrayList<String>();
      @Override
      public void onTestFinished(@NotNull SMTestProxy test) {
        final SMTestProxy.SMRootTestProxy root = test.getRoot();
        if ((root == null || root.getHandler() == handler)) {
          final String fullTestName = test.getLocationUrl();
          if (fullTestName != null && fullTestName.startsWith(JavaTestLocator.TEST_PROTOCOL)) {
            myCompletedMethodNames.add(((JavaTestConfigurationBase)configuration).getFrameworkPrefix() + fullTestName.substring(JavaTestLocator.TEST_PROTOCOL.length() + 3));
            if (myCompletedMethodNames.size() > 50) {
              final String[] fullTestNames = ArrayUtil.toStringArray(myCompletedMethodNames);
              myCompletedMethodNames.clear();
              processTracesAlarm.addRequest(new Runnable() {
                @Override
                public void run() {
                  processAvailableTraces(configuration, fullTestNames);
                }
              }, 100);
            }
          }
        }
      }

      @Override
      public void onTestingFinished(@NotNull SMTestProxy.SMRootTestProxy testsRoot) {
        if (testsRoot.getHandler() == handler) {
          processTracesAlarm.cancelAllRequests();
          processTracesAlarm.addRequest(new Runnable() {
            @Override
            public void run() {
              processAvailableTraces(configuration);
              Disposer.dispose(processTracesAlarm);
            }
          }, 0);
          connection.disconnect();
        }
      }
    });
  }
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
protected boolean isApplicableFor(@NotNull final RunConfigurationBase configuration) {
  return configuration instanceof JavaTestConfigurationBase && Registry.is("testDiscovery.enabled");
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
public void updateJavaParameters(RunConfigurationBase configuration, JavaParameters javaParameters, RunnerSettings runnerSettings) throws ExecutionException {
    //Check if the action was ran by DynatraceExecutor
    if (!(runnerSettings instanceof DynatraceRunnerSettings)) {
        return;
    }
    //get global settings
    DynatraceSettingsProvider.State settings = DynatraceSettingsProvider.getInstance().getState();

    try {
        DynatraceClient client = SDKClient.getInstance();
        //get local (runconfiguration) settings
        DynatraceConfigurableStorage executionSettings = DynatraceConfigurableStorage.getOrCreateStorage(configuration);

        SessionStorage ss = configuration.getProject().getComponent(SessionStorage.class);
        if (executionSettings.isRecordSessionPerLaunch() && !ss.isRecording(executionSettings.getSystemProfile())) {
            ss.startRecording(executionSettings.getSystemProfile());
        }

        StringBuilder builder = new StringBuilder("-agentpath:");
        builder.append(settings.getAgent().getAgentLibrary()).append('=');
        builder.append("wait=").append(settings.getServer().getTimeout()).append(',');
        builder.append("name=").append(executionSettings.getAgentName()).append(',');
        builder.append("server=").append(settings.getAgent().getCollectorHost()).append(':');
        builder.append(settings.getAgent().getCollectorPort());

        if (executionSettings.getAdditionalParameters() != null && !executionSettings.getAdditionalParameters().isEmpty()) {
            builder.append(',').append(executionSettings.getAdditionalParameters());
        }

        if(configuration instanceof JavaTestConfigurationBase) {
            Calendar now = Calendar.getInstance();

            //fetch test id
            TestAutomation testAutomation = new TestAutomation(client);
            CreateTestRunRequest request = new CreateTestRunRequest(executionSettings.getSystemProfile(), String.valueOf(new SimpleDateFormat("HH:mm:ss").format(now.getTime())));
            request.setVersionMajor(String.valueOf(now.get(Calendar.YEAR)));
            request.setVersionMinor(String.valueOf(now.get(Calendar.MONTH) + 1));
            request.setVersionRevision(String.valueOf(now.get(Calendar.DAY_OF_WEEK)));

            TestRun testRun = testAutomation.createTestRun(request);

            builder.append(',').append("optionTestRunIdJava=").append(testRun.getId());
            IDEADescriptor.getInstance().log(Level.INFO, "TestRun", "", Messages.getMessage("execution.configuration.tests.running", testRun.getId()), false);
        }

        //mutate java parameters
        javaParameters.getVMParametersList().add(builder.toString());
    } catch (Exception e) {
        //IDEADescriptor.getInstance(configuration.getProject()).log(Level.SEVERE, Messages.getMessage("notifications.error.title"), "", Messages.getMessage("notifications.error.configuration"), true);
        throw new ExecutionException(e);
    }
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
protected Module findModule(JavaTestConfigurationBase configuration, Module contextModule)
{
	return null;
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
public boolean isConfigurationFromContext(JavaTestConfigurationBase configuration, ConfigurationContext configurationContext)
{
	final Pair<String, String> position = getPosition(getSourceMethod(configurationContext.getLocation()));
	return position != null && position.equals(getPosition(configuration));
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
protected void attachToProcess(@NotNull final RunConfigurationBase configuration, @NotNull final ProcessHandler handler, @Nullable RunnerSettings runnerSettings)
{
	if(runnerSettings == null && isApplicableFor(configuration))
	{
		final String frameworkPrefix = ((JavaTestConfigurationBase) configuration).getFrameworkPrefix();
		final String moduleName = ((JavaTestConfigurationBase) configuration).getConfigurationModule().getModuleName();

		Disposable disposable = Disposer.newDisposable();
		final Alarm processTracesAlarm = new Alarm(Alarm.ThreadToUse.POOLED_THREAD, disposable);
		final MessageBusConnection connection = configuration.getProject().getMessageBus().connect();
		connection.subscribe(SMTRunnerEventsListener.TEST_STATUS, new SMTRunnerEventsAdapter()
		{
			private List<String> myCompletedMethodNames = new ArrayList<>();

			@Override
			public void onTestFinished(@NotNull SMTestProxy test)
			{
				final SMTestProxy.SMRootTestProxy root = test.getRoot();
				if((root == null || root.getHandler() == handler))
				{
					final String fullTestName = test.getLocationUrl();
					if(fullTestName != null && fullTestName.startsWith(JavaTestLocator.TEST_PROTOCOL))
					{
						myCompletedMethodNames.add(frameworkPrefix + fullTestName.substring(JavaTestLocator.TEST_PROTOCOL.length() + 3));
						if(myCompletedMethodNames.size() > 50)
						{
							final String[] fullTestNames = ArrayUtil.toStringArray(myCompletedMethodNames);
							myCompletedMethodNames.clear();
							processTracesAlarm.addRequest(() -> processAvailableTraces(fullTestNames, getTracesDirectory(configuration), moduleName, frameworkPrefix, TestDiscoveryIndex
									.getInstance(configuration.getProject())), 100);
						}
					}
				}
			}

			@Override
			public void onTestingFinished(@NotNull SMTestProxy.SMRootTestProxy testsRoot)
			{
				if(testsRoot.getHandler() == handler)
				{
					processTracesAlarm.cancelAllRequests();
					processTracesAlarm.addRequest(() ->
					{
						processAvailableTraces(configuration);
						Disposer.dispose(disposable);
					}, 0);
					connection.disconnect();
				}
			}
		});
	}
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
@Override
protected boolean isApplicableFor(@NotNull final RunConfigurationBase configuration)
{
	return configuration instanceof JavaTestConfigurationBase && TESTDISCOVERY_ENABLED;
}
 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
protected abstract void setPosition(JavaTestConfigurationBase configuration, PsiLocation<PsiMethod> position); 

import com.intellij.execution.JavaTestConfigurationBase; //导入依赖的package包/类
protected abstract Pair<String, String> getPosition(JavaTestConfigurationBase configuration);