/jboss-as-7.1.1.Final/testsuite/integration/basic/src/test/java/org/jboss/as/test/integration/jpa/hibernate/EntityTest.java

/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2011, Red Hat, Inc., and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.jboss.as.test.integration.jpa.hibernate;

import java.util.HashSet;
import java.util.Set;

import javax.ejb.Stateful;
import javax.persistence.PersistenceContext;
import javax.persistence.PersistenceContextType;

import org.hibernate.Session;
import org.jboss.as.test.integration.jpa.hibernate.entity.Company;
import org.jboss.as.test.integration.jpa.hibernate.entity.Customer;
import org.jboss.as.test.integration.jpa.hibernate.entity.Flight;
import org.jboss.as.test.integration.jpa.hibernate.entity.Ticket;

/**
 * Stateful session bean for testing Hibernate entities:
 * One to Many, Many to One, Many to Many mapping and named queries.
 *
 * @author ZbynÄ›k Roubal?­k
 */
@Stateful
public class EntityTest {

    @PersistenceContext(unitName = "entity_pc", type = PersistenceContextType.EXTENDED)
    private Session session;

    public Customer oneToManyCreate() throws Exception {
        Ticket t = new Ticket();
        t.setNumber("111");

        Customer c = new Customer();
        Set<Ticket> tickets = new HashSet<Ticket>();

        tickets.add(t);
        t.setCustomer(c);
        c.setTickets(tickets);

        session.save(c);

        return c;
    }

    public Flight manyToOneCreate() throws Exception {
        Flight f = new Flight();
        f.setName("Flight number one");
        f.setId(new Long(1));

        Company comp = new Company();
        comp.setName("Airline 1");
        f.setCompany(comp);

        session.save(f);

        return f;
    }

    public void manyToManyCreate() throws Exception {

        Flight f1 = findFlightById(new Long(1));
        Flight f2 = new Flight();

        f2.setId(new Long(2));
        f2.setName("Flight two");

        Company us = new Company();
        us.setName("Airline 2");
        f2.setCompany(us);

        Set<Customer> customers1 = new HashSet<Customer>();
        Set<Customer> customers2 = new HashSet<Customer>();

        Customer c1 = new Customer();
        c1.setName("John");
        customers1.add(c1);

        Customer c2 = new Customer();
        c2.setName("Tom");
        customers2.add(c2);

        Customer c3 = new Customer();
        c3.setName("Pete");
        customers2.add(c3);

        f1.setCustomers(customers1);
        f2.setCustomers(customers2);

        session.save(c1);
        session.save(c2);
        session.save(c3);
        session.save(f2);
    }

    public int testAllCustomersQuery() {

        //session.flush();
        return session.getNamedQuery("allCustomers").list().size();
    }

    public Customer testCustomerByIdQuery() {
        Customer c = new Customer();
        c.setName("Peter");

        session.save(c);
        session.flush();

        return (Customer) session.getNamedQuery("customerById").setParameter("id", c.getId()).uniqueResult();

    }

    public Customer createCustomer(String name) {
        Customer c = new Customer();
        c.setName(name);
        session.save(c);
        return c;
    }

    public void changeCustomer(Long id, String name) {
        Customer c = (Customer) session.load(Customer.class, id);
        c.setName(name);
    }


    public Flight findFlightById(Long id) {

        return (Flight) session.load(Flight.class, id);
    }

    public Company findCompanyById(Long id) {

        return (Company) session.load(Company.class, id);
    }

    public Customer findCustomerById(Long id) {

        return (Customer) session.load(Customer.class, id);
    }

}