使用AbstractTransactionalJUnit4SpringContextTests测试纯JDBC DAO方法:测试方法后未更改的DB中的更改

时间:2013-10-26 19:54:28

标签: java spring testing jdbc transactions

我正在玩纯JDBC和测试。我编写了简单的DAO,它允许执行CRUD操作:

package pl.aszecowka;


import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;


public class JdbcVehicleDao implements VehicleDao {
    private DataSource dataSource;

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    @Override
    public void insert(Vehicle vehicle) {
        String sql = "insert into vehicle(vehicle_no,color, wheel, seat) values(?,?,?,?)";
        Connection conn = null;
        try {
            conn = dataSource.getConnection();
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, vehicle.getVehicleNo());
            ps.setString(2, vehicle.getColor());
            ps.setInt(3, vehicle.getWheel());
            ps.setInt(4, vehicle.getSeat());
            ps.executeUpdate();
            ps.close();
            conn.commit();
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException e) {
                }
            }
        }
    }

    @Override
    public void update(Vehicle vehicle) {
        String sql = "update vehicle set color=?, wheel=?, seat=? where vehicle_no = ?";
        Connection conn = null;
        try {
            conn = dataSource.getConnection();
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, vehicle.getColor());
            ps.setString(4, vehicle.getVehicleNo());
            ps.setInt(2, vehicle.getWheel());
            ps.setInt(3, vehicle.getSeat());
            ps.executeUpdate();
            ps.close();
            conn.commit();
        } catch (SQLException ex) {
            throw new RuntimeException(ex);
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException ex) {
                }
            }
        }
    }

    @Override
    public void delete(Vehicle vehicle) {
        String sql = "delete from vehicle where vehicle_no = ? ";
        Connection conn = null;
        try {
            conn = dataSource.getConnection();
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, vehicle.getVehicleNo());
            int i = ps.executeUpdate();
            System.out.println(i);
            ps.close();
            conn.commit();
        } catch (SQLException ex) {
            throw new RuntimeException(ex);
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException ex) {
                }
            }
        }
    }

    @Override
    public Vehicle findByVehicleNo(String vehicleNo) {
        String sql = "select * from vehicle where vehicle_no = ?";
        Connection conn = null;
        try {
            conn = dataSource.getConnection();
            PreparedStatement ps = conn.prepareStatement(sql);
            ps.setString(1, vehicleNo);
            Vehicle vehicle = null;
            ResultSet rs = ps.executeQuery();
            if (rs.next()) {
                vehicle = new Vehicle(rs.getString("vehicle_no"), rs.getString("color"), rs.getInt("wheel"), rs.getInt("seat"));
            }
            rs.close();
            ps.close();
            return vehicle;
        } catch (SQLException ex) {
            throw new RuntimeException(ex);
        } finally {
            if (conn != null) {
                try {
                    conn.close();
                } catch (SQLException ex) {
                }
            }
        }

    }
}

这是我的弹簧配置:

<bean id="dataSource" class="org.apache.commons.dbcp.BasicDataSource">
            <property name="driverClassName" value="org.apache.derby.jdbc.ClientDriver"/>
            <property name="url" value="jdbc:derby://localhost:1527/vehicle;create=true"/>
            <property name="username" value="app"/>
            <property name="password" value="app"/>
            <property name="initialSize" value="2"/>
            <property name="maxActive" value="5"/>
           <property name="defaultAutoCommit" value="false" />
        </bean>

        <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
            <property name="dataSource" ref="dataSource"/>
        </bean>

        <bean id="vehicleDao" class="pl.aszecowka.JdbcVehicleDao">
            <property name="dataSource" ref="dataSource"/>
        </bean>

我的测试:

@ContextConfiguration("/beans.xml")
public class JdbcVehicleDaoTest extends AbstractTransactionalJUnit4SpringContextTests {

    @Resource
    private VehicleDao vehicleDao;


