Skip to content

Commit 83fd44e

Browse files
committed
Fix failing test case due to type attribution validation
1 parent bcd31ce commit 83fd44e

File tree

1 file changed

+22
-2
lines changed

1 file changed

+22
-2
lines changed

src/test/java/org/openrewrite/java/testing/junit5/AddParameterizedTestAnnotationTest.java

Lines changed: 22 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)