@@ -419,29 +419,49 @@ void replacesArgumentSource() {
419419 //language=java
420420 java (
421421 """
422+ import org.junit.jupiter.api.extension.ExtensionContext;
423+ import org.junit.jupiter.params.provider.Arguments;
424+ import org.junit.jupiter.params.provider.ArgumentsProvider;
422425 import org.junit.jupiter.params.provider.ArgumentsSource;
423426 import org.junit.jupiter.api.Test;
427+ import java.util.stream.Stream;
424428 import static org.junit.jupiter.api.Assertions.*;
425-
429+
426430 class TestClass {
427431 @Test
428432 @ArgumentsSource(MyArgumentsProvider.class)
429433 void testWithArgumentsSource(String argument) {
430434 assertNotNull(argument);
431435 }
436+ static class MyArgumentsProvider implements ArgumentsProvider {
437+ @Override
438+ public Stream<? extends Arguments> provideArguments(ExtensionContext context) {
439+ return Stream.of("apple", "banana").map(Arguments::of);
440+ }
441+ }
432442 }
433443 """ ,
434444 """
445+ import org.junit.jupiter.api.extension.ExtensionContext;
435446 import org.junit.jupiter.params.ParameterizedTest;
447+ import org.junit.jupiter.params.provider.Arguments;
448+ import org.junit.jupiter.params.provider.ArgumentsProvider;
436449 import org.junit.jupiter.params.provider.ArgumentsSource;
450+ import java.util.stream.Stream;
437451 import static org.junit.jupiter.api.Assertions.*;
438-
452+
439453 class TestClass {
440454 @ParameterizedTest
441455 @ArgumentsSource(MyArgumentsProvider.class)
442456 void testWithArgumentsSource(String argument) {
443457 assertNotNull(argument);
444458 }
459+ static class MyArgumentsProvider implements ArgumentsProvider {
460+ @Override
461+ public Stream<? extends Arguments> provideArguments(ExtensionContext context) {
462+ return Stream.of("apple", "banana").map(Arguments::of);
463+ }
464+ }
445465 }
446466 """
447467 )
0 commit comments