88 * http://www.eclipse.org/legal/epl-v10.html
99 */
1010
11- package org .junit .jupiter .migrationsupport .rules ;
11+ package org .junit .jupiter .migrationsupport .parameterized ;
1212
1313import static java .util .Collections .singletonList ;
1414import static java .util .function .Function .identity ;
2727import java .util .stream .IntStream ;
2828import java .util .stream .Stream ;
2929
30- import org .junit .jupiter .api .extension .BeforeTestExecutionCallback ;
3130import org .junit .jupiter .api .extension .ContainerExtensionContext ;
3231import org .junit .jupiter .api .extension .Extension ;
3332import org .junit .jupiter .api .extension .ExtensionContext ;
3433import org .junit .jupiter .api .extension .ParameterContext ;
3534import org .junit .jupiter .api .extension .ParameterResolutionException ;
3635import org .junit .jupiter .api .extension .ParameterResolver ;
37- import org .junit .jupiter .api .extension .TestExtensionContext ;
36+ import org .junit .jupiter .api .extension .TestInstancePostProcessor ;
3837import org .junit .jupiter .api .extension .TestTemplateInvocationContext ;
3938import org .junit .jupiter .api .extension .TestTemplateInvocationContextProvider ;
4039import org .junit .platform .commons .meta .API ;
@@ -215,15 +214,15 @@ public List<Extension> getAdditionalExtensions() {
215214 };
216215 }
217216
218- private static class InjectionExtension implements BeforeTestExecutionCallback {
217+ private static class InjectionExtension implements TestInstancePostProcessor {
219218 private final Object [] parameters ;
220219
221220 public InjectionExtension (Object [] parameters ) {
222221 this .parameters = parameters ;
223222 }
224223
225224 @ Override
226- public void beforeTestExecution ( TestExtensionContext context ) throws Exception {
225+ public void postProcessTestInstance ( Object testInstance , ExtensionContext context ) throws Exception {
227226 List <Field > parameters = parametersFields (context );
228227
229228 if (!parameters .isEmpty () && parameters .size () != this .parameters .length ) {
@@ -233,7 +232,7 @@ public void beforeTestExecution(TestExtensionContext context) throws Exception {
233232 for (Field param : parameters ) {
234233 Parameterized .Parameter annotation = param .getAnnotation (Parameterized .Parameter .class );
235234 int paramIndex = annotation .value ();
236- param .set (context . getTestInstance () , this .parameters [paramIndex ]);
235+ param .set (testInstance , this .parameters [paramIndex ]);
237236 }
238237 }
239238 }
0 commit comments