//$Id: Company.java,v 1.1 2005/05/12 13:33:52 epbernard Exp $
package org.hibernate.test.annotations;

import javax.persistence.*;
import java.io.Serializable;
import java.util.List;


/**
* Corporate like Air France
*
* @author Emmanuel Bernard
*/
@Entity(name="Corporation")
@EntityListener(Audit.class)
public class Company implements Serializable {
   private Integer id;
   private String name;
    private Flight flight;
    protected boolean updated = false;

    @Id(generate = GeneratorType.AUTO)
    public Integer getId() {
        return id;
    }

   public String getName() {
      return name;
   }

    @OneToOne( cascade = {CascadeType.REMOVE}, fetch = FetchType.LAZY)
    @JoinColumn(name = "comp_id")
    public Flight getFlight() {
        return flight;
    }

    public void setFlight(Flight flights) {
        this.flight = flights;
    }

   public void setId(Integer integer) {
      id = integer;
   }


   public void setName(String string) {
      name = string;
   }

    @PrePersist
    @PreUpdate
    public void testCallbacks(){
        updated = true;
    }

    @Transient
    public boolean isUpdated() {
        return updated;
    }

}

   comp = new Company();
        comp.setName(java.util.UUID.randomUUID().toString());
        Flight flight = new Flight();
        flight.setName(java.util.UUID.randomUUID().toString());
        flight.setDepartureDates(new LinkedList<DepartureDate>());
        for (int j = 0; j < 3; j++) {
            DepartureDate date = new DepartureDate();
            date.setDate(new Date());
            date.setDescription(java.util.UUID.randomUUID().toString());
            flight.getDepartureDates().add(date);
        }
        comp.setFlight(flight);


        long tickStart = System.currentTimeMillis();
        tx.begin();
        try {
            flight = em.merge(comp.getFlight());
            comp.setFlight(flight);
            comp = em.merge(comp);
        }
        catch (Exception e) {
            tx.rollback();
            throw e;
        }

        if (!comp.isUpdated()){
            throw new RuntimeException("Callbacks was not called!!!!");
        };
        tx.commit();
        em.close();