我正在尝试调用此类的第二个方法,并且始终获取null
。请注意,它返回new User()
,但是在测试类中,我总是得到null
:
@Stateless
public class UserDAO2 {
public Connection getConnFromPool(int i) {
return null;
}
public User readByUserid(String s) {
System.out.println("In DAO 2");
Connection c = getConnFromPool(1);
return new User();
}
}
和测试类:
@RunWith(MockitoJUnitRunner.class)
public class UserBeanUnitTest {
@InjectMocks
private UserDAO2 dao2;
@Before
public void setup() {
dao2 = Mockito.mock(UserDAO2.class);
MockitoAnnotations.initMocks(this);
}
@Test
public void testBean() {
Mockito.when(dao2.getConnFromPool(1)).thenReturn(null);
User expectedUser = new User();
expectedUser.setSk(1);
expectedUser.setFirstName("David");
expectedUser.setLastName("Gahan");
expectedUser.setUserid("user1");
User user = dao2.readByUserid("user1"); // <-- this method always returns null
assertThat(user).isEqualTo(expectedUser); // <-- test fails as expectedUser != null
}
}
也请注意,System.out.println
从不打印。如何解决此问题以实际拨打dao.readByUserid()
?
[如果需要测试某个类的方法,并且在其中要调用要模拟的同一类的另一个方法,则需要使用@Spy
:
@RunWith(MockitoJUnitRunner.class)
public class UserDAO2Test {
@InjectMocks
@Spy
private UserDAO2 dao;
@Test
public void testBean() {
Mockito.doReturn(null).when(dao).getConnFromPool(1);
User expectedUser = new User();
expectedUser.setSk(1);
expectedUser.setFirstName("David");
expectedUser.setLastName("Gahan");
expectedUser.setUserid("user1");
User user = dao.readByUserid("user1");
assertThat(user).isEqualTo(expectedUser);
}
}
请注意,我使用模拟getConnFromPool
稍微修改了该行,因为使用该技术时需要它。
请参见docs进行监视。