/*
* 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.collection.types;
import static org.fest.assertions.Assertions.assertThat;
import static org.hibernate.ogm.utils.TestHelper.getCurrentDialectType;
import static org.hibernate.ogm.utils.TestHelper.getNumberOfAssociations;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.ogm.datastore.document.options.AssociationStorageType;
import org.hibernate.ogm.utils.OgmTestCase;
import org.junit.Test;
/**
* @author Emmanuel Bernard <emmanuel@hibernate.org>
*/
public class MapTest extends OgmTestCase {
@Test
public void testMapAndElementCollection() throws Exception {
Session session = openSession();
Transaction tx = session.beginTransaction();
Address home = new Address();
home.setCity( "Paris" );
Address work = new Address();
work.setCity( "San Francisco" );
User user = new User();
user.getAddresses().put( "home", home );
user.getAddresses().put( "work", work );
user.getNicknames().add( "idrA" );
user.getNicknames().add( "day[9]" );
session.persist( home );
session.persist( work );
session.persist( user );
User user2 = new User();
user2.getNicknames().add( "idrA" );
user2.getNicknames().add( "day[9]" );
session.persist( user2 );
tx.commit();
session.clear();
if ( getCurrentDialectType().isDocumentStore() ) {
assertThat( getNumberOfAssociations( sessions, AssociationStorageType.IN_ENTITY ) )
.describedAs( "Map contents and element collections should be stored within the entity document" )
.isEqualTo( 3 );
assertThat( getNumberOfAssociations( sessions, AssociationStorageType.ASSOCIATION_DOCUMENT ) )
.describedAs( "Map contents should be stored within the entity document" )
.isEqualTo( 0 );
}
tx = session.beginTransaction();
user = (User) session.get( User.class, user.getId() );
assertThat( user.getNicknames() ).as( "Should have 2 nick1" ).hasSize( 2 );
assertThat( user.getNicknames() ).as( "Should contain nicks" ).contains( "idrA", "day[9]" );
user.getNicknames().remove( "idrA" );
tx.commit();
session.clear();
tx = session.beginTransaction();
user = (User) session.get( User.class, user.getId() );
// TODO do null value
assertThat( user.getAddresses() ).as( "Map should have 2 elements" ).hasSize( 2 );
assertThat( user.getAddresses().get( "home" ).getCity() ).as( "home address should be under home" ).isEqualTo(
home.getCity() );
assertThat( user.getNicknames() ).as( "Should have 1 nick1" ).hasSize( 1 );
assertThat( user.getNicknames() ).as( "Should contain nick" ).contains( "day[9]" );
session.delete( user );
session.delete( session.load( Address.class, home.getId() ) );
session.delete( session.load( Address.class, work.getId() ) );
user2 = (User) session.get( User.class, user2.getId() );
assertThat( user2.getNicknames() ).as( "Should have 2 nicks" ).hasSize( 2 );
assertThat( user2.getNicknames() ).as( "Should contain nick" ).contains( "idrA", "day[9]" );
session.delete( user2 );
tx.commit();
session.close();
checkCleanCache();
}
@Test
public void testRemovalOfMapEntry() throws Exception {
// Create user with two addresses
Session session = openSession();
Transaction tx = session.beginTransaction();
Address home = new Address();
home.setCity( "Paris" );
Address work = new Address();
work.setCity( "San Francisco" );
User user = new User();
user.getAddresses().put( "home", home );
user.getAddresses().put( "work", work );
session.persist( home );
session.persist( work );
session.persist( user );
tx.commit();
session.clear();
// Load user and remove one address
tx = session.beginTransaction();
user = (User) session.get( User.class, user.getId() );
user.getAddresses().remove( "work" );
tx.commit();
session.clear();
// assert
tx = session.beginTransaction();
user = (User) session.get( User.class, user.getId() );
assertThat( user.getAddresses() ).hasSize( 1 );
assertThat( user.getAddresses().containsKey( "home" ) ).isTrue();
assertThat( user.getAddresses().get( "home" ).getCity() ).isEqualTo( home.getCity() );
// clean up
session.delete( user );
session.delete( session.load( Address.class, home.getId() ) );
session.delete( session.load( Address.class, work.getId() ) );
tx.commit();
session.close();
checkCleanCache();
}
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[] { User.class, Address.class };
}
}