com.evidence.data.DbUnitTest.java Source code

Introduction

Here is the source code for com.evidence.data.DbUnitTest.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package com.evidence.data;

import java.io.File;
import java.net.MalformedURLException;
import java.sql.Connection;
import java.sql.SQLException;

import javax.inject.Inject;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

import org.dbunit.DatabaseUnitException;
import org.dbunit.database.DatabaseConfig;
import org.dbunit.database.DatabaseConnection;
import org.dbunit.database.IDatabaseConnection;
import org.dbunit.dataset.DataSetException;
import org.dbunit.dataset.IDataSet;
import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
import org.dbunit.ext.hsqldb.HsqldbDataTypeFactory;
import org.dbunit.operation.DatabaseOperation;
import org.hibernate.internal.SessionImpl;
import org.junit.After;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.springframework.security.authentication.AuthenticationManager;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;

/**
 * @author Michal Bocek
 * @since 1.0.0
 */
@Transactional
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({ "classpath:spring/transaction.xml", "classpath:spring/applicationConfig.xml",
        "classpath:spring/security.xml" })
public class DbUnitTest {

    @PersistenceContext(name = "entityManagerFactory")
    private EntityManager entityManager;

    @Inject
    private AuthenticationManager authenticationManager;

    @Before
    public void init() throws DatabaseUnitException, SQLException, MalformedURLException {
        Authentication authentication = new UsernamePasswordAuthenticationToken("admin@evidence.com", "password");
        Authentication authenticate = authenticationManager.authenticate(authentication);
        SecurityContextHolder.getContext().setAuthentication(authenticate);
        // insert data into database
        DatabaseOperation.CLEAN_INSERT.execute(getConnection(), getDataSet());
    }

    @After
    public void after() throws DatabaseUnitException, SQLException, MalformedURLException {
        // delete data from database
        final SessionImpl session = (SessionImpl) entityManager.getDelegate();
        final Connection con = session.connection(); // NOPMD
        con.createStatement().execute("SET DATABASE REFERENTIAL INTEGRITY FALSE;");
        DatabaseOperation.DELETE_ALL.execute(getConnection(), getDataSet());
        con.createStatement().execute("SET DATABASE REFERENTIAL INTEGRITY TRUE;");
        SecurityContextHolder.getContext().setAuthentication(null);
    }

    private IDatabaseConnection getConnection() throws DatabaseUnitException {
        // get connection
        final SessionImpl session = (SessionImpl) entityManager.getDelegate();
        final Connection con = session.connection(); // NOPMD
        //DatabaseMetaData databaseMetaData = con.getMetaData();
        final IDatabaseConnection connection = new DatabaseConnection(con);
        final DatabaseConfig config = connection.getConfig();
        config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new HsqldbDataTypeFactory());
        return connection;
    }

    private IDataSet getDataSet() throws MalformedURLException, DataSetException {
        final FlatXmlDataSetBuilder builder = new FlatXmlDataSetBuilder();
        builder.setColumnSensing(true);
        return builder.build(new File("src/test/resources/dataset.xml"));
    }

    protected EntityManager getEntityManager() {
        return this.entityManager;
    }
}