SchedulePlan1Test.java 2.55 KB
package com.bsth.service.schedule.rules;

import com.bsth.service.schedule.BaseTest;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.dbunit.operation.DatabaseOperation;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;

import javax.sql.DataSource;

/**
 * 排班计划测试(手工排班测试)。
 */
public class SchedulePlan1Test extends BaseTest {
    /** 日志记录器 */
    private static final Logger logger = LoggerFactory.getLogger(SchedulePlan1Test.class);

    @Autowired
    private DataSource dataSource;

    /**
     * 初始化数据。
     */
    @Before
    public void initData() throws Exception {
        logger.info("载入数据......");

        // 获取数据库连接
        IDatabaseConnection iDatabaseConnection = new DatabaseConnection(dataSource.getConnection());
        // 获取Dbunit数据源
        Resource res = new ClassPathResource("testdata/d2.xml");
        FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder();
        flatXmlDataSetBuilder.setColumnSensing(false);
        flatXmlDataSetBuilder.setCaseSensitiveTableNames(false);
        IDataSet iDataSet = flatXmlDataSetBuilder.build(res.getInputStream());

        // 载入数据
        DatabaseOperation.CLEAN_INSERT.execute(iDatabaseConnection, iDataSet);
    }

    /**
     * 清除数据。
     */
    @After
    public void destoryData() throws Exception {
        logger.info("清除数据......");
        // 获取数据库连接
        IDatabaseConnection iDatabaseConnection = new DatabaseConnection(dataSource.getConnection());
        // 获取Dbunit数据源
        Resource res = new ClassPathResource("testdata/d2.xml");
        FlatXmlDataSetBuilder flatXmlDataSetBuilder = new FlatXmlDataSetBuilder();
        flatXmlDataSetBuilder.setColumnSensing(false);
        flatXmlDataSetBuilder.setCaseSensitiveTableNames(false);
        IDataSet iDataSet = flatXmlDataSetBuilder.build(res.getInputStream());

        // 清除数据
        DatabaseOperation.DELETE_ALL.execute(iDatabaseConnection, iDataSet);
    }


    @Test
    public void t1() {

    }




















}