Search in sources :

Example 1 with AUIDGeneratorItem

use of org.jpox.samples.valuegeneration.AUIDGeneratorItem in project tests by datanucleus.

the class ValueGeneratorTest method testAUIDGenerator.

/**
 * Test the use of "auid" generator. This is for all datastores.
 */
public void testAUIDGenerator() throws Exception {
    try {
        HashSet idSet = new HashSet();
        Class idClass = null;
        PersistenceManager pm = pmf.getPersistenceManager();
        Transaction tx = pm.currentTransaction();
        try {
            tx.begin();
            // Create a few objects.
            AUIDGeneratorItem item = null;
            item = new AUIDGeneratorItem("First item");
            pm.makePersistent(item);
            idSet.add(item.getIdentifier());
            item = new AUIDGeneratorItem("Second item");
            pm.makePersistent(item);
            idSet.add(item.getIdentifier());
            item = new AUIDGeneratorItem("Third item");
            pm.makePersistent(item);
            idSet.add(item.getIdentifier());
            item = new AUIDGeneratorItem("Fourth item");
            pm.makePersistent(item);
            idSet.add(item.getIdentifier());
            idClass = JDOHelper.getObjectId(item).getClass();
            tx.commit();
        } catch (JDOUserException e) {
            fail("Exception thrown during insert of objects in \"auid\" generator test " + e.getMessage());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
        pm = pmf.getPersistenceManager();
        tx = pm.currentTransaction();
        try {
            tx.begin();
            HashSet idSetCopy = new HashSet(idSet);
            // Retrieve the items
            Query q = pm.newQuery(pm.getExtent(AUIDGeneratorItem.class, true));
            Collection c = (Collection) q.execute();
            // Check on the number of items
            assertTrue("Number of AUIDGeneratorItem's retrieved is incorrect (" + c.size() + ") : should have been 4", c.size() == 4);
            Iterator iter = c.iterator();
            while (iter.hasNext()) {
                Object o = iter.next();
                if (AUIDGeneratorItem.Oid.class.equals(idClass)) {
                    idSetCopy.remove(((AUIDGeneratorItem) o).getIdentifier());
                }
            }
            tx.commit();
            if (AUIDGeneratorItem.Oid.class.equals(idClass)) {
                assertEquals("Wrong number of different IDs", 4, idSet.size());
                assertTrue("Retrieved IDs did not match created IDs", 0 == idSetCopy.size());
            }
        } catch (JDOUserException ue) {
            assertTrue("Exception thrown during test " + ue.getMessage(), false);
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
    } finally {
        clean(AUIDGeneratorItem.class);
    }
}
Also used : AUIDGeneratorItem(org.jpox.samples.valuegeneration.AUIDGeneratorItem) Transaction(javax.jdo.Transaction) Query(javax.jdo.Query) PersistenceManager(javax.jdo.PersistenceManager) Iterator(java.util.Iterator) Collection(java.util.Collection) JDOUserException(javax.jdo.JDOUserException) HashSet(java.util.HashSet)

Aggregations

Collection (java.util.Collection)1 HashSet (java.util.HashSet)1 Iterator (java.util.Iterator)1 JDOUserException (javax.jdo.JDOUserException)1 PersistenceManager (javax.jdo.PersistenceManager)1 Query (javax.jdo.Query)1 Transaction (javax.jdo.Transaction)1 AUIDGeneratorItem (org.jpox.samples.valuegeneration.AUIDGeneratorItem)1