Можно ли улучшить тесты, например сократить количество кода?

Да, тесты можно улучшить и сократить количество кода несколькими способами:

• Использовать параметризованные тесты (например, с JUnit 5 @ParameterizedTest), чтобы избежать дублирования кода при проверке разных входных данных.
• Вынести повторяющиеся настройки и инициализацию в методы с аннотациями @BeforeEach или @BeforeAll.
• Применять библиотеки для упрощения мокирования и ассертов, например, Mockito и AssertJ, которые делают код тестов более выразительным и компактным.
• Использовать вспомогательные методы или фабрики для создания тестовых данных, чтобы не писать их вручную в каждом тесте.

Пример параметризованного теста с JUnit 5:

@ParameterizedTest
@ValueSource(strings = {"input1", "input2", "input3"})
void testWithMultipleInputs(String input) {
    // тестируемый код
    assertNotNull(process(input));
}