如何解决DbUnit的Maven插件在执行过程中出现的侵犯外键约束错误(MySQLIntegrityConstraintViolationException)

如果是在使用UbUnit进行单元测试时遇到


其实这个问题不止在通过DbUnit的Maven插件操作数据库时会遇到,在基于DbUni进行单元测试时也会遇到,一般的处理方法是在每次获取连接时前执行一个"set  @@session.foreign_key_checks = 0"的statement,以保证本次会话不作外键约束检查,以下是我常用的一个基于DbUnit的单元测试基类,其中第59行就是关于禁止外键约束检查的设置。(注:本例使用的数据库是MySql)

package oobbs.domainmodel;

import java.sql.SQLException;

import javax.sql.DataSource;

import org.dbunit.DataSourceDatabaseTester;
import org.dbunit.DefaultOperationListener;
import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.xml.XmlDataSet;
import org.dbunit.ext.mysql.MySqlDataTypeFactory;
import org.junit.After;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

import testutil.ApplicationContextSupport;

/**
 * This base class does not extends any DBTestCase of dbunit,or use any Tester,Their's implement is not good and flexible.
 * Here,we prepare and set connection manully!
 * 
 * @author Laurence Geng
 */
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"classpath:/applicationContext-infrastructure.xml",
								   "classpath:/applicationContext-domainModel.xml",
								   "classpath:/applicationContext-test.xml"})
public abstract class DbBasedTest{
	
    /** The data source. */
    @Autowired
    protected DataSource dataSource;

    /** The dbunitTestUtil can fill test data from xml into test db before testing. */
    protected DataSourceDatabaseTester dbunitTestUtil;
    
    /**
     * Inits dbunitTestUtil. 
	 * The connectionRetrieved method is called back when setUp() executes.
	 * At this time,we should set connection-specific setting: set foreign key check disabled
	 * so as dbunit can invert test data, and set data type factory be MySqlDataTypeFactory so as
	 * dbunit can convert correct type when invert data to mysql.
	 *
     * @throws Exception the exception
     */
    protected void initDbunitTestUtil() throws Exception{
    	dbunitTestUtil = new DataSourceDatabaseTester(dataSource);
		dbunitTestUtil.setDataSet(new XmlDataSet(new ClassPathResource("dbunit-test-data.xml").getInputStream()));
		dbunitTestUtil.setOperationListener( new DefaultOperationListener(){
            public void connectionRetrieved(IDatabaseConnection connection) {
                try {
                	//Disable foreign key check!
					connection.getConnection().prepareStatement("set @@session.foreign_key_checks = 0").execute();
					// When a new connection has been created then invoke the setUp method
					// so that user defined DatabaseConfig parameters can be set.
					connection.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new MySqlDataTypeFactory());
				} catch (SQLException e) {
					e.printStackTrace();
				}
            }});
    }
	
	/**
	 * Before test method.
	 * 
	 * @throws Exception the exception
	 */
	@Before
    public void beforeTestMethod() throws Exception { 
		initDbunitTestUtil();
		dbunitTestUtil.onSetup();
    }
    
    /**
     * After test method.
     * 
     * @throws Exception the exception
     */
    @After
    public void afterTestMethod() throws Exception {
    	dbunitTestUtil.onTearDown();
    }
    
}


如果是在使用DbUnit的Maven插件时遇到


而如果是使用Maven的DbUnit插件,以命令行的方式执行数据导入工作的话,就只能从数据库连接的url上下手了,方法也很简单就是在原插件的配置上添加foreign_key_checks = 0这个变量,以下是一个例子,请注意第14行标记的部分:${jdbc.url}&sessionVariables=foreign_key_checks=0,它在标准url后面追加了对变量foreign_key_checks的设置。

			
				org.codehaus.mojo
				dbunit-maven-plugin
				1.0-beta-3
				
					
						${jdbc.groupId}
						${jdbc.artifactId}
						${jdbc.version}
					
				
				
					${jdbc.driverClassName}
					${jdbc.url}&sessionVariables=foreign_key_checks=0
					${jdbc.username}
					${jdbc.password}
				
				
					
						default-cli
						
							operation
						
						
							CLEAN_INSERT
							src/test/resources/dbunit-test-data.xml
							org.dbunit.ext.mysql.MySqlDataTypeFactory
							true
						
					
				
			


备注

mysql中,变量的作用域有两种session和global,改变变量值的方法为:


要想设置一个GLOBAL变量的值,使用下面的语法:


mysql> SET GLOBAL sort_buffer_size=value;
mysql> SET @@global.sort_buffer_size=value;


要想设置一个SESSION变量的值,使用下面的语法:


mysql> SET SESSION sort_buffer_size=value;
mysql> SET @@session.sort_buffer_size=value;
mysql> SET sort_buffer_size=value;

你可能感兴趣的:(如何解决DbUnit的Maven插件在执行过程中出现的侵犯外键约束错误(MySQLIntegrityConstraintViolationException))