lysu'''

<dependency>
  <groupId>com.h2database</groupId>
  <artifactId>h2</artifactId>
  <version>1.3.160</version>
  <scope>test</scope>
</dependency>

<jdbc:embedded-database id="dataSource">
  <jdbc:script location="classpath:schema.sql"/>
  <jdbc:script location="classpath:test-data.sql"/>
</jdbc:embedded-database>

public final class H2DatabaseUtil {
  private static final Logger logger = LoggerFactory.getLogger(H2DatabaseUtil.class);
  
  private H2DatabaseUtil() {
  }

  public static DataSource getDataSource() {
    if (logger.isDebugEnabled()) {
      logger.debug("get H2 data source.");
    }
    SimpleDriverDataSource dataSource = new SimpleDriverDataSource();
    dataSource.setDriverClass(org.h2.Driver.class);
    dataSource.setUsername("sa");
    dataSource.setPassword("");
    dataSource.setUrl("jdbc:h2:mem:testdb;MODE=MYSQL;DB_CLOSE_DELAY=-1");
    return dataSource;
  }

  public static void initSchema(DataSource ds) {
    if (logger.isDebugEnabled()) {
      logger.debug("start init schema.");
    }
    try {
      QueryRunner queryRunner = new QueryRunner(ds);
      queryRunner.update("drop all objects");
      if (logger.isDebugEnabled()) {
       logger.debug("delete all object k.o.");
      }
      try {
        queryRunner.update("runscript from '"
             + new DefaultResourceLoader()
             .getResource("schema.sql").getURL().toString() + "'");
        if (logger.isDebugEnabled()) {
         logger.debug("rebuild the schema finish! f~resh meat...");
        }
      } catch (IOException e) {
        logger.error(e.getMessage(), e);
      }
    } catch (SQLException e) {
      logger.error(e.getMessage(),e);
    }
  }  
}

這里使用了DbUtils來(lái)簡(jiǎn)化建表語(yǔ)句的執(zhí)行，至于DataSource類型的選擇，可以根據(jù)情況自己更換。

然后可以創(chuàng)建一個(gè)Test的基類讓測(cè)試代碼方便使用dataSource

public abstract class AbstractDBAwareTestCase {
  protected QueryRunner queryRunner ;
  protected DataSource dataSource;
  
  protected void setUp() throws Exception {
    dataSource = getDataSource();
    initSchema(dataSource);
    queryRunner = new QueryRunner(dataSource);
  }
}

之后寫(xiě)的單元測(cè)試只需繼承此類就可以獲取已建好庫(kù)的數(shù)據(jù)進(jìn)行測(cè)試。

在得到數(shù)據(jù)源后，我們要做的操作是用Dbunit向這個(gè)嶄新的數(shù)據(jù)庫(kù)里頭灌入數(shù)據(jù)。

覺(jué)得Dbunit還是不夠好用？這里推薦直接copy更改spring-side的Fixtures類

（http://springside.googlecode.com/svn/springside4/trunk/modules/core/src/main/java/org/springside/modules/test/data/Fixtures.java）

這個(gè)類有個(gè)問(wèn)題是需要對(duì)spring的resource相關(guān)類有依賴（獲取classpath下的文件之類的），如果有問(wèn)題還需更改下。

最后看個(gè)簡(jiǎn)單Test的使用列子：

public class AgentRepositoryTest extends AbstractDBAwareTestCase {
  private IAgentRepository agentRepository;

  @Before
  public void setUp() throws Exception {
    super.setUp();
    reloadAllTable(dataSource, "/data/agent-data.xml");
    agentRepository = Lookup.getDefault().lookup(IAgentRepository.class);
  }
  
  @Test
  public void should_find_agent_by_id() throws Exception {
    Agent agent = agentRepository.findById(dataSource, 1);
    assertNotNull(agent);
    assertThat(agent.getSimpleName(), is("demoa"));
  }
｝