com.jeroensteenbeeke.hyperion.data.TestPersistenceModel.java Source code

Introduction

Here is the source code for com.jeroensteenbeeke.hyperion.data.TestPersistenceModel.java

Source

/**
 * Tysan Clan Website
 * Copyright (C) 2008-2013 Jeroen Steenbeeke and Ties van de Ven
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.jeroensteenbeeke.hyperion.data;

import static org.junit.Assert.assertEquals;

import java.io.Serializable;

import org.apache.wicket.model.IModel;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.boot.registry.StandardServiceRegistryBuilder;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class TestPersistenceModel {
    private SessionFactory sf;

    private Session current;

    @Before
    public void initHibernate() {
        Configuration config = new Configuration();
        config = config.addAnnotatedClass(Lolcat.class);

        config = config.setProperty("hibernate.hbm2ddl.auto", "create");
        config = config.setProperty("hibernate.dialect", "org.hibernate.dialect.H2Dialect");
        config = config.setProperty("hibernate.show_sql", "true");
        config = config.setProperty("hibernate.connection.driver_class", "org.h2.Driver");
        config = config.setProperty("hibernate.connection.url", "jdbc:h2:mem:tysan");
        config = config.setProperty("hibernate.connection.username", "sa");
        config = config.setProperty("hibernate.connection.password", "");
        config = config.setProperty("hibernate.connection.pool_size", "50");
        config = config.setProperty("hibernate.connection.provider_class",
                "org.hibernate.connection.DriverManagerConnectionProvider");
        // config = config.setProperty("hibernate.c3p0.min_size", "5");
        // config = config.setProperty("hibernate.c3p0.max_size", "20");
        // config = config.setProperty("hibernate.c3p0.timeout", "1800");
        // config = config.setProperty("hibernate.c3p0.max_statements", "50");

        final ServiceRegistry serviceRegistry = new StandardServiceRegistryBuilder()
                .applySettings(config.getProperties()).build();

        sf = config.buildSessionFactory(serviceRegistry);
    }

    @Test
    public void testPersistenceModel() {
        current = sf.openSession();

        Lolcat t1 = new Lolcat();
        Lolcat t2 = new Lolcat();

        t2.setParent(t1);

        current.save(t1);
        current.save(t2);

        IModel<Lolcat> model = new PersistenceModel<Lolcat>(t1) {
            private static final long serialVersionUID = 1L;

            @Override
            protected void injectDAO() {
                setDao(new BaseEntityFinder() {

                    @SuppressWarnings("unchecked")
                    @Override
                    public <T extends DomainObject> T getEntity(Class<T> entityClass, Serializable id) {
                        return (T) current.load(entityClass, id);
                    }
                });
            }

        };

        current.close();

        model.detach();

        current = sf.openSession();

        assertEquals(1, model.getObject().getChildren().size());

        current.close();
    }

    @After
    public void tearItAllDown() {
        sf.close();
    }
}