HikeTest.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.demos.ogm101.part1;

import static org.fest.assertions.Assertions.assertThat;

import java.math.BigDecimal;
import java.util.Date;

import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

/**
 * Shows how to persist and retrieve entities to/from NoSQL stores using Hibernate OGM.
 *
 * @author Gunnar Morling
 */
public class HikeTest {

	private static EntityManagerFactory entityManagerFactory;

	@BeforeClass
	public static void setUpEntityManagerFactory() {
		entityManagerFactory = Persistence.createEntityManagerFactory( "hikePu" );
	}

	@AfterClass
	public static void closeEntityManagerFactory() {
		entityManagerFactory.close();
	}

	@Test
	public void canPersistAndLoadPersonAndHikes() {
		EntityManager entityManager = entityManagerFactory.createEntityManager();

		entityManager.getTransaction().begin();

		// create a Person
		Person bob = new Person( "Bob", "McRobb" );

		// and two hikes
		Hike cornwall = new Hike(
				"Visiting Land's End", new Date(), new BigDecimal( "5.5" ),
				new HikeSection( "Penzance", "Mousehole" ),
				new HikeSection( "Mousehole", "St. Levan" ),
				new HikeSection( "St. Levan", "Land's End" )
		);
		Hike isleOfWight = new Hike(
				"Exploring Carisbrooke Castle", new Date(), new BigDecimal( "7.5" ),
				new HikeSection( "Freshwater", "Calbourne" ),
				new HikeSection( "Calbourne", "Carisbrooke Castle" )
		);

		// let Bob organize the two hikes
		cornwall.setOrganizer( bob );
		bob.getOrganizedHikes().add( cornwall );

		isleOfWight.setOrganizer( bob );
		bob.getOrganizedHikes().add( isleOfWight );

		// persist organizer (will be cascaded to hikes)
		entityManager.persist( bob );

		entityManager.getTransaction().commit();

		// get a new EM to make sure data is actually retrieved from the store and not Hibernate’s internal cache
		entityManager.close();
		entityManager = entityManagerFactory.createEntityManager();

		// load it back
		entityManager.getTransaction().begin();

		Person loadedPerson = entityManager.find( Person.class, bob.getId() );
		assertThat( loadedPerson ).isNotNull();
		assertThat( loadedPerson.getFirstName() ).isEqualTo( "Bob" );
		assertThat( loadedPerson.getOrganizedHikes() ).onProperty( "description" ).containsOnly( "Visiting Land's End", "Exploring Carisbrooke Castle" );

		entityManager.getTransaction().commit();

		entityManager.close();
	}
}