DBunit、TestContext实践 【转】

1、定义接口UserDao.java
package com.bao.dbunit.dao;

import com.bao.dbunit.entity.User;

public interface UserDao {
   public User getUserByNick(String nick);
  
   public void save(User user);
  
   public void update(User user);
  
   public void remove(String nick);
}

Pojo类: User.java
package com.bao.dbunit.entity;

public class User {
   private String nick;
   private String password;

   public String getNick() {
     return nick;
  }

   public void setNick(String nick) {
     this.nick = nick;
  }

   public String getPassword() {
     return password;
  }

   public void setPassword(String password) {
     this.password = password;
  }

}



2、实现接口
1)帮助类:BaseDao.java
package com.bao.dbunit.dao.impl;

import org.springframework.jdbc.core.support.JdbcDaoSupport;

public class BaseDao extends JdbcDaoSupport {

}

2)实现类:DefaultUserDao.java
package com.bao.dbunit.dao.impl;

import java.sql.ResultSet;
import java.sql.SQLException;
import org.springframework.jdbc.core.RowMapper;
import com.bao.dbunit.dao.UserDao;
import com.bao.dbunit.entity.User;

public class DefaultUserDao extends BaseDao implements UserDao {

   private static String QUERY_BY_NICK = "select * from user where user.nick = ?";
  
   private static String REMOVE_USER = "delete from user where user.nick = ?";
  
   private static String INSERT_USER = "insert into user(nick,password) values(?, ?)";
  
   private static String UPDATE_USER = "update user set user.password = ? where user.nick = ?";
  @Override
   public User getUserByNick(String nick) {
     return (User) getJdbcTemplate().queryForObject(QUERY_BY_NICK, new Object[]{nick}, new RowMapper(){
      @Override
       public Object mapRow(ResultSet rs, int index) throws SQLException {
        User user = new User();
        user.setNick(rs.getString( "nick"));
        user.setPassword(rs.getString( "password"));
         return user;
      }
    });
  }

  @Override
   public void remove(String nick) {
    getJdbcTemplate().update(REMOVE_USER, new Object[]{nick});
  }

  @Override
   public void save(User user) {
    getJdbcTemplate().update(INSERT_USER, new Object[]{user.getNick(), user.getPassword()});
  }

  @Override
   public void update(User user) {
    getJdbcTemplate().update(UPDATE_USER, new Object[]{user.getPassword(), user.getNick()});
  }

}

3、相关配置:applicationContext.xml
<?xml version= "1.0" encoding= "UTF-8"?>
<beans xmlns= "http://www.springframework.org/schema/beans"
  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xmlns:aop="http://www.springframework.org/schema/aop"
  xmlns:tx="http://www.springframework.org/schema/tx"
  xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
    http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd
    http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx.xsd">

  <bean name="dataSource" class="org.springframework.jdbc.datasource.DriverManagerDataSource">
    <property name="driverClassName" value="com.mysql.jdbc.Driver"></property>
    <property name="url" value="jdbc:mysql://localhost/dbunit"></property>
    <property name="password" value="root" />
    <property name="username" value="root" />
  </bean>

  <bean name="jdbcTemplate" class="org.springframework.jdbc.core.JdbcTemplate">
    <property name="dataSource" ref="dataSource" />
  </bean>
  
  <bean name="baseDao" abstract="true" class="com.bao.dbunit.dao.impl.BaseDao">
    <property name="jdbcTemplate"    ref="jdbcTemplate"/>
  </bean>
  
  <bean name="userDao" parent="baseDao" class="com.bao.dbunit.dao.impl.DefaultUserDao" />
  
</beans>

sql脚本:ddl.sql
create table user(
  nick varchar(12) not null,
  password varchar(12) not null,
   primary key(nick)
);


4、进行测试:UserDaoTest.java
package com.bao.dbunit.dao;

import javax.sql.DataSource;

import org.dbunit.Assertion;
import org.dbunit. database.DatabaseConnection;
import org.dbunit. database.IDatabaseConnection;
import org.dbunit. database.QueryDataSet;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSet;
import org.dbunit.operation.DatabaseOperation;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.jdbc.datasource.DataSourceUtils;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;
import org.springframework.test.context. transaction.TransactionConfiguration;

import com.bao.dbunit.dao.UserDao;
import com.bao.dbunit.entity. User;


@ContextConfiguration(locations = { "classpath:testApplicationContext.xml" })
@TransactionConfiguration(defaultRollback = true)
public class UserDaoTest extends AbstractTransactionalJUnit4SpringContextTests {
  @Autowired
  private UserDao userDao;

  @Autowired
  private DataSource dataSource;

  private IDatabaseConnection conn;

  @Before
   public void initDbunit() throws Exception {
    conn = new DatabaseConnection(DataSourceUtils.getConnection(dataSource));
  }

  @Test
   public void saveUser() throws Exception {
     User user = new User();
     user.setNick("user001");
     user.setPassword("password001");
    userDao. save( user);

    QueryDataSet actual = new QueryDataSet(conn);
    actual.addTable(" user",
        " select * from user where user.nick = 'user001'");

    IDataSet expected = new FlatXmlDataSet(new ClassPathResource(
        "com/bao/dbunit/dao/user001.xml").getFile());

    Assertion.assertEquals(expected, actual);
  }

