为了账号安全，请及时绑定邮箱和手机立即绑定
首页 手记 DBunit、Spring TestContext实践

DBunit、Spring TestContext实践

标签：
MySQL

 

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>

【注意】

1、保证加载的依赖的正确性，使用了外面的test-context.jar，则可能出现找不到beans配置文件的问题，而不会报找不到依赖的问题。

 

2、基于Spring的单元测试框架搭建及分布式数据源单元测试

http://aaronjiu.iteye.com/blog/435574

 

 

 

 

©著作权归作者所有：来自51CTO博客作者tianya23的原创作品，如需转载，请注明出处，否则将追究法律责任

职场休闲dbunitSpring


点击查看更多内容
发表于 2019.01.30 18:42， 共 1095 人浏览

本文原创发布于慕课网 ，转载请注明出处，谢谢合作

举报
为 TA 点赞

若觉得本文不错，就分享一下吧！

评论

作者其他优质文章

正在加载中

largeQ

手记
篇
粉丝
92
获赞与收藏
585

关注作者，订阅最新文章

相关文章推荐

阅读免费教程

  • 推荐
  • 评论
  • 收藏
  • 共同学习，写下你的评论
感谢您的支持，我会继续努力的～
扫码打赏，你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫，即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消