Best junit code snippet using org.junit.runners.Annotation Type Parameterized.AfterParam
Source:BlockJUnit4ClassRunnerWithParameters.java
1package org.junit.runners.parameterized;2import java.lang.annotation.Annotation;3import java.lang.reflect.Field;4import java.util.List;5import org.junit.internal.runners.statements.RunAfters;6import org.junit.internal.runners.statements.RunBefores;7import org.junit.runner.RunWith;8import org.junit.runner.notification.RunNotifier;9import org.junit.runners.BlockJUnit4ClassRunner;10import org.junit.runners.Parameterized;11import org.junit.runners.Parameterized.Parameter;12import org.junit.runners.model.FrameworkField;13import org.junit.runners.model.FrameworkMethod;14import org.junit.runners.model.InitializationError;15import org.junit.runners.model.Statement;16/**17 * A {@link BlockJUnit4ClassRunner} with parameters support. Parameters can be18 * injected via constructor or into annotated fields.19 */20public class BlockJUnit4ClassRunnerWithParameters extends21 BlockJUnit4ClassRunner {22 private enum InjectionType {23 CONSTRUCTOR, FIELD24 }25 private final Object[] parameters;26 private final String name;27 public BlockJUnit4ClassRunnerWithParameters(TestWithParameters test)28 throws InitializationError {29 super(test.getTestClass());30 parameters = test.getParameters().toArray(31 new Object[test.getParameters().size()]);32 name = test.getName();33 }34 @Override35 public Object createTest() throws Exception {36 InjectionType injectionType = getInjectionType();37 switch (injectionType) {38 case CONSTRUCTOR:39 return createTestUsingConstructorInjection();40 case FIELD:41 return createTestUsingFieldInjection();42 default:43 throw new IllegalStateException("The injection type "44 + injectionType + " is not supported.");45 }46 }47 private Object createTestUsingConstructorInjection() throws Exception {48 return getTestClass().getOnlyConstructor().newInstance(parameters);49 }50 private Object createTestUsingFieldInjection() throws Exception {51 List<FrameworkField> annotatedFieldsByParameter = getAnnotatedFieldsByParameter();52 if (annotatedFieldsByParameter.size() != parameters.length) {53 throw new Exception(54 "Wrong number of parameters and @Parameter fields."55 + " @Parameter fields counted: "56 + annotatedFieldsByParameter.size()57 + ", available parameters: " + parameters.length58 + ".");59 }60 Object testClassInstance = getTestClass().getJavaClass().newInstance();61 for (FrameworkField each : annotatedFieldsByParameter) {62 Field field = each.getField();63 Parameter annotation = field.getAnnotation(Parameter.class);64 int index = annotation.value();65 try {66 field.set(testClassInstance, parameters[index]);67 } catch (IllegalAccessException e) {68 IllegalAccessException wrappedException = new IllegalAccessException(69 "Cannot set parameter '" + field.getName()70 + "'. Ensure that the field '" + field.getName()71 + "' is public.");72 wrappedException.initCause(e);73 throw wrappedException;74 } catch (IllegalArgumentException iare) {75 throw new Exception(getTestClass().getName()76 + ": Trying to set " + field.getName()77 + " with the value " + parameters[index]78 + " that is not the right type ("79 + parameters[index].getClass().getSimpleName()80 + " instead of " + field.getType().getSimpleName()81 + ").", iare);82 }83 }84 return testClassInstance;85 }86 @Override87 protected String getName() {88 return name;89 }90 @Override91 protected String testName(FrameworkMethod method) {92 return method.getName() + getName();93 }94 @Override95 protected void validateConstructor(List<Throwable> errors) {96 validateOnlyOneConstructor(errors);97 if (getInjectionType() != InjectionType.CONSTRUCTOR) {98 validateZeroArgConstructor(errors);99 }100 }101 @Override102 protected void validateFields(List<Throwable> errors) {103 super.validateFields(errors);104 if (getInjectionType() == InjectionType.FIELD) {105 List<FrameworkField> annotatedFieldsByParameter = getAnnotatedFieldsByParameter();106 int[] usedIndices = new int[annotatedFieldsByParameter.size()];107 for (FrameworkField each : annotatedFieldsByParameter) {108 int index = each.getField().getAnnotation(Parameter.class)109 .value();110 if (index < 0 || index > annotatedFieldsByParameter.size() - 1) {111 errors.add(new Exception("Invalid @Parameter value: "112 + index + ". @Parameter fields counted: "113 + annotatedFieldsByParameter.size()114 + ". Please use an index between 0 and "115 + (annotatedFieldsByParameter.size() - 1) + "."));116 } else {117 usedIndices[index]++;118 }119 }120 for (int index = 0; index < usedIndices.length; index++) {121 int numberOfUse = usedIndices[index];122 if (numberOfUse == 0) {123 errors.add(new Exception("@Parameter(" + index124 + ") is never used."));125 } else if (numberOfUse > 1) {126 errors.add(new Exception("@Parameter(" + index127 + ") is used more than once (" + numberOfUse + ")."));128 }129 }130 }131 }132 @Override133 protected Statement classBlock(RunNotifier notifier) {134 Statement statement = childrenInvoker(notifier);135 statement = withBeforeParams(statement);136 statement = withAfterParams(statement);137 return statement;138 }139 private Statement withBeforeParams(Statement statement) {140 List<FrameworkMethod> befores = getTestClass()141 .getAnnotatedMethods(Parameterized.BeforeParam.class);142 return befores.isEmpty() ? statement : new RunBeforeParams(statement, befores);143 }144 private class RunBeforeParams extends RunBefores {145 RunBeforeParams(Statement next, List<FrameworkMethod> befores) {146 super(next, befores, null);147 }148 @Override149 protected void invokeMethod(FrameworkMethod method) throws Throwable {150 int paramCount = method.getMethod().getParameterTypes().length;151 method.invokeExplosively(null, paramCount == 0 ? (Object[]) null : parameters);152 }153 }154 private Statement withAfterParams(Statement statement) {155 List<FrameworkMethod> afters = getTestClass()156 .getAnnotatedMethods(Parameterized.AfterParam.class);157 return afters.isEmpty() ? statement : new RunAfterParams(statement, afters);158 }159 private class RunAfterParams extends RunAfters {160 RunAfterParams(Statement next, List<FrameworkMethod> afters) {161 super(next, afters, null);162 }163 @Override164 protected void invokeMethod(FrameworkMethod method) throws Throwable {165 int paramCount = method.getMethod().getParameterTypes().length;166 method.invokeExplosively(null, paramCount == 0 ? (Object[]) null : parameters);167 }168 }169 @Override170 protected Annotation[] getRunnerAnnotations() {171 Annotation[] allAnnotations = super.getRunnerAnnotations();172 Annotation[] annotationsWithoutRunWith = new Annotation[allAnnotations.length - 1];173 int i = 0;174 for (Annotation annotation: allAnnotations) {175 if (!annotation.annotationType().equals(RunWith.class)) {176 annotationsWithoutRunWith[i] = annotation;177 ++i;178 }179 }180 return annotationsWithoutRunWith;181 }182 private List<FrameworkField> getAnnotatedFieldsByParameter() {183 return getTestClass().getAnnotatedFields(Parameter.class);184 }185 private InjectionType getInjectionType() {186 if (fieldsAreAnnotated()) {187 return InjectionType.FIELD;188 } else {189 return InjectionType.CONSTRUCTOR;190 }191 }192 private boolean fieldsAreAnnotated() {193 return !getAnnotatedFieldsByParameter().isEmpty();194 }195}...
Annotation Type Parameterized.AfterParam
Using AI Code Generation
1import org.junit.runners.Parameterized;2import org.junit.runners.Parameterized.Parameters;3import org.junit.runners.Parameterized.Parameter;4import org.junit.runner.RunWith;5import org.junit.Test;6import static org.junit.Assert.assertEquals;7@RunWith(Parameterized.class)8public class AfterParamTest {9 private int input;10 private int expected;11 @Parameter(value = 0)12 public void setInput(int input) {13 this.input = input;14 }15 @Parameter(value = 1)16 public void setExpected(int expected) {17 this.expected = expected;18 }19 public static Object[][] data() {20 return new Object[][] {21 {1, 1},22 {2, 2},23 {3, 3},24 {4, 4},25 {5, 5}26 };27 }28 public void test() {29 assertEquals(expected, input);30 }31}32OK (1 test)
Annotation Type Parameterized.AfterParam
Using AI Code Generation
1import org.junit.runners.Parameterized;2public class ParameterizedExample {3 public static Iterable<Object[]> data1() {4 return Arrays.asList(new Object[][] { { 1, 1 }, { 2, 2 }, { 8, 8 }, { 4, 4 } });5 }6 @Parameterized.Parameter(0)7 public int fInput;8 @Parameterized.Parameter(1)9 public int fExpected;10 public void test() {11 assertEquals(fExpected, fInput);12 }13}14import org.junit.runners.Parameterized;15public class ParameterizedExample {16 public static Iterable<Object[]> data1() {17 return Arrays.asList(new Object[][] { { 1, 1 }, { 2, 2 }, { 8, 8 }, { 4, 4 } });18 }19 @Parameterized.Parameter(0)20 public int fInput;21 @Parameterized.Parameter(1)22 public int fExpected;23 public void test() {24 assertEquals(fExpected, fInput);25 }26}27import org.junit.runners.Parameterized;28public class ParameterizedExample {29 public static Iterable<Object[]> data1() {30 return Arrays.asList(new Object[][] { { 1, 1 }, { 2, 2 }, { 8, 8 }, { 4, 4 } });31 }32 @Parameterized.Parameter(0)33 public int fInput;34 @Parameterized.Parameter(1)35 public int fExpected;36 public void test() {37 assertEquals(fExpected, fInput);38 }39}40import org.junit.runners.Parameterized;41public class ParameterizedExample {42 public static Iterable<Object[]> data1() {43 return Arrays.asList(new Object[][] { { 1, 1 }, { 2, 2 }, { 8, 8 }, { 4, 4 } });44 }45 @Parameterized.Parameter(0)46 public int fInput;47 @Parameterized.Parameter(1)
Annotation Type Parameterized.AfterParam
Using AI Code Generation
1import org.junit.runners.Parameterized;2import org.junit.runners.Parameterized.AfterParam;3public class ParameterizedTest {4 public static void afterParam(Object param) {5 System.out.println("After Parameterized Test: " + param);6 }7}8import org.junit.runners.Parameterized;9import org.junit.runners.Parameterized.BeforeParam;10public class ParameterizedTest {11 public static void beforeParam(Object param) {12 System.out.println("Before Parameterized Test: " + param);13 }14}15import org.junit.runners.Parameterized;16import org.junit.runners.Parameterized.Parameters;17public class ParameterizedTest {18 public static Collection<Object[]> data() {19 return Arrays.asList(new Object[][] { { 0, false }, { 5, false }, { 6, true }, { 10, true }, { 11, false }, });20 }21 private int fInput;22 private boolean fExpected;23 public ParameterizedTest(int input, boolean expected) {24 fInput = input;25 fExpected = expected;26 }27 public void test() {28 assertEquals(fExpected, fInput % 2 == 0);29 }30}31import org.junit.runners.Parameterized;32import org.junit.runners.Parameterized.UseParametersRunnerFactory;33public class ParameterizedTest {34 @UseParametersRunnerFactory(CustomRunnerFactory.class)35 public static class CustomRunnerFactory implements ParametersRunnerFactory {36 public Runner createRunnerForTestWithParameters(TestWithParameters test) throws InitializationError {37 return new BlockJUnit4ClassRunnerWithParameters(test);38 }39 }40}41import org.junit.runners.Parameterized;42import org.junit.runners.Parameterized.UseParametersRunnerFactory;43public class ParameterizedTest {44 @UseParametersRunnerFactory(CustomRunnerFactory.class)45 public static class CustomRunnerFactory implements ParametersRunnerFactory {46 public Runner createRunnerForTestWithParameters(TestWithParameters test) throws InitializationError {47 return new BlockJUnit4ClassRunnerWithParameters(test);48 }49 }50}
Annotation Type Parameterized.AfterParam
Using AI Code Generation
1@RunWith(Parameterized.class)2public class TestRunner {3 private String name;4 private int age;5 private String address;6 public TestRunner(String name, int age, String address) {7 this.name = name;8 this.age = age;9 this.address = address;10 }11 public static Collection<Object[]> data() {12 Object[][] data = new Object[][] { { "John", 20, "USA" }, { "Peter", 25, "UK" }, { "Mark", 30, "China" } };13 return Arrays.asList(data);14 }15 public void test() {16 System.out.println("Name: " + name + ", Age: " + age + ", Address: " + address);17 }18}
LambdaTest also has a detailed JUnit tutorial explaining its features, importance, advanced use cases, best practices, and more to help you get started with running your automation testing scripts.
Here are the detailed JUnit testing chapters to help you get started:
You can also check out our JUnit certification if you wish to take your career in Selenium automation testing with JUnit to the next level.
Get 100 minutes of automation test minutes FREE!!