Search in sources :

Example 1 with MapFKKeyItem

use of org.jpox.samples.one_many.map.MapFKKeyItem in project tests by datanucleus.

the class MapForeignKeyTest method testMapWithValueAsFieldInKey.

/**
 * Test for persistence of a Map with the value stored in the key object.
 */
public void testMapWithValueAsFieldInKey() {
    try {
        Object containerId = null;
        MapHolder container = new MapHolder();
        MapFKKeyItem item1 = new MapFKKeyItem("First", "First element", "Item1");
        MapFKKeyItem item3 = new MapFKKeyItem("Third", "Third element", "Item3");
        MapFKKeyItem item2 = new MapFKKeyItem("Second", "Second element", "Item2");
        container.getFkMapValue().put(item1, item1.getValue());
        container.getFkMapValue().put(item3, item3.getValue());
        container.getFkMapValue().put(item2, item2.getValue());
        // Persist the objects
        PersistenceManager pm = pmf.getPersistenceManager();
        Transaction tx = pm.currentTransaction();
        try {
            tx.begin();
            pm.makePersistent(container);
            tx.commit();
            containerId = JDOHelper.getObjectId(container);
        } catch (Exception e) {
            e.printStackTrace();
            fail(e.toString());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
        // Retrieve the object and inspect it
        pm = pmf.getPersistenceManager();
        tx = pm.currentTransaction();
        try {
            tx.begin();
            container = (MapHolder) pm.getObjectById(containerId);
            assertTrue("Container map was not found!", container != null);
            Map map = container.getFkMapValue();
            assertEquals("Number of items in the container map is incorrect", map.size(), 3);
            // Check Entry set
            Set entries = map.entrySet();
            Iterator entryIter = entries.iterator();
            while (entryIter.hasNext()) {
                Map.Entry entry = (Map.Entry) entryIter.next();
                MapFKKeyItem item = (MapFKKeyItem) entry.getKey();
                if (entry.getValue().equals("Item1")) {
                    assertEquals("item has incorrect name for value Item1", item.getName(), "First");
                } else if (entry.getValue().equals("Item2")) {
                    assertEquals("item has incorrect name for value Item2", item.getName(), "Second");
                } else if (entry.getValue().equals("Item3")) {
                    assertEquals("item has incorrect name for value Item3", item.getName(), "Third");
                } else {
                    fail("Unknown Map entry found with value " + entry.getValue());
                }
            }
            // Check Key set
            Set keys = map.keySet();
            assertEquals("Number of keys in Map.keySet() is incorrect", keys.size(), 3);
            Iterator keyIter = keys.iterator();
            boolean item1Present = false;
            boolean item2Present = false;
            boolean item3Present = false;
            while (keyIter.hasNext()) {
                Object obj = keyIter.next();
                assertEquals("Type of key objects returned from Map.keySet().iterator() is incorrect", obj.getClass().getName(), MapFKKeyItem.class.getName());
                MapFKKeyItem key = (MapFKKeyItem) obj;
                if (key.getName().equals("First")) {
                    item1Present = true;
                } else if (key.getName().equals("Second")) {
                    item2Present = true;
                } else if (key.getName().equals("Third")) {
                    item3Present = true;
                }
            }
            assertTrue("Item1 was not present in the keySet", item1Present);
            assertTrue("Item2 was not present in the keySet", item2Present);
            assertTrue("Item3 was not present in the keySet", item3Present);
            // Check Value set
            Collection values = map.values();
            assertEquals("Number of values in Map.values() is incorrect", values.size(), 3);
            Iterator valueIter = values.iterator();
            item1Present = false;
            item2Present = false;
            item3Present = false;
            while (valueIter.hasNext()) {
                Object obj = valueIter.next();
                assertEquals("Type of value objects returned from Map.values().iterator() is incorrect", obj.getClass().getName(), String.class.getName());
                String value = (String) obj;
                if (value.equals("Item1")) {
                    item1Present = true;
                } else if (value.equals("Item2")) {
                    item2Present = true;
                } else if (value.equals("Item3")) {
                    item3Present = true;
                }
            }
            assertTrue("Item1 was not present in the values()", item1Present);
            assertTrue("Item2 was not present in the values()", item2Present);
            assertTrue("Item3 was not present in the values()", item3Present);
            tx.commit();
        } catch (Exception e) {
            e.printStackTrace();
            fail(e.toString());
        } finally {
            if (tx.isActive()) {
                tx.rollback();
            }
            pm.close();
        }
    } finally {
        // Clean out our data
        clean(MapHolder.class);
    }
}
Also used : Set(java.util.Set) MapHolder(org.jpox.samples.one_many.map.MapHolder) PersistenceManager(javax.jdo.PersistenceManager) MapFKKeyItem(org.jpox.samples.one_many.map.MapFKKeyItem) Transaction(javax.jdo.Transaction) Iterator(java.util.Iterator) Collection(java.util.Collection) Map(java.util.Map)

Aggregations

Collection (java.util.Collection)1 Iterator (java.util.Iterator)1 Map (java.util.Map)1 Set (java.util.Set)1 PersistenceManager (javax.jdo.PersistenceManager)1 Transaction (javax.jdo.Transaction)1 MapFKKeyItem (org.jpox.samples.one_many.map.MapFKKeyItem)1 MapHolder (org.jpox.samples.one_many.map.MapHolder)1