JUnit: @Before only for some test methods?
JavaUnit TestingJunitJava Problem Overview
I have some common set up code that I've factored out to a method marked with @Before
. However, it is not necessary for all this code to run for every single test. Is there a way to mark it so the @Before
method only runs before certain tests?
Java Solutions
Solution 1 - Java
Just move out the tests that don't need the setup code into a separate test class. If you have some other code common to the tests that would be helpful to keep, move that out into a helper class.
Solution 2 - Java
@Nested + @BeforeEach
Totally agree with the point of moving the related code to an inner class. So here what I have done.
- Create an inner class inside your test class
- Annotate the inner class with @Nested
- Move all the test methods you want to use in the inner class
- Write the init code inside the inner class and annotate it with @ForEach
Here is the code:
class Testing {
@Test
public void testextmethod1() {
System.out.println("test ext method 1");
}
@Nested
class TestNest{
@BeforeEach
public void init() {
System.out.println("Init");
}
@Test
public void testmethod1() {
System.out.println("This is method 1");
}
@Test
public void testmethod2() {
System.out.println("This is method 2");
}
@Test
public void testmethod3() {
System.out.println("This is method 3");
}
}
@Test
public void testextmethod2() {
System.out.println("test ext method 2");
}
}
Here is the output
test ext method 1
test ext method 2
Init
This is method 1
Init
This is method 2
Init
This is method 3
Note: I am not sure if this is supported in Junit4. I am doing this in JUnit5
Solution 3 - Java
It is possible to achieve also via Assume from JUnit. And then you can check the method name for which you want to process @Before
.
public class MyTest {
@Rule
public TestName testName = new TestName();
@Before
public void setUp() {
assumeTrue(testName.getMethodName().equals("myMethodName"));
// setup follows
}
}
Check the topic for more insights about @Rule
.
Solution 4 - Java
Or use TestNG. It gives you finer grained control over tests.
Solution 5 - Java
Not sure about @Before, but I recently came up with a strategy for @After block to run selectively. The implementation was straight forward. I have some flags set to default values as part of the test class. They are reset to default values in @Before class. In the class I need to do things specific to a flag, I set those flags & in @After I check for flag values to do the respective jobs.
Solution 6 - Java
JUnit 4.12 provide Enclosed Runner like
@RunWith(Enclosed.class)
public class GlobalTest{
@RunWith(MockitoJUnitRunner.class)
public class InnerTest{
}
}