TestSpark is a plugin for generating unit tests. TestSpark natively integrates different AI-based test generation tools and techniques in the IDE.
TestSpark currently supports two test generation strategies:
- LLM-based test generation (using OpenAI, HuggingFace, and JetBrains internal AI Assistant platform)
- Local search-based test generation (using EvoSuite)
For this type of test generation, TestSpark sends request to different Large Language Models. Also, it automatically checks if tests are valid before presenting it to users.
This feature needs a token from OpenAI, HuggingFace, or the AI Assistant platform.
- Supports Java (any version) and Kotlin (K2 mode should be disabled, checkout the Settings section on README).
- Generates unit tests for capturing failures.
- Generate tests for Java classes, methods, and single lines.
For this type of test generation, TestSpark uses EvoSuite, which is the most powerful search-based local test generator.
- Supports up to Java 11.
- Generates tests for different test criteria: line coverage, branch coverage, I/O diversity, exception coverage, mutation score.
- Generates unit tests for capturing failures.
- Generate tests for Java classes, methods, and single lines.
Initially implemented by CISELab at SERG @ TU Delft, TestSpark is currently developed and maintained by ICTL at JetBrains Research.
TestSpark is currently designed to serve as an experimental tool. Please keep in mind that tests generated by TestSpark are meant to augment your existing test suites. They are not meant to replace writing tests manually.
If you are running the plugin for the first time, checkout the Settings section.
-
Using IDE built-in plugin system:
Settings/Preferences > Plugins > Marketplace > Search for "TestSpark" > Install Plugin
-
Manually:
Download the latest release and install it manually using Settings/Preferences > Plugins > ⚙️ > Install plugin from disk...
- Generating Tests
- Working with Test Cases
- Coverage
- Integrating tests into the project
- Settings
- Disable K2 for Kotlin Test Generation
- Telemetry
To initiate the generation process, right-click on the part of the code for which tests need to be generated and select TestSpark.
After that, a window will open where users need to configure generation settings.
Firstly users need to select the test generator (LLM-based test generator or EvoSuite, which is a Local search-based test generator).
Also, in this window, it is necessary to select the part of the code for which tests need to be generated. The selection consists of no more than three items -- class/interface, method/constructor (if applicable), line (if applicable).
After clicking the Next button, the plugin provides the opportunity to configure the basic parameters of the selected generator. Advanced parameter settings can be done in Settings. All settings, both in Settings and in this window, are saved, so you can disable the ability to configure generators before each generation process to perform this process more quickly.
In the case of LLM, two additional pages are provided for basic settings.
In the first page, users configure LLM Platform, LLM Token, LLM Model, LLM JUnit version, and Prompt Selection. More detailed descriptions of each item can be found in Settings.
After that, in the next page, you can provide some test samples for LLM.
Tests can be entered manually.
Also, tests can be chosen tests from the current project.
Test Cases can be modified, reset to their initial state, and deleted.
For EvoSuite, you need to enter the local path to Java 11 and select the generation algorithm, after which the generation process will start.
After configuring the test generators, click the OK button, after which the generation process will start, and a list of generated test cases will appear on the right side of the IDE.
During the test generation, users can observe the current state of the generation process.
After receiving the results, the user can interact with the test cases in various ways. They can view the result (whether it's passed or failed), also select, delete, modify, reset, like/dislike, fix by LLM and execute the tests to update the results.
Hitting the "Apply to test suite" button will add the selected tests to a test class of your choice.
Additionally, the top row of the tool window has buttons for selecting all tests, deselecting all tests, running all tests and removing them. The user also has an overview of how many tests they currently have selected and passed.
Users can select test cases.
Users can remove test cases.
Users can modify the code of test cases.
Users can reset the code to its original.
Users can reset the code to the last run.
Users can run the test to update the execution result.
Effortlessly identify passed and failed test cases with green and red color highlights for instant result comprehension is available. In case of failure, it is possible to find out the current error.
Users can copy the test.
Users can like/dislike the test for future analysis and improvement of the generation process.
Users can send a request to LLM with modification which users prefer for the test case.
Users can choose a default query, the list of which is set up in the Settings.
Users can also manually punch in a new request.
Once a test suite is generated, basic statistics about it can be seen in the tool window, coverage tab. The statistics include line coverage, branch coverage, weak mutation coverage. The table adjusts dynamically - it only calculates the statistics for the selected tests in the test suite.
Once test are generated, the lines which are covered by the tests will be highlighted (default color: green). The gutter next to the lines will have a green rectangle as well. If the rectangle is clicked, a popup will show the names of the tests which cover the selected line. If any of the test names are clicked, the corresponding test in the toolwindow will be highlighted with the same accent color. The effect lasts 10 seconds. Coverage visualisation adjusts dynamically - it only shows results for the tests that are selected in the TestSpark tab.
The tests can be added to an existing file:
Or to a new file:
For LLM-based Kotlin test generation, you need to disable the K2 mode for now.
The plugin is configured mainly through the Settings menu. The plugin settings can be found under Settings > Tools > TestSpark. Here, the user is able to select options for the plugin:
Before running the plugin for the first time, we highly recommend going to the Environment settings section of TestSpark settings. The settings include compilation path (path to compiled code) and compilation command. Both commands have defaults. However, we recommend especially that you check compilation command. For this command the user requires maven, gradle or any other builder program which can be accessed via command. Leaving this field with a faulty value may cause unintended behaviour.
The plugin supports changing the color for coverage visualisation and killed mutants visualisation (one setting for both). To change the color, go to Settings > Tools > TestSpark and use the color picker under Accessibility settings:
The plugin has been designed with translation in mind. The vast majority of the plugins labels, tooltips, messages, etc. is stored in .property files. For more information on translation, refer to the contributing readme.
The settings submenu Settings > Tools > TestSpark > EvoSuite allows the user to tweak EvoSuite parameters to their liking.
At the moment EvoSuite can be executed only with Java 11, so if the user has a more modern version by default, it is necessary to download Java 11 and set the path to the java file.
To accelerate the test generation process, users can disable the display of the EvoSuite Setup Page.
EvoSuite has hundreds of parameters, not all can be packed in a settings menu. However, the most commonly used and rational settings were added here:
The settings submenu Settings > Tools > TestSpark > LLM allows the user to tweak LLM parameters to their liking.
Selecting a platform to interact with the LLM. By default, only OpenAI is available, but for JetBrains employees there is an option to interact via Graize. More details in the TestSpark for JetBrains employees section.
Users have to set their own token for LLM, the plugin does not provide a default option.
Once the correct token is entered, it will be possible to select an LLM model for test generation.
In addition to the token, users are recommended to configure settings for the LLM process.
To expedite the test generation process, users can disable the display of the LLM Setup Page.
Additionally, they can also disable the display of the LLM Samples Page.
Users can customize the list of default requests to the LLM in test cases.
The plugin uses JUnit to generate tests. It is possible to select the JUnit version and prioritize the versions used by the current project.
Users have the opportunity to adjust the prompt that is sent to the LLM platform.
To create a request to the LLM, it is necessary to provide a prompt which contains information about the details of the generation. The most necessary data are located in the mandatory section, without them the prompt is not valid.
Users can modify, create new templates, delete and use different variants in practice.
❗ Pro tip: don't forget to hit the "save" button at the bottom. ❗
One of the biggest future plans of our client is to leverage the data that is gathered by TestSpark’s telemetry. This will help them with future research, including the development of an interactive way of using EvoSuite. The general idea behind this feature is to learn from the stored user corrections in order to improve test generation.
To opt into telemetry, go to Settings > Tools > TestSpark and tick the Enable telemetry checkbox. If you want, change the directory where telemetry is stored.
JetBrains employees have the ability to send queries to OpenAI models through the Grazie platform.
-
To include test generation using Grazie in the build process, you need to pass Space username and token as properties:
gradle buildPlugin -Dspace.username=<USERNAME> -Dspace.pass=<TOKEN>. -
To include test generation using Grazie in the runIdeForUiTests process, you need to pass Space username and token as properties:
gradle runIdeForUiTests -Dspace.username=<USERNAME> -Dspace.pass=<TOKEN>. -
<TOKEN>is generated by Space, which has access toAutomatically generating unit testsmaven packages.
Store Space username and token in ~/.gradle/gradle.properties
...
spaceUsername=<USERNAME>
spacePassword=<TOKEN>
...
LLM Settings with Grazie platform option:
The plugin is Open-Source and publicly hosted on github. Anyone can look into the code and suggest changes. You can find the plugin page here.
In addition, learn more about the structure of the plugin here.
Plugin based on the IntelliJ Platform Plugin Template.