Add tests of bulk deletes/updates

git-svn-id: https://svn.jboss.org/repos/hibernate/core/trunk@14370 1b8cb986-b30d-0410-93ca-fae66ebed9b2

cache-jbosscache2/src/test/java/org/hibernate/test/cache/jbc2/functional/bulk/OptimisticBulkOperationsTest.java

@@ -0,0 +1,57 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+
+package org.hibernate.test.cache.jbc2.functional.bulk;
+
+
+/**
+ * Optimistic version of BulkOperationsUnitTestCase.
+ * 
+ * @author Brian Stansberry
+ *
+ */
+public class OptimisticBulkOperationsTest 
+   extends PessimisticBulkOperationsTest
+{
+
+   /**
+    * @param name
+    */
+   public OptimisticBulkOperationsTest(String name)
+   {
+      super(name);
+   }
+
+   @Override
+   protected String getEntityCacheConfigName()
+   {
+      return "optimistic-entity";
+   }
+
+   public void testBulkOperationsFailureExpected() throws Exception
+   {
+      super.testBulkOperations();
+   }
+   
+   
+
+}

cache-jbosscache2/src/test/java/org/hibernate/test/cache/jbc2/functional/bulk/PessimisticBulkOperationsTest.java

@@ -0,0 +1,307 @@
+/*
+ * JBoss, Home of Professional Open Source.
+ * Copyright 2006, Red Hat Middleware LLC, and individual contributors
+ * as indicated by the @author tags. See the copyright.txt file in the
+ * distribution for a full listing of individual contributors.
+ *
+ * This is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of
+ * the License, or (at your option) any later version.
+ *
+ * This software is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this software; if not, write to the Free
+ * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
+ * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
+ */
+package org.hibernate.test.cache.jbc2.functional.bulk;
+
+import java.util.HashSet;
+import java.util.List;
+import java.util.Set;
+
+import org.hibernate.FlushMode;
+import org.hibernate.cache.RegionFactory;
+import org.hibernate.cache.jbc2.MultiplexedJBossCacheRegionFactory;
+import org.hibernate.cache.jbc2.builder.MultiplexingCacheInstanceManager;
+import org.hibernate.cfg.Configuration;
+import org.hibernate.classic.Session;
+import org.hibernate.test.cache.jbc2.functional.CacheTestCaseBase;
+import org.hibernate.test.cache.jbc2.functional.Contact;
+import org.hibernate.test.cache.jbc2.functional.Customer;
+import org.hibernate.test.tm.SimpleJtaTransactionManagerImpl;
+import org.hibernate.transaction.CMTTransactionFactory;
+
+/**
+ * Sample client for the jboss container.
+ *
+ * @author Brian Stansberry
+ * @version $Id: EntityUnitTestCase.java 60697 2007-02-20 05:08:31Z bstansberry@jboss.com $
+ */
+public class PessimisticBulkOperationsTest
+extends CacheTestCaseBase
+{
+   public PessimisticBulkOperationsTest(String name)
+   {
+      super(name);
+   } 
+
+   @Override
+   protected Class<? extends RegionFactory> getCacheRegionFactory()
+   {
+      return MultiplexedJBossCacheRegionFactory.class;
+   }
+
+   @Override
+   protected void configureCacheFactory(Configuration cfg)
+   {
+      cfg.setProperty(MultiplexingCacheInstanceManager.ENTITY_CACHE_RESOURCE_PROP, 
+                      getEntityCacheConfigName());
+   }
+
+   protected String getEntityCacheConfigName() {
+       return "pessimistic-entity";
+   }
+
+   @Override
+   protected boolean getUseQueryCache()
+   {
+      return false;
+   }
+
+   @Override
+   protected Class getTransactionFactoryClass() {
+       return CMTTransactionFactory.class;
+   }
+   
+   public void testBulkOperations() throws Exception
+   {
+      System.out.println("*** testBulkOperations()");
+      try
+      {
+         createContacts();
+         
+         List<Integer> rhContacts = getContactsByCustomer("Red Hat");
+         assertNotNull("Red Hat contacts exist", rhContacts);
+         assertEquals("Created expected number of Red Hat contacts", 10, rhContacts.size());
+         
+         assertEquals("Deleted all Red Hat contacts", 10, deleteContacts());
+         
+         List<Integer> jbContacts = getContactsByCustomer("JBoss");
+         assertNotNull("JBoss contacts exist", jbContacts);
+         assertEquals("JBoss contacts remain", 10, jbContacts.size());
+         
+         for (Integer id : rhContacts)
+         {
+            assertNull("Red Hat contact " + id + " cannot be retrieved",
+                       getContact(id));
+         }
+         rhContacts = getContactsByCustomer("Red Hat");
+         if (rhContacts != null)
+         {
+            assertEquals("No Red Hat contacts remain", 0, rhContacts.size());
+         }
+         
+         updateContacts("Kabir", "Updated");
+         for (Integer id : jbContacts)
+         {
+            Contact contact = getContact(id);
+            assertNotNull("JBoss contact " + id + " exists", contact);
+            String expected = ("Kabir".equals(contact.getName())) ? "Updated" : "2222";
+            assertEquals("JBoss contact " + id + " has correct TLF",
+                         expected, contact.getTlf());
+         }
+         
+         List<Integer> updated = getContactsByTLF("Updated");
+         assertNotNull("Got updated contacts", updated);
+         assertEquals("Updated contacts", 5, updated.size());
+      }
+      finally
+      {
+         // cleanup the db so we can run this test multiple times w/o restarting the cluster
+         cleanup();
+      }
+   }
+   
+   public void createContacts() throws Exception
+   {
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          for (int i = 0; i < 10; i++)
+              createCustomer(i);
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+      }
+      catch (Exception e) {
+         SimpleJtaTransactionManagerImpl.getInstance().rollback();
+         throw e;
+      }
+   }
+
+   public int deleteContacts() throws Exception
+   {
+      String deleteHQL = "delete Contact where customer in ";
+      deleteHQL += " (select customer FROM Customer as customer ";
+      deleteHQL += " where customer.name = :cName)";
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          int rowsAffected = session.createQuery(deleteHQL)
+                                .setFlushMode(FlushMode.AUTO)
+                                .setParameter("cName", "Red Hat")
+                                .executeUpdate();
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+          return rowsAffected;
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }
+   }
+   
+   public List<Integer> getContactsByCustomer(String customerName)
+       throws Exception
+   {
+      String selectHQL = "select contact.id from Contact contact";
+      selectHQL += " where contact.customer.name = :cName";
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          List results = session.createQuery(selectHQL)
+                            .setFlushMode(FlushMode.AUTO)
+                            .setParameter("cName", customerName)
+                            .list();
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+          return results;  
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }    
+   }
+   
+   public List<Integer> getContactsByTLF(String tlf) throws Exception
+   {
+      String selectHQL = "select contact.id from Contact contact";
+      selectHQL += " where contact.tlf = :cTLF";
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          List results = session.createQuery(selectHQL)
+                            .setFlushMode(FlushMode.AUTO)
+                            .setParameter("cTLF", tlf)
+                            .list();
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+          return results;       
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }     
+   }
+
+   public int updateContacts(String name, String newTLF) throws Exception
+   {
+      String updateHQL = "update Contact set tlf = :cNewTLF where name = :cName";
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          int rowsAffected = session.createQuery(updateHQL)
+                                .setFlushMode(FlushMode.AUTO)
+                                .setParameter("cNewTLF", newTLF)
+                                .setParameter("cName", name)
+                                .executeUpdate();
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+          return rowsAffected;       
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }     
+   }
+   
+   public Contact getContact(Integer id) throws Exception
+   {
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          Contact contact = (Contact) session.get(Contact.class, id);
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+          return contact;       
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }     
+   }
+   
+   public void cleanup() throws Exception
+   {
+      String deleteContactHQL = "delete from Contact";
+      String deleteCustomerHQL = "delete from Customer";
+
+      SimpleJtaTransactionManagerImpl.getInstance().begin();
+      try {
+          
+          Session session = getSessions().getCurrentSession();
+          session.createQuery(deleteContactHQL)
+                 .setFlushMode(FlushMode.AUTO)
+                 .executeUpdate();
+          session.createQuery(deleteCustomerHQL)
+                 .setFlushMode(FlushMode.AUTO)
+                 .executeUpdate();
+          SimpleJtaTransactionManagerImpl.getInstance().commit();
+      }
+      catch (Exception e) {
+          SimpleJtaTransactionManagerImpl.getInstance().rollback();
+          throw e;         
+      }
+      
+   }
+   
+   private Customer createCustomer(int id) throws Exception
+   {
+      System.out.println("CREATE CUSTOMER " + id);
+      try
+      {
+         Customer customer = new Customer();
+         customer.setName((id % 2 == 0) ? "JBoss" : "Red Hat");
+         Set<Contact> contacts = new HashSet<Contact>();
+         
+         Contact kabir = new Contact();
+         kabir.setCustomer(customer);
+         kabir.setName("Kabir");
+         kabir.setTlf("1111");
+         contacts.add(kabir);
+         
+         Contact bill = new Contact();
+         bill.setCustomer(customer);
+         bill.setName("Bill");
+         bill.setTlf("2222");
+         contacts.add(bill);
+
+         customer.setContacts(contacts);
+
+         getSessions().getCurrentSession().persist(customer);
+         return customer;
+      }
+      finally
+      {
+         System.out.println("CREATE CUSTOMER " +  id + " -  END");         
+      }
+   }
+}