OneToOneBidirectionalTest.java

/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.userguide.associations;

import javax.persistence.CascadeType;
import javax.persistence.Column;
import javax.persistence.Entity;
import javax.persistence.FetchType;
import javax.persistence.GeneratedValue;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import javax.persistence.OneToOne;

import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;

import org.junit.Assert;
import org.junit.Test;

import org.jboss.logging.Logger;

import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;

/**
 * @author Vlad Mihalcea
 */
public class OneToOneBidirectionalTest extends BaseEntityManagerFunctionalTestCase {

	private static final Logger log = Logger.getLogger( OneToOneBidirectionalTest.class );

	@Override
	protected Class<?>[] getAnnotatedClasses() {
		return new Class<?>[] {
				Phone.class,
				PhoneDetails.class,
		};
	}

	@Test
	public void testLifecycle() {
		doInJPA( this::entityManagerFactory, entityManager -> {
			Phone phone = new Phone( "123-456-7890" );
			PhoneDetails details = new PhoneDetails( "T-Mobile", "GSM" );

			phone.addDetails( details );
			entityManager.persist( phone );
		} );
	}

	@Test
	public void testConstraint() {
		doInJPA( this::entityManagerFactory, entityManager -> {
			//tag::associations-one-to-one-bidirectional-lifecycle-example[]
			Phone phone = new Phone( "123-456-7890" );
			PhoneDetails details = new PhoneDetails( "T-Mobile", "GSM" );

			phone.addDetails( details );
			entityManager.persist( phone );
			//end::associations-one-to-one-bidirectional-lifecycle-example[]
		} );
		try {
			doInJPA( this::entityManagerFactory, entityManager -> {

				Phone phone = entityManager.find( Phone.class, 1L );

				//tag::associations-one-to-one-bidirectional-constraint-example[]
				PhoneDetails otherDetails = new PhoneDetails( "T-Mobile", "CDMA" );
				otherDetails.setPhone( phone );
				entityManager.persist( otherDetails );
				entityManager.flush();
				entityManager.clear();

				//throws javax.persistence.PersistenceException: org.hibernate.HibernateException: More than one row with the given identifier was found: 1
				phone = entityManager.find( Phone.class, phone.getId() );
				//end::associations-one-to-one-bidirectional-constraint-example[]
				phone.getDetails().getProvider();
			} );
			Assert.fail( "Expected: HHH000327: Error performing load command : org.hibernate.HibernateException: More than one row with the given identifier was found: 1" );
		}
		catch (Exception expected) {
			log.error( "Expected", expected );
		}
	}

	//tag::associations-one-to-one-bidirectional-example[]
	@Entity(name = "Phone")
	public static class Phone {

		@Id
		@GeneratedValue
		private Long id;

		@Column(name = "`number`")
		private String number;

		@OneToOne(mappedBy = "phone", cascade = CascadeType.ALL, orphanRemoval = true, fetch = FetchType.LAZY)
		private PhoneDetails details;

		public Phone() {
		}

		public Phone(String number) {
			this.number = number;
		}

		public Long getId() {
			return id;
		}

		public String getNumber() {
			return number;
		}

		public PhoneDetails getDetails() {
			return details;
		}

		public void addDetails(PhoneDetails details) {
			details.setPhone( this );
			this.details = details;
		}

		public void removeDetails() {
			if ( details != null ) {
				details.setPhone( null );
				this.details = null;
			}
		}
	}

	@Entity(name = "PhoneDetails")
	public static class PhoneDetails {

		@Id
		@GeneratedValue
		private Long id;

		private String provider;

		private String technology;

		@OneToOne(fetch = FetchType.LAZY)
		@JoinColumn(name = "phone_id")
		private Phone phone;

		public PhoneDetails() {
		}

		public PhoneDetails(String provider, String technology) {
			this.provider = provider;
			this.technology = technology;
		}

		public String getProvider() {
			return provider;
		}

		public String getTechnology() {
			return technology;
		}

		public void setTechnology(String technology) {
			this.technology = technology;
		}

		public Phone getPhone() {
			return phone;
		}

		public void setPhone(Phone phone) {
			this.phone = phone;
		}
	}
	//end::associations-one-to-one-bidirectional-example[]
}