net.jforum.core.hibernate.AbstractDAOTestCase.java Source code

Introduction

Here is the source code for net.jforum.core.hibernate.AbstractDAOTestCase.java

Source

/*
 * Copyright (c) JForum Team. All rights reserved.
 *
 * The software in this package is published under the terms of the LGPL
 * license a copy of which has been included with this distribution in the
 * license.txt file.
 *
 * The JForum Project
 * http://www.jforum.net
 */
package net.jforum.core.hibernate;

import net.jforum.repository.HibernateGenericDAO;

import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.hibernate.cfg.AnnotationConfiguration;
import org.hibernate.cfg.Configuration;
import org.junit.After;
import org.junit.Before;

/**
 * @author Rafael Steil
 */
public abstract class AbstractDAOTestCase<T> {
    private Session session;
    private SessionFactory sessionFactory;

    /**
     * Persists an entity.
     * After inserting, the transaction will be commited and
     * the session cleared.
     * @param entity the entity to persist
     * @param dao the dao to use
     */
    protected void insert(T entity, HibernateGenericDAO<T> dao) {
        dao.add(entity);
        commit();
        beginTransaction();
    }

    /**
     * Updates an entity.
     * After updating, the transaction will be commited and
     * the session cleared.
     * @param entity the entity to update
     * @param dao the dao to use
     */
    protected void update(T entity, HibernateGenericDAO<T> dao) {
        dao.update(entity);
        commit();
        beginTransaction();
    }

    /**
     * Deletes an entity
     * @param entity the entity to delete
     * @param dao the dao to use
     */
    protected void delete(T entity, HibernateGenericDAO<T> dao) {
        dao.remove(entity);
        commit();
        beginTransaction();
    }

    protected Session session() {
        return session;
    }

    /**
     * First commits the transaction
     */
    protected void commit() {
        session.getTransaction().commit();
    }

    @Before
    public void setUp() throws Exception {
        Configuration config = new AnnotationConfiguration();
        config.configure("/hibernate-tests.cfg.xml");
        sessionFactory = config.buildSessionFactory();
        session = sessionFactory.openSession();
        beginTransaction();
    }

    protected Transaction beginTransaction() {
        return session.beginTransaction();
    }

    @After
    public void tearDown() throws Exception {
        if (session != null) {
            session.close();
        }
        if (sessionFactory != null) {
            sessionFactory.close();
        }
    }
}