Examples with AppIdStringGeneratorItem org.datanucleus.tests.applicationid.AppIdStringGeneratorItem used on opensource projects

Example 1 with AppIdStringGeneratorItem

use of org.datanucleus.tests.applicationid.AppIdStringGeneratorItem in project tests by datanucleus.

the class ValueGeneratorTest method testAppIdentityGenerator.

/**
 * Test the use of "identity" generator for application-identity.
 */
public void testAppIdentityGenerator() throws Exception {
    try {
        HashSet idSet = new HashSet();
        PersistenceManager pm = pmf.getPersistenceManager();
        Transaction tx = pm.currentTransaction();
        try {
            tx.begin();
            // Create a few objects.
            for (int i = 0; i < 4; i++) {
                AppIdStringGeneratorItem item = new AppIdStringGeneratorItem("First item");
                pm.makePersistent(item);
                pm.flush();
                idSet.add(JDOHelper.getObjectId(item));
            }
            tx.commit();
            assertEquals("Number of unique ids persisted is wrong", 4, idSet.size());
        } catch (JDOUserException e) {
            fail("Exception thrown during insert of objects with \"identity\" generator : " + e.getMessage());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
        pm = pmf.getPersistenceManager();
        tx = pm.currentTransaction();
        try {
            tx.begin();
            Iterator idIter = idSet.iterator();
            while (idIter.hasNext()) {
                Object id = idIter.next();
                AppIdStringGeneratorItem obj = (AppIdStringGeneratorItem) pm.getObjectById(id);
                assertEquals("Incorrect name", "First item", obj.getName());
            }
            tx.commit();
        } catch (JDOUserException ue) {
            assertTrue("Exception thrown during test " + ue.getMessage(), false);
        } 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(AppIdStringGeneratorItem.class, true));
            Collection c = (Collection) q.execute();
            // Check on the number of items
            assertTrue("Number of AppIdStringGeneratorItems retrieved is incorrect (" + c.size() + ") : should have been 4", c.size() == 4);
            Iterator<AppIdStringGeneratorItem> iter = c.iterator();
            while (iter.hasNext()) {
                AppIdStringGeneratorItem item = iter.next();
                Object id = JDOHelper.getObjectId(item);
                if (idSetCopy.contains(id)) {
                    idSetCopy.remove(id);
                }
            }
            tx.commit();
            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(AppIdStringGeneratorItem.class);
    }
}