  @Test
   public void updateUser() throws Exception {

    IDataSet origen = new FlatXmlDataSet(new ClassPathResource(
        "com/bao/dbunit/dao/user001.xml").getFile());

    DatabaseOperation. INSERT. execute(conn, origen);

     User user = new User();
     user.setNick("user001");
     user.setPassword("password002");
    userDao. update( user);

    QueryDataSet actual = new QueryDataSet(conn);
    actual.addTable(" user",
        " select * from user where user.nick = 'user001'");

    IDataSet expected = new FlatXmlDataSet(new ClassPathResource(
        "com/bao/dbunit/dao/user001_updated.xml").getFile());

    Assertion.assertEquals(expected, actual);
  }

  @Test
   public void removeUser() throws Exception {
    IDataSet origen = new FlatXmlDataSet(new ClassPathResource(
        "com/bao/dbunit/dao/user001.xml").getFile());
    DatabaseOperation. INSERT. execute(conn, origen);

    userDao.remove("user001");

    QueryDataSet actual = new QueryDataSet(conn);
    actual.addTable(" user", " select * from user where nick = 'user001'");

    Assert.assertEquals(0, actual.getTable(" user").getRowCount());
    
  }
  
  @Test
   public void findUser() throws Exception {
    IDataSet data = new FlatXmlDataSet(new ClassPathResource(
        "com/bao/dbunit/dao/user001.xml").getFile());
    DatabaseOperation. INSERT. execute(conn, data);

    
     User user = userDao.getUserByNick("user001");

    Assert.assertEquals("password001", user.getPassword());
  }

}

测试相关的配置文件:
1)com.bao.dbunit.dao包下面 user001_updated.xml
<? xml version ="1.0" encoding ="UTF-8" ?>
< dataset >
   < user nick ="user001" password ="password002" />
</ dataset >

2)com.bao.dbunit.dao包下面 user001.xml
<? xml version ="1.0" encoding ="UTF-8" ?>
< dataset >
   < user nick ="user001" password ="password001" />
</ dataset >


3) testApplicationContext.xml
<? xml version ="1.0" encoding ="UTF-8" ?>
< beans xmlns ="http://www.springframework.org/schema/beans"
   xmlns:xsi ="http://www.w3.org/2001/XMLSchema-instance"
   xmlns:aop ="http://www.springframework.org/schema/aop"
   xmlns:tx ="http://www.springframework.org/schema/tx"
  xsi:schemaLocation="http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd
    http://www.springframework.org/schema/aop http://www.springframework.org/schema/aop/spring-aop.xsd
    http://www.springframework.org/schema/tx http://www.springframework.org/schema/tx/spring-tx.xsd" >

   < bean name ="dataSource" class ="org.springframework.jdbc.datasource.DriverManagerDataSource" >
     < property name ="driverClassName" value ="com.mysql.jdbc.Driver" > </ property >
     < property name ="url" value ="jdbc:mysql://localhost/dbunit" > </ property >
     < property name ="password" value ="root" />
     < property name ="username" value ="root" />
   </ bean >

   < bean name ="transactionManager" class ="org.springframework.jdbc.datasource.DataSourceTransactionManager" >
     < property name ="dataSource" ref ="dataSource" />
   </ bean >
  
   < tx:annotation-driven transaction-manager ="transactionManager" />
  
   < bean name ="jdbcTemplate" class ="org.springframework.jdbc.core.JdbcTemplate" >
     < property name ="dataSource" ref ="dataSource" />
   </ bean >
  
   < bean name ="baseDao" abstract ="true" class ="com.bao.dbunit.dao.impl.BaseDao" >
     < property name ="jdbcTemplate"     ref ="jdbcTemplate" />
   </ bean >
  
   < bean name ="userDao" parent ="baseDao" class ="com.bao.dbunit.dao.impl.DefaultUserDao" />
  
</ beans >


测试顺利通过,且数据库里面的数据进行了回滚

附录:pom.xml
   < dependencies >
     < dependency >
       < groupId >org.springframework </ groupId >
       < artifactId >spring </ artifactId >
       < version >2.5.6 </ version >
     </ dependency >
     < dependency >
       < groupId >mysql </ groupId >
       < artifactId >mysql-connector-java </ artifactId >
       < version >5.1.6 </ version >
     </ dependency >
     < dependency >
       < groupId >commons-logging </ groupId >
       < artifactId >commons-logging </ artifactId >
       < version >1.1.1 </ version >
     </ dependency >

     < dependency >
       < groupId >org.springframework </ groupId >
       < artifactId >spring-test </ artifactId >
       < version >2.5.6 </ version >
       < scope >test </ scope >
     </ dependency >
     < dependency >
       < groupId >junit </ groupId >
       < artifactId >junit </ artifactId >
       < version >4.4 </ version >
       < scope >test </ scope >
     </ dependency >
     < dependency >
       < groupId >org.easymock </ groupId >
       < artifactId >easymock </ artifactId >
       < version >2.4 </ version >
       < scope >test </ scope >
     </ dependency >
     < dependency >
       < groupId >org.dbunit </ groupId >
       < artifactId >dbunit </ artifactId >
       < version >2.4.6 </ version >
       < scope >test </ scope >
     </ dependency >
     < dependency >
       < groupId >org.slf4j </ groupId >
       < artifactId >slf4j-nop </ artifactId >
       < version >1.5.6 </ version >
       < scope >test </ scope >
     </ dependency >
   </ dependencies >
   < build >
     < plugins >
       < plugin >
         < artifactId >maven-compiler-plugin </ artifactId >
         < configuration >
           < target >1.6 </ target >
           < source >1.6 </ source >
           < encoding >UTF-8 </ encoding >
         </ configuration >
       </ plugin >
     </ plugins >
   </ build >

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页