OneToOneTest.java

/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * 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.ogm.backendtck.associations.onetoone;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;

import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.ogm.utils.OgmTestCase;
import org.junit.Test;

/**
 * @author Emmanuel Bernard
 */
public class OneToOneTest extends OgmTestCase {

	@Test
	public void testUnidirectionalManyToOne() throws Exception {
		final Session session = openSession();
		Transaction transaction = session.beginTransaction();
		Horse horse = new Horse( "palefrenier" );
		horse.setName( "Palefrenier" );
		Cavalier cavalier = new Cavalier( "caroline" );
		cavalier.setName( "Caroline" );
		cavalier.setHorse( horse );
		session.persist( horse );
		session.persist( cavalier );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		cavalier = (Cavalier) session.get( Cavalier.class, cavalier.getId() );
		horse = cavalier.getHorse();
		session.delete( cavalier );
		session.delete( horse );
		transaction.commit();

		session.close();

		checkCleanCache();
	}

	@Test
	public void testUnidirectionalOneToOne() throws Exception {
		final Session session = openSession();
		Transaction transaction = session.beginTransaction();
		Vehicule vehicule = new Vehicule();
		vehicule.setBrand( "Mercedes" );
		Wheel wheel = new Wheel();
		wheel.setVehicule( vehicule );
		session.persist( vehicule );
		session.persist( wheel );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		wheel = (Wheel) session.get( Wheel.class, wheel.getId() );
		vehicule = wheel.getVehicule();
		session.delete( wheel );
		session.delete( vehicule );
		transaction.commit();
		session.close();
	}

	@Test
	public void testBidirectionalOneToOne() throws Exception {
		final Session session = openSession();
		Transaction transaction = session.beginTransaction();
		Husband husband = new Husband( "alex" );
		husband.setName( "Alex" );
		Wife wife = new Wife( "bea" );
		wife.setName( "Bea" );
		husband.setWife( wife );
		wife.setHusband( husband );
		session.persist( husband );
		session.persist( wife );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		husband = (Husband) session.get( Husband.class, husband.getId() );
		assertNotNull( husband );
		assertNotNull( husband.getWife() );
		session.clear();
		wife = (Wife) session.get( Wife.class, wife.getId() );
		assertNotNull( wife );
		husband = wife.getHusband();
		assertNotNull( husband );
		Wife bea2 = new Wife( "still_bea" );
		session.persist( bea2 );
		bea2.setName( "Still Bea" );
		husband.setWife( bea2 );
		wife.setHusband( null );
		bea2.setHusband( husband );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		husband = (Husband) session.get( Husband.class, husband.getId() );
		assertNotNull( husband );
		assertNotNull( husband.getWife() );
		session.clear();
		wife = (Wife) session.get( Wife.class, wife.getId() );
		assertNotNull( wife );
		assertNull( wife.getHusband() );
		session.delete( wife );
		bea2 = (Wife) session.get( Wife.class, bea2.getId() );
		assertNotNull( bea2 );
		husband = bea2.getHusband();
		assertNotNull( husband );
		bea2.setHusband( null );
		husband.setWife( null );
		session.delete( husband );
		session.delete( wife );
		session.delete( bea2 );
		transaction.commit();
		session.close();
	}

	@Test
	public void testBidirectionalOneToOneCompositeId() throws Exception {
		final Session session = openSession();
		Transaction transaction = session.beginTransaction();
		PatchCable patchCable = new PatchCable( new MediaId( "Belkin", "cat7" ) );
		NetworkSwitch networkSwitch = new NetworkSwitch( new MediaId( "Frisco", "AS500" ) );
		patchCable.setNetworkSwitch( networkSwitch );
		networkSwitch.setPatchCable( patchCable );
		session.persist( patchCable );
		session.persist( networkSwitch );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		patchCable = session.get( PatchCable.class, patchCable.getId() );
		assertNotNull( patchCable );
		assertNotNull( patchCable.getNetworkSwitch() );
		session.clear();
		networkSwitch = session.get( NetworkSwitch.class, networkSwitch.getId() );
		assertNotNull( networkSwitch );
		patchCable = networkSwitch.getPatchCable();
		assertNotNull( patchCable );
		assertNotNull( patchCable.getId() );
		transaction.commit();
		session.clear();

		transaction = session.beginTransaction();
		patchCable = session.get( PatchCable.class, patchCable.getId() );
		assertNotNull( patchCable );
		assertNotNull( patchCable.getNetworkSwitch() );
		session.clear();
		networkSwitch = session.get( NetworkSwitch.class, networkSwitch.getId() );
		assertNotNull( networkSwitch );
		session.delete( patchCable );
		session.delete( networkSwitch );
		transaction.commit();
		session.close();
	}

	@Override
	protected Class<?>[] getAnnotatedClasses() {
		return new Class<?>[] {
				Horse.class,
				Cavalier.class,
				Vehicule.class,
				Wheel.class,
				Husband.class,
				Wife.class,
				NetworkSwitch.class,
				PatchCable.class
		};
	}
}