You can do that by instantiating your main class with parametrized constructor in your junit and then creating a spy from it.
Let’s suppose your main class is A
. Where B
and C
are its dependencies
public class A {
private B b;
private C c;
public A(B b,C c)
{
this.b=b;
this.c=c;
}
void method() {
System.out.println("A's method called");
b.method();
c.method();
System.out.println(method2());
}
protected int method2() {
return 10;
}
}
Then you can write junit for this using your parametrized class as below
@RunWith(MockitoJUnitRunner.class)
public class ATest {
A a;
@Mock
B b;
@Mock
C c;
@Test
public void test() {
a=new A(b, c);
A spyA=Mockito.spy(a);
doReturn(20).when(spyA).method2();
spyA.method();
}
}
Output of test class
A's method called
20
- Here
B
andC
are mocked object that you injected in your classA
using parametrized constructor. - Then we created a
spy
ofA
calledspyA
. - We checked if
spy
is really working by modifying the return value of a protected methodmethod2
in classA
which could not have been possible ifspyA
was not an actualspy
ofA
.