/*
* 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.datastore.mongodb.test.dialectinvocations;
import static org.fest.assertions.Assertions.assertThat;
import static org.junit.Assert.assertNotNull;
import java.util.List;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.ogm.cfg.OgmProperties;
import org.hibernate.ogm.dialect.impl.ForwardingGridDialect;
import org.hibernate.ogm.dialect.spi.GridDialect;
import org.hibernate.ogm.utils.InvokedOperationsLoggingDialect;
import org.hibernate.ogm.utils.OgmTestCase;
import org.junit.Before;
import org.junit.Test;
/**
* Asserts the number and order of grid dialect operations.
*
* @author Gunnar Morling
*/
public class GridDialectOperationInvocationsTest extends OgmTestCase {
@Before
public void resetOperationsLog() {
getOperationsLogger().reset();
}
@Test
public void insertUpdateAndDelete() throws Exception {
Session session = openSession();
Transaction transaction = session.beginTransaction();
// insert
StockItem item = new StockItem();
item.setId( "item-1" );
item.setItemName( "Fairway Wood 19°" );
item.setCount( 25 );
session.persist( item );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
// load and update
StockItem loadedItem = (StockItem) session.get( StockItem.class, item.getId() );
assertNotNull( "Cannot load persisted object", loadedItem );
loadedItem.setCount( 24 );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
// delete
loadedItem = (StockItem) session.get( StockItem.class, item.getId() );
assertNotNull( "Cannot load persisted object", loadedItem );
session.delete( loadedItem );
transaction.commit();
session.clear();
session.close();
assertThat( getOperations() ).containsExactly(
"getTuple",
"createTuple",
"executeBatch",
"getTuple",
"executeBatch",
"getTuple",
"executeBatch"
);
}
@Test
public void insertGetOfEntityAndElementCollectionAndDelete() throws Exception {
Session session = openSession();
Transaction transaction = session.beginTransaction();
// insert
Product product = new Product(
"product-1",
"Hybrid 25°",
new Vendor( "Leaveland", 10 ), new Vendor( "Headaway", 9 )
);
session.persist( product );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
// load and update
Product loadedProduct = (Product) session.get( Product.class, "product-1" );
assertNotNull( "Cannot load persisted object", loadedProduct );
assertThat( loadedProduct.getVendors() ).onProperty( "name" ).containsOnly( "Leaveland", "Headaway" );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
// delete
loadedProduct = (Product) session.get( Product.class, "product-1" );
assertNotNull( "Cannot load persisted object", loadedProduct );
session.delete( loadedProduct );
transaction.commit();
session.clear();
session.close();
assertThat( getOperations() ).containsExactly(
"getTuple",
"createTuple",
"getAssociation",
"createAssociation",
"insertOrUpdateAssociation",
"executeBatch",
"getTuple",
"getAssociation",
"executeBatch",
"getTuple",
"getAssociation",
"removeAssociation",
"removeTuple",
"executeBatch"
);
}
@Test
public void insertAndUpdateInSameTransaction() throws Exception {
Session session = openSession();
Transaction transaction = session.beginTransaction();
// insert
StockItem item = new StockItem();
item.setId( "item-1" );
item.setItemName( "Fairway Wood 19°" );
item.setCount( 25 );
session.persist( item );
session.flush();
// update
item.setCount( 24 );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
StockItem loadedItem = (StockItem) session.get( StockItem.class, item.getId() );
assertNotNull( "Cannot load persisted object", loadedItem );
session.delete( loadedItem );
transaction.commit();
session.close();
assertThat( getOperations() ).containsExactly(
"getTuple",
"createTuple",
"executeBatch",
"executeBatch",
"getTuple",
"executeBatch"
);
}
@Test
public void insertSearchUpdateDelete() throws Exception {
Session session = openSession();
Transaction transaction = session.beginTransaction();
// insert
StockItem item = new StockItem();
item.setId( "item-1" );
item.setItemName( "Fairway Wood 19°" );
item.setCount( 25 );
session.persist( item );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
// search and update
StockItem loadedItem = (StockItem) session.createQuery( "from StockItem si" ).uniqueResult();
assertNotNull( "Cannot load persisted object", loadedItem );
loadedItem.setCount( 24 );
transaction.commit();
session.clear();
transaction = session.beginTransaction();
loadedItem = (StockItem) session.get( StockItem.class, item.getId() );
assertNotNull( "Cannot load persisted object", loadedItem );
assertThat( loadedItem.getCount() ).isEqualTo( 24 );
session.delete( loadedItem );
transaction.commit();
session.close();
assertThat( getOperations() ).containsExactly(
"getTuple",
"createTuple",
"executeBatch",
"executeBatch",
"executeBackendQuery",
"executeBatch",
"getTuple",
"executeBatch"
);
}
@Override
protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[] { StockItem.class, Product.class };
}
@Override
protected void configure(Configuration cfg) {
cfg.getProperties().put( OgmProperties.GRID_DIALECT, InvokedOperationsLoggingDialect.class );
}
private InvokedOperationsLoggingDialect getOperationsLogger() {
GridDialect gridDialect = sfi().getServiceRegistry().getService( GridDialect.class );
if ( gridDialect instanceof ForwardingGridDialect ) {
while ( gridDialect instanceof ForwardingGridDialect ) {
gridDialect = ( (ForwardingGridDialect<?>) gridDialect ).getGridDialect();
if ( gridDialect instanceof InvokedOperationsLoggingDialect ) {
return (InvokedOperationsLoggingDialect) gridDialect;
}
}
}
else if ( gridDialect instanceof InvokedOperationsLoggingDialect ) {
return (InvokedOperationsLoggingDialect) gridDialect;
}
throw new IllegalStateException( "Unexpected dialect type: " + gridDialect.getClass() );
}
private List<String> getOperations() {
return getOperationsLogger().getOperations();
}
}