    @Test
    public void testCRUD()
    {
        String vehicleNo = "ABCDEF";
        String color = "blue";
        int wheel = 4;
        int seat = 4;
        Vehicle vehicle = new Vehicle(vehicleNo, color, wheel, seat);
        vehicleDao.insert(vehicle);
        Vehicle fromDB = vehicleDao.findByVehicleNo(vehicleNo);
        Assert.assertNotNull(fromDB);
        Assert.assertEquals(vehicleNo, fromDB.getVehicleNo());
        Assert.assertEquals(color, fromDB.getColor());
        Assert.assertEquals(wheel, fromDB.getWheel());
        Assert.assertEquals(seat, fromDB.getSeat());

        color = "blue";
        seat = 5;
        wheel = 12;

        fromDB.setColor(color);
        fromDB.setSeat(seat);
        fromDB.setWheel(wheel);
        vehicleDao.update(fromDB);
        fromDB = vehicleDao.findByVehicleNo(fromDB.getVehicleNo());
        Assert.assertNotNull(fromDB);
        Assert.assertEquals(vehicleNo, fromDB.getVehicleNo());
        Assert.assertEquals(color, fromDB.getColor());
        Assert.assertEquals(wheel, fromDB.getWheel());
        Assert.assertEquals(seat, fromDB.getSeat());

        vehicleDao.delete(fromDB);
        fromDB = vehicleDao.findByVehicleNo(fromDB.getVehicleNo());
        Assert.assertNull(fromDB);
    }

    @Test
    public void testCheckIfTestRollbackWorks()
    {
        Vehicle vehicle = new Vehicle("ABCDEF", "blue", 4, 4);
        vehicleDao.insert(vehicle);
    }
}

在我的测试类中,我想避免任何自定义的“清理”方法,这些方法还原了测试期间所做的所有更改,因此我扩展了 AbstractTransactionalJUnit4SpringContextTests 。此类使用 @Transactional 进行注释,据我所知,这意味着所有测试方法的事务将在最后回滚。
在开始时,回滚不起作用,例如,如果testCheckIfTestRollbackWorks()作为第一个运行,第二个是testCRUD,则testCRUD因抛出 SQLIntegrityConstraintViolationException 而失败:语句被中止,因为它会导致重复由'VEHICLE'上定义的'SQL131026082556940'标识的唯一或主键约束或唯一索引中的键值。
经过一些研究,我发现 autoCommit 可能导致此类行为的信息,然后我将defaultAutoCommit属性设置为false。
但是,testCRUD不起作用,因为尽管我打电话 vehicleDao.insert(车辆); 我使用时找不到这个物体:vehicleDao.findByVehicleNo(vehicleNo);
我认为原因很简单:我的DAO没有执行任何提交。所以我添加了插入,更新和删除方法“ conn.commit()”。
但是现在,在另一个测试中再次看到一个测试的变化:(我假设测试回滚它自己的tranasaction,但不是这个在DAO方法中提交的。) 如何解决这个问题的任何提示?

2 个答案:

答案 0 :(得分:1)

  

这个类用@Transactional注释,据我所知   表示所有测试方法的事务将回滚到   端。

是的但是仅适用于使用单个和/或弹簧管理事务的代码。您的代码没有,您自己获得连接并在该连接上进行提交。因此数据已经提交,提交的数据无法回滚。接下来,春天(或在弹簧控制下)交易是不可见的,因此春天无法做到这一点。

如果您希望Spring影响/推动您的交易,请重写您的代码以使用JdbcTemplate或使用TransactionAwareDataSourceProxy。有关详细信息,请参阅Spring Reference Guide

链接:

  1. JdbcTemplate javadoc | reference
  2. TransactionAwareDataSourceProxy javadoc | reference
  3. 参考指南的
  4. JDBC Section

答案 1 :(得分:0)

  

@Transactional,据我所知,这意味着交易   所有测试方法都将在最后回滚。

事实并非如此,如果没有发生配置注释的异常,它将提交事务。

@Transactional具有您应该配置的rollbackFor属性。