使用 Spring 2.5 TestContext 测试框架

虽然 TestUserService.handleUserLogin() 测试方法已经可以成功运行，但是它在测试功能上是不完善的，读者朋友可以已经发现了它存在以下两个问题：

我们仅仅执行了 UserService#handleUserLogin(user) 方法，但验证该方法执行结果的正确性。

在测试方法中直接使用 ID 为 1 的 User 对象进行测试，这相当于要求在数据库 t_user 表必须已经存在 ID 为 1 的记录，如果 t_user 中不存在这条记录，将导致测试方法执行失败。

准备测试数据并检测运行结果

在这节里，我们将着手解决上面所提出的两个问题，在测试用例中准备测试数据并到数据库中检测业务执行结果的正确性。

准备测试数据

相比于在测试方法中直接访问预定的数据记录，在测试方法执行前通过程序准备一些测试数据，然后在此基础上运行测试方法是比较好的策略，因为后者不需要对数据库的状态做假设。在 TestContext 中，你可以通过使用 JUnit 4.4 的 @Before 注解达到这个目的，请看下面的代码：

清单5. 为测试方法准备数据
package　com.baobaotao.service;　
　
import　java.sql.Connection;　
import　java.sql.PreparedStatement;　
import　java.sql.SQLException;　
import　java.util.Date;　
　
import　org.junit.Before;　
import　org.junit.Test;　
import　org.springframework.beans.factory.annotation.Autowired;　
import　org.springframework.jdbc.core.PreparedStatementCreator;　
import　org.springframework.jdbc.support.GeneratedKeyHolder;　
import　org.springframework.jdbc.support.KeyHolder;　
import　org.springframework.test.context.ContextConfiguration;　
import　org.springframework.test.context.junit4.　
AbstractTransactionalJUnit4SpringContextTests;　
　
import　com.baobaotao.dao.UserDao;　
import　com.baobaotao.domain.User;　
　
@ContextConfiguration　
public　class　TestUserService　
extends　AbstractTransactionalJUnit4SpringContextTests　{　
　　@Autowired　
　　private　UserService　userService;　
　　
　　@Autowired　
　　private　UserDao　userDao;　
　　
　　private　int　userId;　
　　
　　@Before　//①　准备测试数据　
　　public　void　prepareTestData()　{　
　　　　final　String　sql　=　"insert　into　t_user(user_name,password)　values('tom','1234')";　
　　　　simpleJdbcTemplate.update(sql);　
　　　　KeyHolder　keyHolder　=　new　GeneratedKeyHolder();　
　　　　simpleJdbcTemplate.getJdbcOperations().update(　
　　　　　　new　PreparedStatementCreator()　{　
　　　　　　　　public　PreparedStatement　createPreparedStatement(Connection　conn)　
　　　　　　　　　　throws　SQLException　{　
　　　　　　　　　　PreparedStatement　ps　=　conn.prepareStatement(sql);　
　　　　　　　　　　return　ps;　
　　　　　　　　}　
　　　　　　},　keyHolder);　
　　　　userId　=　keyHolder.getKey().intValue();//①－1　记录测试数据的id　
　　}　
　　
　　@Test　
　　public　void　handleUserLogin(){　
　　　　User　user　=　userDao.getUserById(userId);　//②　获取测试数据　
　　　　user.setLastIp("127.0.0.1");　
　　　　Date　now　=　new　Date();　
　　　　user.setLastVisit(now.getTime());　
　　　　userService.handleUserLogin(user);　
　　}　
}